X-Git-Url: https://vcs.fsf.org/?p=rainbowstream.git;a=blobdiff_plain;f=rainbowstream%2Fdb.py;h=e0c7e84b2d2ffe7f160ef8f45c54795e1cace8d8;hp=0756b3a5edc9469983ab872e29bd3a2e815b9b1a;hb=908d5012ca72d4a47d88e10cea211e94af03e1d6;hpb=c8f6a1733d068e53594a1a84a784fda7150688ed

diff --git a/rainbowstream/db.py b/rainbowstream/db.py
index 0756b3a..e0c7e84 100644
--- a/rainbowstream/db.py
+++ b/rainbowstream/db.py
@@ -14,6 +14,9 @@ class RainbowDB():
         self.engine = create_engine('sqlite:///rainbow.db', echo=False)
 
     def store(self, tweet_id):
+        """
+        Store tweet id
+        """
         Session = sessionmaker(bind=self.engine)
         session = Session()
         m = Map(tweet_id)
@@ -21,13 +24,28 @@ class RainbowDB():
         session.commit()
 
     def rainbow_query(self, rid):
+        """
+        Query base of rainbow id
+        """
         Session = sessionmaker(bind=self.engine)
         session = Session()
         res = session.query(Map).filter("rainbow_id =:rid").params(rid=rid).all()
         return res
 
     def tweet_query(self, tid):
+        """
+        Query base of tweet id
+        """
         Session = sessionmaker(bind=self.engine)
         session = Session()
         res = session.query(Map).filter("tweet_id =:tid").params(tid=tid).all()
         return res
+
+    def truncate(self):
+        """
+        Truncate table
+        """
+        Session = sessionmaker(bind=self.engine)
+        session = Session()
+        session.query(Map).delete()
+        session.commit()