1 from sqlalchemy
import MetaData
, Table
, engine_from_config
, orm
3 from ..defaults
import get_default_db_uri
4 from .tables
import metadata
5 from .multilang
import MultilangSession
, MultilangScopedSession
8 def connect(uri
=None, session_args
={}, engine_args
={}, engine_prefix
=''):
9 """Connects to the requested URI. Returns a session object.
11 With the URI omitted, attempts to connect to a default SQLite database
12 contained within the package directory.
14 Calling this function also binds the metadata object to the created engine.
17 # If we didn't get a uri, fall back to the default
19 uri
= engine_args
.get(engine_prefix
+ 'url', None)
21 uri
= get_default_db_uri()
23 ### Do some fixery for MySQL
24 if uri
[0:5] == 'mysql':
25 # MySQL uses latin1 for connections by default even if the server is
26 # otherwise oozing with utf8; charset fixes this
27 if 'charset' not in uri
:
28 uri
+= '?charset=utf8'
30 # Tables should be InnoDB, in the event that we're creating them, and
31 # use UTF-8 goddammit!
32 for table
in metadata
.tables
.values():
33 table
.kwargs
['mysql_engine'] = 'InnoDB'
34 table
.kwargs
['mysql_charset'] = 'utf8'
37 engine_args
[engine_prefix
+ 'url'] = uri
38 engine
= engine_from_config(engine_args
, prefix
=engine_prefix
)
39 conn
= engine
.connect()
40 metadata
.bind
= engine
42 all_session_args
= dict(autoflush
=True, autocommit
=False, bind
=engine
)
43 all_session_args
.update(session_args
)
44 sm
= orm
.sessionmaker(class_
=MultilangSession
, **all_session_args
)
45 session
= MultilangScopedSession(sm
)