diff options
Diffstat (limited to 'f1elo/db.py')
| -rw-r--r-- | f1elo/db.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/f1elo/db.py b/f1elo/db.py index 554a061..58e7a8a 100644 --- a/f1elo/db.py +++ b/f1elo/db.py @@ -5,13 +5,16 @@ import __main__ from sqlalchemy import create_engine, event from sqlalchemy.orm import sessionmaker -config = json.load(open(path.join(path.dirname(__main__.__file__), 'config', 'db.json'))) +config = json.load( + open(path.join(path.dirname(__main__.__file__), 'config', 'db.json'))) if config['engine'] == 'mysql': - engine = create_engine("mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}?charset=utf8".format(config)) + engine = create_engine( + "mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}?charset=utf8".format(config)) elif config['engine'] == 'sqlite': engine = create_engine("sqlite:///{0[file]}".format(config)) + def fk_pragma(conn, record): - conn.execute('PRAGMA FOREIGN_KEYS=ON'); + conn.execute('PRAGMA FOREIGN_KEYS=ON') event.listen(engine, 'connect', fk_pragma) Session = sessionmaker(bind=engine) |