theme on DB

[rainbowstream.git] / rainbowstream / table_def.py

diff --git a/rainbowstream/table_def.py b/rainbowstream/table_def.py
index f2435ab..9948ad7 100644 (file)
--- a/rainbowstream/table_def.py
+++ b/rainbowstream/table_def.py
@@ -27,5 +27,18 @@ class Message(Base):
         self.message_id = message_id
 
 
+class Theme(Base):
+
+    __tablename__ = "theme"
+
+    theme_id = Column(Integer, primary_key=True)
+    theme_name = Column(String)
+    changed = Column(Boolean, default=False)
+
+    def __init__(self, theme_name):
+        self.theme_name = theme_name
+        self.Boolean = False
+
+
 def init_db():
     Base.metadata.create_all(engine)