Revision 2715ade4 snf-pithos-backend/pithos/backends/lib/sqlalchemy/dbwrapper.py
b/snf-pithos-backend/pithos/backends/lib/sqlalchemy/dbwrapper.py | ||
---|---|---|
1 | 1 |
# Copyright 2011-2012 GRNET S.A. All rights reserved. |
2 |
#
|
|
2 |
# |
|
3 | 3 |
# Redistribution and use in source and binary forms, with or |
4 | 4 |
# without modification, are permitted provided that the following |
5 | 5 |
# conditions are met: |
6 |
#
|
|
6 |
# |
|
7 | 7 |
# 1. Redistributions of source code must retain the above |
8 | 8 |
# copyright notice, this list of conditions and the following |
9 | 9 |
# disclaimer. |
10 |
#
|
|
10 |
# |
|
11 | 11 |
# 2. Redistributions in binary form must reproduce the above |
12 | 12 |
# copyright notice, this list of conditions and the following |
13 | 13 |
# disclaimer in the documentation and/or other materials |
14 | 14 |
# provided with the distribution. |
15 |
#
|
|
15 |
# |
|
16 | 16 |
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS |
17 | 17 |
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
18 | 18 |
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
... | ... | |
25 | 25 |
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN |
26 | 26 |
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
27 | 27 |
# POSSIBILITY OF SUCH DAMAGE. |
28 |
#
|
|
28 |
# |
|
29 | 29 |
# The views and conclusions contained in the software and |
30 | 30 |
# documentation are those of the authors and should not be |
31 | 31 |
# interpreted as representing official policies, either expressed |
... | ... | |
34 | 34 |
from sqlalchemy import create_engine |
35 | 35 |
#from sqlalchemy.event import listen |
36 | 36 |
from sqlalchemy.engine import Engine |
37 |
from sqlalchemy.pool import NullPool |
|
37 |
from sqlalchemy.pool import NullPool, QueuePool
|
|
38 | 38 |
from sqlalchemy.interfaces import PoolListener |
39 | 39 |
|
40 | 40 |
|
41 | 41 |
class DBWrapper(object): |
42 | 42 |
"""Database connection wrapper.""" |
43 |
|
|
43 |
|
|
44 | 44 |
def __init__(self, db): |
45 | 45 |
if db.startswith('sqlite://'): |
46 | 46 |
class ForeignKeysListener(PoolListener): |
47 | 47 |
def connect(self, dbapi_con, con_record): |
48 | 48 |
db_cursor = dbapi_con.execute('pragma foreign_keys=ON;') |
49 |
db_cursor = dbapi_con.execute('pragma case_sensitive_like=ON;') |
|
49 |
db_cursor = dbapi_con.execute( |
|
50 |
'pragma case_sensitive_like=ON;') |
|
50 | 51 |
self.engine = create_engine(db, connect_args={'check_same_thread': False}, poolclass=NullPool, listeners=[ForeignKeysListener()]) |
51 | 52 |
#elif db.startswith('mysql://'): |
52 | 53 |
# db = '%s?charset=utf8&use_unicode=0' %db |
53 | 54 |
# self.engine = create_engine(db, convert_unicode=True) |
54 | 55 |
else: |
56 |
#self.engine = create_engine(db, pool_size=0, max_overflow=-1) |
|
55 | 57 |
self.engine = create_engine(db, poolclass=NullPool) |
56 |
#self.engine.echo = True |
|
58 |
self.engine.echo = True |
|
59 |
self.engine.echo_pool = False |
|
57 | 60 |
self.conn = self.engine.connect() |
58 | 61 |
self.trans = None |
59 |
|
|
62 |
|
|
60 | 63 |
def close(self): |
61 | 64 |
self.conn.close() |
62 |
|
|
65 |
|
|
63 | 66 |
def execute(self): |
64 | 67 |
self.trans = self.conn.begin() |
65 |
|
|
68 |
|
|
66 | 69 |
def commit(self): |
67 | 70 |
self.trans.commit() |
68 | 71 |
self.trans = None |
69 |
|
|
72 |
|
|
70 | 73 |
def rollback(self): |
71 | 74 |
self.trans.rollback() |
72 | 75 |
self.trans = None |
Also available in: Unified diff