2 from sqlalchemy
import create_engine
3 from sqlalchemy
.orm
import sessionmaker
4 from .table_def
import *
15 if not os
.path
.isfile('rainbow.db'):
17 self
.engine
= create_engine('sqlite:///rainbow.db', echo
=False)
20 def tweet_store(self
, tweet_id
):
24 Session
= sessionmaker(bind
=self
.engine
)
31 def rainbow_to_tweet_query(self
, rid
):
33 Query base of rainbow id
35 Session
= sessionmaker(bind
=self
.engine
)
37 res
= session
.query(Tweet
).filter_by(rainbow_id
=rid
).all()
41 def tweet_to_rainbow_query(self
, tid
):
43 Query base of tweet id
45 Session
= sessionmaker(bind
=self
.engine
)
47 res
= session
.query(Tweet
).filter_by(tweet_id
=tid
).all()
51 def message_store(self
, message_id
):
55 Session
= sessionmaker(bind
=self
.engine
)
57 m
= Message(message_id
)
62 def rainbow_to_message_query(self
, rid
):
64 Query base of rainbow id
66 Session
= sessionmaker(bind
=self
.engine
)
68 res
= session
.query(Message
).filter_by(rainbow_id
=rid
).all()
72 def message_to_rainbow_query(self
, mid
):
74 Query base of message id
76 Session
= sessionmaker(bind
=self
.engine
)
78 res
= session
.query(Message
).filter_by(message_id
=mid
).all()
82 def theme_store(self
, theme_name
):
86 Session
= sessionmaker(bind
=self
.engine
)
88 th
= Theme(theme_name
)
93 def theme_update(self
, theme_name
):
97 Session
= sessionmaker(bind
=self
.engine
)
99 res
= session
.query(Theme
).all()
101 r
.theme_name
= theme_name
105 def theme_query(self
):
109 Session
= sessionmaker(bind
=self
.engine
)
111 res
= session
.query(Theme
).all()
115 def semaphore_store(self
, flag
):
119 Session
= sessionmaker(bind
=self
.engine
)
126 def semaphore_update(self
, flag
):
128 Update semaphore flag
130 Session
= sessionmaker(bind
=self
.engine
)
132 res
= session
.query(Semaphore
).all()
138 def semaphore_query(self
):
142 Session
= sessionmaker(bind
=self
.engine
)
144 res
= session
.query(Semaphore
).all()