fix table_def
[rainbowstream.git] / rainbowstream / table_def.py
index f2435ab..fda3948 100644 (file)
@@ -27,5 +27,16 @@ class Message(Base):
         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)