* DB generation from model metadata

1 files changed, 8 insertions, 0 deletions

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