Added a couple of convenience methods to the classes to get the representations I needed, but put that in the classes

src/datatypes.py

@@ -103,6 +103,12 @@ class Category:
     def product_count(self):
         return len(self.products)
 
+    @property
+    def singular_name(self):
+        if self.name[-1] == "s":
+            return self.name[:-1]  # Clip the s
+        return self.name
+
 
 class Feature:
     all = {}
@@ -120,6 +126,7 @@ class Feature:
     def __repr__(self):
         return self.name
 
+
 class Product:
     all = {}
 
@@ -150,6 +157,11 @@ class Product:
     def for_ids(ids: List[str]):
         return[Product.all[i] for i in ids]
 
+    @staticmethod
+    def all_as_list():
+        return list(Product.all.values())
+
+
 class Review:
     all = {}
 
@@ -158,8 +170,3 @@ class Review:
         self.rating = rating
         self.review_text = review_text
         self.product = product
-
-
-if __name__ == "__main__":
-    DataLoader.load_data()
-    print('test')