Move weird functions to Orders class

This commit is contained in:
Pablo Martin 2023-08-03 16:50:40 +02:00
parent 2099ddc0e0
commit 55b06dbbb0
2 changed files with 54 additions and 52 deletions

View file

@ -1,4 +1,4 @@
from typing import Dict, Collection
from typing import List, Dict, Collection
from camisatoshi_wordpress_reports.constants import order_keys, custom_meta_data_keys
@ -41,6 +41,37 @@ class Order:
class Orders:
def __init__(self, orders: Collection[Order]):
self._orders = orders
self._orders = orders
def __len__(self):
return len(self._orders)
def filter_orders_by_skus(self, skus: List[str]) -> "Orders":
filtered_orders = []
for order in self._orders:
for sku in skus:
if order.contains_sku(sku):
filtered_orders.append(order)
return Orders(filtered_orders)
def filter_orders_without_sats_received(self) -> "Orders":
orders_without_sats_received = []
for order in self._orders:
if not order.contains_meta_data_entry(custom_meta_data_keys.sats_received):
orders_without_sats_received.append(order)
continue
return Orders(orders_without_sats_received)
def filter_unsettled_orders(self) -> "Orders":
unsettled_orders = []
for order in self._orders:
if not order.is_settled_with_um():
unsettled_orders.append(order)
return Orders(unsettled_orders)