rainbowstream/table_def.py

   1 from sqlalchemy import *
   2 from sqlalchemy.ext.declarative import declarative_base
   3
   4 engine = create_engine('sqlite:///rainbow.db', echo=False)
   5 Base = declarative_base()
   6
   7
   8 class Tweet(Base):
   9
  10     __tablename__ = "tweet"
  11
  12     rainbow_id = Column(Integer, primary_key=True)
  13     tweet_id = Column(Integer)
  14
  15     def __init__(self, tweet_id):
  16         self.tweet_id = tweet_id
  17
  18
  19 class Message(Base):
  20
  21     __tablename__ = "message"
  22
  23     rainbow_id = Column(Integer, primary_key=True)
  24     message_id = Column(Integer)
  25
  26     def __init__(self, message_id):
  27         self.message_id = message_id
  28
  29
  30 class Theme(Base):
  31
  32     __tablename__ = "theme"
  33
  34     theme_id = Column(Integer, primary_key=True)
  35     theme_name = Column(String(20))
  36
  37     def __init__(self, theme_name):
  38         self.theme_name = theme_name
  39
  40
  41 class Semaphore(Base):
  42
  43     __tablename__ = "semaphore"
  44
  45     semaphore_id = Column(Integer, primary_key=True)
  46     flag = Column(Boolean)
  47
  48     def __init__(self, flag):
  49         self.flag = flag
  50
  51
  52 def init_db():
  53     Base.metadata.create_all(engine)