projects
/
rainbowstream.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
refactoring and add various comments
[rainbowstream.git]
/
rainbowstream
/
db.py
diff --git
a/rainbowstream/db.py
b/rainbowstream/db.py
index 2ab23bce0e80a371f05792886c32644a6e82e991..1714005d77a4154d9052144b595e5b978a855640 100644
(file)
--- a/
rainbowstream/db.py
+++ b/
rainbowstream/db.py
@@
-9,10
+9,14
@@
class RainbowDB():
engine = None
def __init__(self):
engine = None
def __init__(self):
+ """
+ Init DB
+ """
if not os.path.isfile('rainbow.db'):
init_db()
self.engine = create_engine('sqlite:///rainbow.db', echo=False)
if not os.path.isfile('rainbow.db'):
init_db()
self.engine = create_engine('sqlite:///rainbow.db', echo=False)
+
def tweet_store(self, tweet_id):
"""
Store tweet id
def tweet_store(self, tweet_id):
"""
Store tweet id
@@
-23,6
+27,7
@@
class RainbowDB():
session.add(t)
session.commit()
session.add(t)
session.commit()
+
def rainbow_to_tweet_query(self, rid):
"""
Query base of rainbow id
def rainbow_to_tweet_query(self, rid):
"""
Query base of rainbow id
@@
-32,6
+37,7
@@
class RainbowDB():
res = session.query(Tweet).filter_by(rainbow_id=rid).all()
return res
res = session.query(Tweet).filter_by(rainbow_id=rid).all()
return res
+
def tweet_to_rainbow_query(self, tid):
"""
Query base of tweet id
def tweet_to_rainbow_query(self, tid):
"""
Query base of tweet id
@@
-41,6
+47,7
@@
class RainbowDB():
res = session.query(Tweet).filter_by(tweet_id=tid).all()
return res
res = session.query(Tweet).filter_by(tweet_id=tid).all()
return res
+
def message_store(self, message_id):
"""
Store message id
def message_store(self, message_id):
"""
Store message id
@@
-51,6
+58,7
@@
class RainbowDB():
session.add(m)
session.commit()
session.add(m)
session.commit()
+
def rainbow_to_message_query(self, rid):
"""
Query base of rainbow id
def rainbow_to_message_query(self, rid):
"""
Query base of rainbow id
@@
-60,6
+68,7
@@
class RainbowDB():
res = session.query(Message).filter_by(rainbow_id=rid).all()
return res
res = session.query(Message).filter_by(rainbow_id=rid).all()
return res
+
def message_to_rainbow_query(self, mid):
"""
Query base of message id
def message_to_rainbow_query(self, mid):
"""
Query base of message id
@@
-69,6
+78,7
@@
class RainbowDB():
res = session.query(Message).filter_by(message_id=mid).all()
return res
res = session.query(Message).filter_by(message_id=mid).all()
return res
+
def theme_store(self, theme_name):
"""
Store theme
def theme_store(self, theme_name):
"""
Store theme
@@
-79,6
+89,7
@@
class RainbowDB():
session.add(th)
session.commit()
session.add(th)
session.commit()
+
def theme_update(self, theme_name):
"""
Update theme
def theme_update(self, theme_name):
"""
Update theme
@@
-90,6
+101,7
@@
class RainbowDB():
r.theme_name = theme_name
session.commit()
r.theme_name = theme_name
session.commit()
+
def theme_query(self):
"""
Query theme
def theme_query(self):
"""
Query theme
@@
-99,22
+111,35
@@
class RainbowDB():
res = session.query(Theme).all()
return res
res = session.query(Theme).all()
return res
- def list_store(self, list_id, list_name):
+
+ def semaphore_store(self, flag):
"""
"""
- Store
list id and name
+ Store
semaphore flag
"""
Session = sessionmaker(bind=self.engine)
session = Session()
"""
Session = sessionmaker(bind=self.engine)
session = Session()
-
l = List(list_id, list_name
)
- session.add(
l
)
+
th = Semaphore(flag
)
+ session.add(
th
)
session.commit()
session.commit()
- def list_name_to_id_query(self, list_name):
+
+ def semaphore_update(self, flag):
"""
"""
- Query base of list id
+ Update semaphore flag
"""
Session = sessionmaker(bind=self.engine)
session = Session()
"""
Session = sessionmaker(bind=self.engine)
session = Session()
- res = session.query(List).filter_by(list_name=list_name).all()
- return res
+ res = session.query(Semaphore).all()
+ for r in res:
+ r.flag = flag
+ session.commit()
+
+ def semaphore_query(self):
+ """
+ Query semaphore
+ """
+ Session = sessionmaker(bind=self.engine)
+ session = Session()
+ res = session.query(Semaphore).all()
+ return res[0].flag