X-Git-Url: https://vcs.fsf.org/?p=rainbowstream.git;a=blobdiff_plain;f=rainbowstream%2Ftable_def.py;h=c312bde02a8313303f3feede9255ee7164b7da24;hp=b6d14c56699d7fbedf35efdbf1609554c773594e;hb=b2b933a939fc0a10b2a6300089ac2a72d2cfb17b;hpb=18cab06ae81a383f8eaf3117d3377c0424b86782

diff --git a/rainbowstream/table_def.py b/rainbowstream/table_def.py
index b6d14c5..c312bde 100644
--- a/rainbowstream/table_def.py
+++ b/rainbowstream/table_def.py
@@ -4,6 +4,7 @@ from sqlalchemy.ext.declarative import declarative_base
 engine = create_engine('sqlite:///rainbow.db', echo=False)
 Base = declarative_base()
 
+
 class Map(Base):
 
     __tablename__ = "map"
@@ -14,4 +15,6 @@ class Map(Base):
     def __init__(self, tweet_id):
         self.tweet_id = tweet_id
 
-Base.metadata.create_all(engine)
\ No newline at end of file
+
+def init_db():
+    Base.metadata.create_all(engine)