57428c42bc81f8eb7062e3c31ec623b8c4d5fd57
2 from sqlalchemy
import create_engine
3 from sqlalchemy
.orm
import sessionmaker
4 from table_def
import Map
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 store(self
, tweet_id
):
21 Session
= sessionmaker(bind
=self
.engine
)
27 def rainbow_query(self
, rid
):
29 Query base of rainbow id
31 Session
= sessionmaker(bind
=self
.engine
)
33 res
= session
.query(Map
)
34 .filter("rainbow_id =:rid").params(rid
=rid
).all()
37 def tweet_query(self
, tid
):
39 Query base of tweet id
41 Session
= sessionmaker(bind
=self
.engine
)
43 res
= session
.query(Map
)
44 .filter("tweet_id =:tid").params(tid
=tid
).all()
51 Session
= sessionmaker(bind
=self
.engine
)
53 session
.query(Map
).delete()