Week 1 project

w1/data_processor.py

@@ -79,5 +79,13 @@ def aggregate(self, column_name: str) -> float:
         aggregate should be 105.58
         """
         ######################################## YOUR CODE HERE ##################################################
+        gen_data_reader = (row for row in self.data_reader)
+        _= next(gen_data_reader)
+
+        aggregate = 0
+
+        for row in tqdm(gen_data_reader):
+            aggregate += self.to_float(row[column_name])
+        return aggregate
 
         ######################################## YOUR CODE HERE ##################################################

w1/main.py

@@ -44,6 +44,22 @@ def revenue_per_region(dp: DataProcessor) -> Dict:
     }
     """
     ######################################## YOUR CODE HERE ##################################################
+    gen_data_reader = (row for row in dp.data_reader)
+
+    _= next(gen_data_reader)
+
+    revenue_per_region = {}
+
+    for row in tqdm(gen_data_reader):
+        country = row['Country']
+        total_price = dp.to_float((row['TotalPrice']))
+
+        if country in revenue_per_region:
+            revenue_per_region[country] += total_price
+        else:
+            revenue_per_region[country] = total_price
+
+    return revenue_per_region
 
     ######################################## YOUR CODE HERE ##################################################
 

w1/utils.py

@@ -112,6 +112,11 @@ def __iter__(self) -> Generator:
         }
         """
     ######################################## YOUR CODE HERE ##################################################
+        with open(self._fp, 'r') as f:
+            rows = f.readlines()
+            for row in rows:
+                row = row.strip().split(self._sep)
+                yield {self._col_names[i]: row[i] for i in range((len(self._col_names)))}
 
     ######################################## YOUR CODE HERE ##################################################