From 8249784c586a088c09598305fd408f376aad26fb Mon Sep 17 00:00:00 2001 From: emkael Date: Mon, 6 Oct 2014 15:51:30 +0200 Subject: * initial commit --- f1elo/db.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 f1elo/db.py (limited to 'f1elo/db.py') diff --git a/f1elo/db.py b/f1elo/db.py new file mode 100644 index 0000000..49ccc3d --- /dev/null +++ b/f1elo/db.py @@ -0,0 +1,25 @@ +import __main__ +from os import path + +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker + +import json + +config = json.load(open(path.dirname(__main__.__file__) + '/config/db.json')) +engine = create_engine("mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}?charset=utf8".format(config)) +Session = sessionmaker(bind=engine) + +from f1elo.model import Driver + +def find_driver(name, country, session): + driver = session.query(Driver).filter(Driver.driver==name).first() + if driver: + return driver + else: + driver = Driver() + driver.driver = name + driver.country = country + session.add(driver) + return driver + -- cgit v1.2.3