* SQLite support + config examples

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):