autopep8 and bumped version

[rainbowstream.git] / rainbowstream / table_def.py

diff --git a/rainbowstream/table_def.py b/rainbowstream/table_def.py
index f8597f6c47242eca01d3e701cf44b95f7bfd3523..822fa1c7ff53d4d9da09eca6cee1ea0f56dfd820 100644 (file)
--- a/rainbowstream/table_def.py
+++ b/rainbowstream/table_def.py
@@ -4,9 +4,10 @@ from sqlalchemy.ext.declarative import declarative_base
 engine = create_engine('sqlite:///rainbow.db', echo=False)
 Base = declarative_base()
 
-class Map(Base):
 
-    __tablename__ = "map"
+class Tweet(Base):
+
+    __tablename__ = "tweet"
 
     rainbow_id = Column(Integer, primary_key=True)
     tweet_id = Column(Integer)
@@ -14,5 +15,39 @@ class Map(Base):
     def __init__(self, tweet_id):
         self.tweet_id = tweet_id
 
+
+class Message(Base):
+
+    __tablename__ = "message"
+
+    rainbow_id = Column(Integer, primary_key=True)
+    message_id = Column(Integer)
+
+    def __init__(self, message_id):
+        self.message_id = message_id
+
+
+class Theme(Base):
+
+    __tablename__ = "theme"
+
+    theme_id = Column(Integer, primary_key=True)
+    theme_name = Column(String(20))
+
+    def __init__(self, theme_name):
+        self.theme_name = theme_name
+
+
+class Semaphore(Base):
+
+    __tablename__ = "semaphore"
+
+    semaphore_id = Column(Integer, primary_key=True)
+    flag = Column(Boolean)
+
+    def __init__(self, flag):
+        self.flag = flag
+
+
 def init_db():
-    Base.metadata.create_all(engine)
\ No newline at end of file
+    Base.metadata.create_all(engine)