Some renaming

This commit is contained in:
counterweight 2023-09-06 14:26:25 +02:00
parent 9917bb71e8
commit 5840a85ded
Signed by: counterweight
GPG key ID: 883EDBAA726BD96C
2 changed files with 13 additions and 13 deletions

View file

@ -14,7 +14,7 @@ from camisatoshi_wordpress_reports.constants import (
bbo_royalty_fee,
)
from camisatoshi_wordpress_reports.report_building import (
ReportChainBuilder,
OrderObtentionChainBuilder,
WoocomerceOrderScope,
keep_orders_containing_sku,
)
@ -134,7 +134,7 @@ def generate_um_report(
def generate_sku_report(start_date, end_date, sku):
logger.info(f"Fetching orders between {start_date} and {end_date}.")
report_chain_builder = ReportChainBuilder()
report_chain_builder = OrderObtentionChainBuilder()
report_chain_builder.add_order_fetching_step(
wc_order_scope=WoocomerceOrderScope(

View file

@ -84,7 +84,7 @@ def validate_orders_satisfy_metadata_filter(
return True
class ReportChain:
class OrderObtentionChain:
def __init__(self):
self.order_fetching_step = None
self.order_filtering_steps = []
@ -127,37 +127,37 @@ class ReportChain:
raise ValueError("Error during order validation step.")
class ReportChainBuilder:
class OrderObtentionChainBuilder:
def __init__(self):
self._wip_report_chain = ReportChain()
self._wip_order_obtention_chain = OrderObtentionChain()
def add_order_fetching_step(
self, wc_order_scope: WoocomerceOrderScope
) -> "ReportChainBuilder":
) -> "OrderObtentionChainBuilder":
"""
Define a scope of orders to get from Woocomerce.
"""
self._order_scope = wc_order_scope
self._wip_report_chain.order_fetching_step = partial(
self._wip_order_obtention_chain.order_fetching_step = partial(
fetch_orders_from_wc, wc_order_scope=wc_order_scope
)
return self
def add_order_filtering_step(self, step) -> "ReportChainBuilder":
def add_order_filtering_step(self, step) -> "OrderObtentionChainBuilder":
self._wip_report_chain.order_filtering_steps.append(step)
self._wip_order_obtention_chain.order_filtering_steps.append(step)
return self
def add_order_validation_step(self, step) -> "ReportChainBuilder":
def add_order_validation_step(self, step) -> "OrderObtentionChainBuilder":
self._wip_report_chain.order_validation_steps.append(step)
self._wip_order_obtention_chain.order_validation_steps.append(step)
return self
def get_report_chain(self) -> ReportChain:
def get_report_chain(self) -> OrderObtentionChain:
return self._wip_report_chain
return self._wip_order_obtention_chain