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 Map(Base):
   9
  10     __tablename__ = "map"
  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 def init_db():
  20     Base.metadata.create_all(engine)