camisatoshi-wordpress-reports/camisatoshi_wordpress_reports/order.py

from typing import Dict


class Order:

    def __init__(self, raw_data: Dict):
        self.raw_data = raw_data

    def __getitem__(self, item):
        return self.raw_data[item]

    @property
    def meta_data_entries(self):
        return {meta_data_entry["key"]: meta_data_entry["value"] for meta_data_entry in self.raw_data["meta_data"]}

    def contains_sku(self, sku: str) -> bool:
        for item in self["line_items"]:
            if item["sku"] == sku:
                return True
        return False

    def contains_meta_data_entry(self, meta_data_entry_key: str) -> bool:
        if meta_data_entry_key in self.meta_data_entries.keys():
            return True
        return False

    def is_settled_with_um(self):
        is_settled = self.meta_data_entries.get("is_settled_with_um", None)

        return bool(is_settled)

    @classmethod
    def from_api_response(cls, raw_data) -> "Order":
        return Order(raw_data)