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)
def __init__(self, tweet_id):
self.tweet_id = tweet_id
-Base.metadata.create_all(engine)
\ No newline at end of file
+
+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)