Move weird functions to Orders class
This commit is contained in:
parent
2099ddc0e0
commit
55b06dbbb0
2 changed files with 54 additions and 52 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue