From dbb63f833282a44fcc55da16968cf62bb76e1d75 Mon Sep 17 00:00:00 2001 From: asdf Date: Wed, 21 Jul 2021 20:22:45 +1000 Subject: [PATCH] Fix category counting error --- data.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/data.py b/data.py index 192e9f1..d192b99 100644 --- a/data.py +++ b/data.py @@ -179,7 +179,7 @@ class LinkData: for record in self.link_data: name = record[4] timestamp = record[2] - if name != "": + if name != "": # only replies have column 4 empty if name not in [cat_record["name"] for cat_record in self.categories]: self.categories.append( {"name": name, "count": 1, "last_updated": timestamp} @@ -197,9 +197,12 @@ class LinkData: except KeyError: continue parent_cat_name = parent_record[4] - if parent_cat_name not in [cat_record["name"] for cat_record in self.categories]: + if parent_cat_name not in [ + cat_record["name"] for cat_record in self.categories + ]: self.categories.append( - {"name": parent_cat_name, "count": 1, "last_updated": timestamp} + # append a record, but set the count to 0 because the parent record will be counted at some stage + {"name": parent_cat_name, "count": 0, "last_updated": timestamp} ) else: for cat_record in self.categories: