From da87925da621f45ed72d8e3aa7b79d3d8eb9a0ca Mon Sep 17 00:00:00 2001
From: emkael <emkael@tlen.pl>
Date: Fri, 17 Oct 2014 16:56:44 +0200
Subject:  * DB generation from model metadata

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

(limited to 'f1elo/interface.py')

diff --git a/f1elo/interface.py b/f1elo/interface.py
index 2065160..4b938ae 100644
--- a/f1elo/interface.py
+++ b/f1elo/interface.py
@@ -1,6 +1,8 @@
 import datetime
 import dateutil.relativedelta
 
+from sqlalchemy import MetaData
+
 from f1elo.db import Session
 from f1elo.elo import Elo
 from f1elo.model import *
@@ -10,6 +12,12 @@ class Interface:
         self.session = Session()
         self.date = date
 
+    def init_db(self, force=False):
+        from f1elo.model import Base
+        if force:
+            Base.metadata.drop_all(self.session.get_bind())
+        Base.metadata.create_all(self.session.get_bind())
+
     def reset(self, date=None, _debug=False):
         if date is None:
             date = self.date
-- 
cgit v1.2.3