From a29f1235a48b9dbe235729fd94f5a6b899168d3d Mon Sep 17 00:00:00 2001 From: Samuel Clay Date: Tue, 4 Aug 2020 08:56:07 -0400 Subject: [PATCH] Remove highlights count when it reaches zero. --- apps/rss_feeds/models.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/apps/rss_feeds/models.py b/apps/rss_feeds/models.py index 2a75bf3ab..f11a576b7 100644 --- a/apps/rss_feeds/models.py +++ b/apps/rss_feeds/models.py @@ -3094,8 +3094,13 @@ class MStarredStoryCounts(mongo.Document): highlighted_count = MStarredStory.objects(user_id=user_id, highlights__exists=True, __raw__={"$where": "this.highlights.length > 0"}).count() - cls.objects(user_id=user_id, - is_highlights=True, slug="highlights").update_one(set__count=highlighted_count, upsert=True) + if highlighted_count > 0: + cls.objects(user_id=user_id, + is_highlights=True, + slug="highlights" + ).update_one(set__count=highlighted_count, upsert=True) + else: + cls.objects(user_id=user_id, is_highlights=True, slug="highlights").delete() return highlighted_count