camisatoshi-wordpress-reports/camisatoshi_wordpress_reports/controllers.py

from pathlib import Path
import datetime
from typing import List, Dict
import logging

from dotenv import dotenv_values
from woocommerce import API

from camisatoshi_wordpress_reports.order import Order, Orders
from camisatoshi_wordpress_reports.constants import custom_meta_data_keys

API_CONFIG = dotenv_values(
    dotenv_path=Path.home() / Path(".camisatoshi-wordpress-reports/.env")
)
WC_API = API(
    url=API_CONFIG["URL"],
    consumer_key=API_CONFIG["CONSUMER_KEY"],
    consumer_secret=API_CONFIG["CONSUMER_SECRET"],
    version=API_CONFIG["VERSION"],
)

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger()


def check_health():
    logger.info(f"Connecting to the configured WooCommerce at {API_CONFIG['URL']}")

    try:
        api_reported_version = WC_API.get("").json()["namespace"]
    except:
        raise ConnectionError("There was an issue connecting to the WooCommerce API.")

    logger.info(f"Informed version of the API: {API_CONFIG['VERSION']}")
    logger.info(f"Version reported by the API itself: {api_reported_version}")

    logger.info("Connection successful. The API is reachable.")


def generate_um_report(
    start_date: datetime.datetime, end_date: datetime.datetime
) -> None:
    logger.info(f"Fetching orders between {start_date} and {end_date}.")

    orders_in_date_range = WC_API.get(
        endpoint="orders",
        params={
            "after": start_date.isoformat(),
            "before": end_date.isoformat(),
            "per_page": 100,
            "status": "processing,completed",
        },
    ).json()
    orders_in_date_range = Orders(
        [
            Order.from_api_response(order_raw_data)
            for order_raw_data in orders_in_date_range
        ]
    )
    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 = orders_in_date_range.filter_orders_by_skus(skus=skus_to_keep)
    logger.info(f"Kept {len(relevant_orders)} orders.")

    logger.info("Checking if all orders have the sats_received entry filled in.")
    orders_without_sats_received = relevant_orders.filter_orders_without_sats_received()
    if orders_without_sats_received:
        logger.warning(
            f"There are {len(orders_without_sats_received)} orders without a properly filled sats_received entry."
        )
        logger.warning(f"See details below.")
        logger.warning(orders_without_sats_received)
        raise ValueError(
            "Not all orders have sats_received. Can't compute sats owed without that."
        )
    logger.info("Success, all orders have sats_received filled in.")

    logger.info("Removing settled orders.")
    unsettled_orders = relevant_orders.filter_unsettled_orders()
    logger.info(f"Kept {len(unsettled_orders)} unsettled 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