From ad4bc8c0dab1f82c102b06fea80df439b307ee69 Mon Sep 17 00:00:00 2001
From: counterweight <counterweightoperator@protonmail.com>
Date: Thu, 3 Aug 2023 09:42:22 +0200
Subject: [PATCH] Some advances.

---
 camisatoshi_wordpress_reports/controllers.py | 67 +++++++++++++++++---
 1 file changed, 57 insertions(+), 10 deletions(-)

diff --git a/camisatoshi_wordpress_reports/controllers.py b/camisatoshi_wordpress_reports/controllers.py
index 0b6a273..25c3d9d 100644
--- a/camisatoshi_wordpress_reports/controllers.py
+++ b/camisatoshi_wordpress_reports/controllers.py
@@ -1,6 +1,7 @@
 from pathlib import Path
 import datetime
 from typing import List, Dict
+import logging
 
 from dotenv import dotenv_values
 from woocommerce import API
@@ -15,9 +16,14 @@ WC_API = API(
     version=API_CONFIG["VERSION"],
 )
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger()
+
 
 def check_health():
-    print(f"Connecting to the configured woocomerce at {API_CONFIG['URL']}")
+    logger.info(
+        f"Connecting to the configured woocomerce at {API_CONFIG['URL']}"
+    )
 
     try:
         api_reported_version = WC_API.get("").json()["namespace"]
@@ -26,24 +32,65 @@ def check_health():
             "There was an issue connecting to the woocomerce API."
         )
 
-    print(f"Informed version of the API: {API_CONFIG['VERSION']}")
-    print(f"Version reported by the API itself: {api_reported_version}")
+    logger.info(f"Informed version of the API: {API_CONFIG['VERSION']}")
+    logger.info(f"Version reported by the API itself: {api_reported_version}")
 
-    print("Connection successful. The API is reachable.")
+    logger.info("Connection successful. The API is reachable.")
 
 
 def generate_um_report(start_date: str, end_date: str):
 
+    start_date = datetime.datetime.strptime(start_date, "%Y-%m-%d").isoformat()
+    end_date = datetime.datetime.strptime(end_date, "%Y-%m-%d").isoformat()
+
+    logger.info(f"Fetching orders between {start_date} and {end_date}.")
+
     orders_in_date_range = WC_API.get(
         endpoint="orders",
         params={
-            "after": datetime.datetime.strptime(
-                start_date, "%Y-%m-%d"
-            ).isoformat(),
-            "before": datetime.datetime.strptime(
-                end_date, "%Y-%m-%d"
-            ).isoformat(),
+            "after": start_date,
+            "before": end_date,
             "per_page": 100,
             "status": "processing,completed",
         },
+    ).json()
+
+    logger.info(f"Received {len(orders_in_date_range)} orders.")
+
+    skus_to_keep = ["TEE-05-BBO-BLACK"]
+
+    logger.info(f"Filtering by SKUs: {skus_to_keep}")
+    relevant_orders = filter_orders_by_sku(
+        orders_in_date_range, skus=skus_to_keep
     )
+
+    logger.info(f"Kept {len(relevant_orders)} orders.")
+
+    # Fetch orders:
+    #   - Between specific dates
+    #   - That contain the hardcoded products
+    #   - That have been paid, hence status is either processing or completed
+    #   - That have not been settled yet (is_settled_with_um: 0)
+
+    # Print to screen:
+    #   - Orders that do not have the `sats_received` metadata informed
+    #   - The unit count for each product
+    #   - The sales sum for each product
+    #   - The sats sum for each product
+    #   - The corresponding payment owed to UM
+    #   - The list of order ids that have been taken into account
+
+    # Update orders:
+    #   - Add metadata entry: is_settled_with_um: 1
+
+
+def filter_orders_by_sku(orders: List[Dict], skus: List[str]):
+
+    filtered_orders = []
+
+    for order in orders:
+        for item in order["line_items"]:
+            if item["sku"] in skus:
+                filtered_orders.append(order)
+
+    return filtered_orders