Constants to stop hardcoding strings everywhere

This commit is contained in:
Pablo Martin 2023-08-03 16:33:53 +02:00
parent 964682716f
commit e87854bba0
2 changed files with 22 additions and 5 deletions

View file

@ -1,8 +1,9 @@
from typing import Dict
from camisatoshi_wordpress_reports.constants import order_keys, custom_meta_data_keys
class Order:
def __init__(self, raw_data: Dict):
self.raw_data = raw_data
@ -11,11 +12,14 @@ class Order:
@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"]}
return {
meta_data_entry["key"]: meta_data_entry["value"]
for meta_data_entry in self.raw_data[order_keys.meta_data]
}
def contains_sku(self, sku: str) -> bool:
for item in self["line_items"]:
if item["sku"] == sku:
for item in self[order_keys.line_items]:
if item[order_keys.line_item_keys.sku] == sku:
return True
return False
@ -25,7 +29,9 @@ class Order:
return False
def is_settled_with_um(self):
is_settled = self.meta_data_entries.get("is_settled_with_um", None)
is_settled = self.meta_data_entries.get(
custom_meta_data_keys.is_settled_with_um, None
)
return bool(is_settled)