2 from sqlalchemy
import create_engine
3 from sqlalchemy
import update
4 from sqlalchemy
.orm
import sessionmaker
5 from .table_def
import *
13 if not os
.path
.isfile('rainbow.db'):
15 self
.engine
= create_engine('sqlite:///rainbow.db', echo
=False)
17 def tweet_store(self
, tweet_id
):
21 Session
= sessionmaker(bind
=self
.engine
)
27 def rainbow_to_tweet_query(self
, rid
):
29 Query base of rainbow id
31 Session
= sessionmaker(bind
=self
.engine
)
33 res
= session
.query(Tweet
).filter_by(rainbow_id
=rid
).all()
36 def tweet_to_rainbow_query(self
, tid
):
38 Query base of tweet id
40 Session
= sessionmaker(bind
=self
.engine
)
42 res
= session
.query(Tweet
).filter_by(tweet_id
=tid
).all()
45 def message_store(self
, message_id
):
49 Session
= sessionmaker(bind
=self
.engine
)
51 m
= Message(message_id
)
55 def rainbow_to_message_query(self
, rid
):
57 Query base of rainbow id
59 Session
= sessionmaker(bind
=self
.engine
)
61 res
= session
.query(Message
).filter_by(rainbow_id
=rid
).all()
64 def message_to_rainbow_query(self
, mid
):
66 Query base of message id
68 Session
= sessionmaker(bind
=self
.engine
)
70 res
= session
.query(Message
).filter_by(message_id
=mid
).all()
73 def theme_store(self
, theme_name
):
77 Session
= sessionmaker(bind
=self
.engine
)
79 th
= Theme(theme_name
)
83 def theme_update(self
, theme_name
):
87 Session
= sessionmaker(bind
=self
.engine
)
89 res
= session
.query(Theme
).all()
91 r
.theme_name
= theme_name
94 def theme_query(self
):
98 Session
= sessionmaker(bind
=self
.engine
)
100 res
= session
.query(Theme
).all()