123

marketplaces/factory.py

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