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

diff --git a/rainbowstream/table_def.py b/rainbowstream/table_def.py
index c312bde..fda3948 100644
--- a/rainbowstream/table_def.py
+++ b/rainbowstream/table_def.py
@@ -5,9 +5,9 @@ engine = create_engine('sqlite:///rainbow.db', echo=False)
 Base = declarative_base()
 
 
-class Map(Base):
+class Tweet(Base):
 
-    __tablename__ = "map"
+    __tablename__ = "tweet"
 
     rainbow_id = Column(Integer, primary_key=True)
     tweet_id = Column(Integer)
@@ -16,5 +16,27 @@ class Map(Base):
         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
+
+
 def init_db():
     Base.metadata.create_all(engine)