aboutsummaryrefslogtreecommitdiff
path: root/mongodb.py
diff options
context:
space:
mode:
Diffstat (limited to 'mongodb.py')
-rw-r--r--mongodb.py17
1 files changed, 14 insertions, 3 deletions
diff --git a/mongodb.py b/mongodb.py
index 3c03c8ed..bc318add 100644
--- a/mongodb.py
+++ b/mongodb.py
@@ -120,15 +120,26 @@ class Articles(object):
for feed_id in self.db.collection_names():
unread_articles += self.nb_unread_articles(feed_id)
return unread_articles
-
-
+
def like_article(self, like, feed_id, article_id):
"""
Like or unlike an article.
"""
collection = self.db[str(feed_id)]
collection.update({"article_id": article_id}, {"$set": {"article_like": like}})
-
+
+ def mark_as_read(self, readed, feed_id=None, article_id=None):
+ """
+ """
+ if feed_id != None and article_id != None:
+ collection = self.db[str(feed_id)]
+ collection.update({"article_id": article_id}, {"$set": {"article_readed": readed}})
+ elif feed_id != None and article_id == None:
+ collection = self.db[str(feed_id)]
+ collection.update({"type": 1}, {"$set": {"article_readed": readed}}, multi=True)
+ else:
+ collection.update({"article_id": article_id}, {"$set": {"article_readed": readed}})
+
def list_collections(self):
"""
List all collections (feed).
bgstack15