From d0c93a1c1813213d2f55fa9b4c6798d131d08484 Mon Sep 17 00:00:00 2001 From: emkael Date: Wed, 4 Jan 2017 00:19:11 +0100 Subject: Model database creation interface --- ausbutler/interface.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 ausbutler/interface.py (limited to 'ausbutler/interface.py') diff --git a/ausbutler/interface.py b/ausbutler/interface.py new file mode 100644 index 0000000..95696c1 --- /dev/null +++ b/ausbutler/interface.py @@ -0,0 +1,14 @@ +from .db import Session + + +class Interface: + + def __init__(self): + self.session = Session() + + def init_db(self, force=False): + from .model import Base + if force: + Base.metadata.drop_all(self.session.get_bind()) + Base.metadata.create_all(self.session.get_bind()) + -- cgit v1.2.3