2 from sqlalchemy
import create_engine
3 from sqlalchemy
.orm
import sessionmaker
4 from .table_def
import *
12 if not os
.path
.isfile('rainbow.db'):
14 self
.engine
= create_engine('sqlite:///rainbow.db', echo
=False)
16 def tweet_store(self
, tweet_id
):
20 Session
= sessionmaker(bind
=self
.engine
)
26 def rainbow_to_tweet_query(self
, rid
):
28 Query base of rainbow id
30 Session
= sessionmaker(bind
=self
.engine
)
32 res
= session
.query(Tweet
).filter_by(rainbow_id
=rid
).all()
35 def tweet_to_rainbow_query(self
, tid
):
37 Query base of tweet id
39 Session
= sessionmaker(bind
=self
.engine
)
41 res
= session
.query(Tweet
).filter_by(tweet_id
=tid
).all()
44 def message_store(self
, message_id
):
48 Session
= sessionmaker(bind
=self
.engine
)
50 m
= Message(message_id
)
54 def rainbow_to_message_query(self
, rid
):
56 Query base of rainbow id
58 Session
= sessionmaker(bind
=self
.engine
)
60 res
= session
.query(Message
).filter_by(rainbow_id
=rid
).all()
63 def message_to_rainbow_query(self
, mid
):
65 Query base of message id
67 Session
= sessionmaker(bind
=self
.engine
)
69 res
= session
.query(Message
).filter_by(message_id
=mid
).all()
72 def theme_store(self
, theme_name
):
76 Session
= sessionmaker(bind
=self
.engine
)
78 th
= Theme(theme_name
)
82 def theme_update(self
, theme_name
):
86 Session
= sessionmaker(bind
=self
.engine
)
88 res
= session
.query(Theme
).all()
90 r
.theme_name
= theme_name
93 def theme_query(self
):
97 Session
= sessionmaker(bind
=self
.engine
)
99 res
= session
.query(Theme
).all()
102 def semaphore_store(self
, flag
):
106 Session
= sessionmaker(bind
=self
.engine
)
112 def semaphore_update(self
, flag
):
114 Update semaphore flag
116 Session
= sessionmaker(bind
=self
.engine
)
118 res
= session
.query(Semaphore
).all()
123 def semaphore_query(self
):
127 Session
= sessionmaker(bind
=self
.engine
)
129 res
= session
.query(Semaphore
).all()