From cd1810a441230188b58d834e68cd15975a58cdb4 Mon Sep 17 00:00:00 2001
From: emkael <emkael@tlen.pl>
Date: Wed, 30 Nov 2016 13:42:37 +0100
Subject:  * loading result SQL on DB init

---
 f1elo/interface.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'f1elo')

diff --git a/f1elo/interface.py b/f1elo/interface.py
index 6cca847..da1b77f 100644
--- a/f1elo/interface.py
+++ b/f1elo/interface.py
@@ -2,7 +2,9 @@ from __future__ import print_function
 
 import datetime
 import sys
+from os import path
 
+import __main__
 import dateutil.relativedelta
 from f1elo.db import Session
 from f1elo.elo import Elo
@@ -21,6 +23,11 @@ class Interface:
         if force:
             Base.metadata.drop_all(self.session.get_bind())
         Base.metadata.create_all(self.session.get_bind())
+        with open(path.join(path.dirname(__main__.__file__), 'sql', 'results.sql')) as result_dump:
+            for line in result_dump.readlines():
+                line = line.strip()
+                if line and line[0:2] != '--':
+                    self.session.bind.execute(line)
 
     def reset(self, date=None, _debug=False):
         if date is None:
-- 
cgit v1.2.3