rewritten crap
This commit is contained in:
26
sender/factory.py
Normal file
26
sender/factory.py
Normal file
@@ -0,0 +1,26 @@
|
||||
from typing import TYPE_CHECKING, Union
|
||||
|
||||
from database.sipro.enums.general import BaseMarketplace
|
||||
from sender.ozon import OzonStocksSender
|
||||
from sender.wildberries import WildberriesStocksSender
|
||||
from sender.yandexmarket import YandexmarketStocksSender
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from updaters.base import BaseMarketplaceUpdater
|
||||
|
||||
|
||||
class SenderFactory:
|
||||
@staticmethod
|
||||
def get_sender(updater: "BaseMarketplaceUpdater") -> Union[
|
||||
WildberriesStocksSender,
|
||||
OzonStocksSender,
|
||||
YandexmarketStocksSender
|
||||
]:
|
||||
match updater.marketplace.base_marketplace:
|
||||
case BaseMarketplace.WILDBERRIES:
|
||||
return WildberriesStocksSender(updater)
|
||||
case BaseMarketplace.OZON:
|
||||
return OzonStocksSender(updater)
|
||||
case BaseMarketplace.YANDEX_MARKET:
|
||||
return YandexmarketStocksSender(updater)
|
||||
raise ValueError()
|
||||
Reference in New Issue
Block a user