Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / backends / lib / sqlalchemy / dbwrapper.py @ 7ed99da8

History | View | Annotate | Download (3 kB)

1
# Copyright 2011-2012 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
#
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
#
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.
33

    
34
from sqlalchemy import create_engine
35
#from sqlalchemy.event import listen
36
from sqlalchemy.engine import Engine
37
from sqlalchemy.pool import NullPool, QueuePool
38
from sqlalchemy.interfaces import PoolListener
39

    
40

    
41
class DBWrapper(object):
42
    """Database connection wrapper."""
43

    
44
    def __init__(self, db):
45
        if db.startswith('sqlite://'):
46
            class ForeignKeysListener(PoolListener):
47
                def connect(self, dbapi_con, con_record):
48
                    db_cursor = dbapi_con.execute('pragma foreign_keys=ON;')
49
                    db_cursor = dbapi_con.execute(
50
                        'pragma case_sensitive_like=ON;')
51
            self.engine = create_engine(db, connect_args={'check_same_thread': False}, poolclass=NullPool, listeners=[ForeignKeysListener()])
52
        #elif db.startswith('mysql://'):
53
        #    db = '%s?charset=utf8&use_unicode=0' %db
54
        #    self.engine = create_engine(db, convert_unicode=True)
55
        else:
56
            #self.engine = create_engine(db, pool_size=0, max_overflow=-1)
57
            self.engine = create_engine(db, poolclass=NullPool)
58
        self.engine.echo = False
59
        self.engine.echo_pool = False
60
        self.conn = self.engine.connect()
61
        self.trans = None
62

    
63
    def close(self):
64
        self.conn.close()
65

    
66
    def execute(self):
67
        self.trans = self.conn.begin()
68

    
69
    def commit(self):
70
        self.trans.commit()
71
        self.trans = None
72

    
73
    def rollback(self):
74
        self.trans.rollback()
75
        self.trans = None