fix serious bug in import HTMLParser in Python > 3.4
[rainbowstream.git] / rainbowstream / table_def.py
CommitLineData
18cab06a
O
1from sqlalchemy import *
2from sqlalchemy.ext.declarative import declarative_base
3
4engine = create_engine('sqlite:///rainbow.db', echo=False)
5Base = declarative_base()
6
b2b933a9 7
305ce127 8class Tweet(Base):
18cab06a 9
305ce127 10 __tablename__ = "tweet"
18cab06a
O
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
b2b933a9 18
305ce127 19class 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
4cf86720
VNM
30class Theme(Base):
31
32 __tablename__ = "theme"
33
34 theme_id = Column(Integer, primary_key=True)
9e3418f1 35 theme_name = Column(String(20))
4cf86720
VNM
36
37 def __init__(self, theme_name):
38 self.theme_name = theme_name
4cf86720
VNM
39
40
9683e61d
O
41class 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
eb9781ed 52def init_db():
b2b933a9 53 Base.metadata.create_all(engine)