Added Base class implementation for reading items from csv file

qtsathish · qtsathish · commit d3074f35bf36 · 2022-01-10T08:38:58.000+05:30
diff --git a/Dec21/InventoryManagement/data/products.csv b/Dec21/InventoryManagement/data/products.csv
diff --git a/Dec21/InventoryManagement/data/stocks.csv b/Dec21/InventoryManagement/data/stocks.csv
diff --git a/Dec21/InventoryManagement/main.py b/Dec21/InventoryManagement/main.py
@@ -11,12 +11,14 @@ def get_input_from_user(field_name):
     return result
 
 if __name__ == '__main__':
-    s1 = Stock(2,11)
-    s1.save()
+    #s1 = Stock(2,11)
+    #s1.save()
 
-    p1 = Product(1, 'test', 'test', 'test', 100.45)
-    p1.save()
-    #print(s1)
+    #p1 = Product(1, 'test', 'test', 'test', 100.45)
+    #p1.save()
+    items = Product.items()
+    for item in items:
+        print(item)
 
     # id = get_input_from_user('id')
     # name = get_input_from_user('name')
diff --git a/Dec21/InventoryManagement/models/baseinventory.py b/Dec21/InventoryManagement/models/baseinventory.py
@@ -2,6 +2,8 @@
 import os
 import csv
 
+import inventory
+
 
 class BaseInventoryModel:
     """
@@ -36,5 +38,23 @@ def save(self):
                 writer.writeheader()
             writer.writerow(field_dict)
 
+    @classmethod
+    def items(cls):
+        """
+        Returns all the objects by reading the records in the csv file
+        """
+        if not os.path.exists(cls._file_name):
+            return []
+        with open(cls._file_name, 'r') as csv_file:
+            reader = csv.DictReader(csv_file)
+            #inventory_items = []
+            #for row in reader:
+            #    inventory_item = cls(**row)
+            #    inventory_items.append(inventory_item)
+            inventory_items = [cls(**row) for row in reader]
+        return inventory_items
+
+
+
 
 
diff --git a/Dec21/InventoryManagement/models/products.py b/Dec21/InventoryManagement/models/products.py
@@ -29,18 +29,7 @@ def __str__(self):
         """
         This is string representation of object
         T"""
-        return(f"{self.id}, {self.name}, {self.description}, {self.category}, {self.mrp}, {self.created_at}")
+        return(f"{self.id}, {self.name}, {self.description}, {self.category}, {self.mrp}, {self.created_at}, {self.updated_at}")
     
 
-    @classmethod
-    def ids(cls):
-        """
-        This method will read all the existing ids from the csv file 
-        """
-        item_ids = []
-        with open(cls.file_name, 'r') as file:
-            reader = csv.reader(file, delimiter = ',')
-            for row in reader:
-                if len(row) > 0:
-                    item_ids.append(row[0])
-        return item_ids
+    
diff --git a/Dec21/InventoryManagement/models/stock.py b/Dec21/InventoryManagement/models/stock.py
@@ -7,9 +7,10 @@ class Stock(BaseInventoryModel):
     """
     This class represents the Stock of the items in the Store
     """
-    def __init__(self, id, quantity, created_at=None, updated_at=None) -> None:
+    def __init__(self, id, product_id, quantity, created_at=None, updated_at=None) -> None:
         super().__init__(created_at, updated_at)
         self.id = id
+        self.product_id = product_id
         self.quantity = quantity
 
     
