123

marketplaces/factory.py

@@ -0,0 +1,19 @@
+from typing import Union
+
+from database import Marketplace
+from database.sipro.enums.general import BaseMarketplace
+from .wildberries import WildberriesMarketplace
+from .ozon import OzonMarketplace
+
+
+class MarketplaceFactory:
+    @staticmethod
+    def get_marketplace(marketplace: Marketplace) -> Union[
+        WildberriesMarketplace,
+        OzonMarketplace,
+    ]:
+        match marketplace.base_marketplace:
+            case BaseMarketplace.OZON:
+                return OzonMarketplace(marketplace)
+            case BaseMarketplace.WILDBERRIES:
+                return WildberriesMarketplace(marketplace)