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 def init_db():
  31     Base.metadata.create_all(engine)