Report kind of complete

camisatoshi_wordpress_reports/order.py

@@ -1,6 +1,7 @@
-from typing import List, Dict, Collection
+from typing import Dict, Collection
 
 from camisatoshi_wordpress_reports.constants import order_keys, custom_meta_data_keys
+from camisatoshi_wordpress_reports.utils import safe_zero_division
 
 
 class Order:
@@ -17,6 +18,34 @@ class Order:
             for meta_data_entry in self.raw_data[order_keys.meta_data]
         }
 
+    def units_of_sku(self, sku: str) -> int:
+        units = 0
+        for line in self[order_keys.line_items]:
+            if line[order_keys.line_item_keys.sku] == sku:
+                units += line[order_keys.line_item_keys.quantity]
+        return units
+
+    def sales_of_sku(self, sku: str) -> float:
+        sales = 0
+        for line in self[order_keys.line_items]:
+            if line[order_keys.line_item_keys.sku] == sku:
+                sales += float(line[order_keys.line_item_keys.total])
+        return sales
+
+    def sats_received_for_sku(self, sku: str) -> float:
+        total_order_eur = float(self[order_keys.total])
+        eur_of_sku = self.sales_of_sku(sku)
+
+        monetary_weight_of_sku_in_order = safe_zero_division(eur_of_sku, total_order_eur)
+        total_order_sats_received = float(
+            self.meta_data_entries[custom_meta_data_keys.sats_received]
+        )
+        sats_received_for_sku = (
+            monetary_weight_of_sku_in_order * total_order_sats_received
+        )
+
+        return sats_received_for_sku
+
     def contains_sku(self, sku: str) -> bool:
         for item in self[order_keys.line_items]:
             if item[order_keys.line_item_keys.sku] == sku:
@@ -58,14 +87,12 @@ class Orders:
             return next_order
         raise StopIteration
 
-
-    def filter_orders_by_skus(self, skus: List[str]) -> "Orders":
+    def filter_orders_by_sku(self, sku: str) -> "Orders":
         filtered_orders = []
 
         for order in self:
-            for sku in skus:
-                if order.contains_sku(sku):
-                    filtered_orders.append(order)
+            if order.contains_sku(sku):
+                filtered_orders.append(order)
 
         return Orders(filtered_orders)