diff options
Diffstat (limited to 'f1elo/db.py')
| -rw-r--r-- | f1elo/db.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/f1elo/db.py b/f1elo/db.py index cd7c4ac..8ecb0ac 100644 --- a/f1elo/db.py +++ b/f1elo/db.py @@ -7,7 +7,10 @@ from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker config = json.load(open(path.join(path.dirname(__main__.__file__), 'config', 'db.json'))) -engine = create_engine("mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}?charset=utf8".format(config)) +if config['engine'] == 'mysql': + engine = create_engine("mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}?charset=utf8".format(config)) +else: + engine = create_engine("sqlite:///{0[file]}".format(config)) Session = sessionmaker(bind=engine) def find_driver(name, country, session): |