rewritten crap

updaters/wildberries_updater.py

@@ -1,11 +1,32 @@
+import json
+
+import jwt
+from celery.backends.database import retry
+
 from queries.general import StockData
+from sender.base import StockRequest
 from updaters.base import BaseMarketplaceUpdater
 
 
 class WildberriesUpdater(BaseMarketplaceUpdater):
 
-    def get_update_for_marketplace(self, stock_data: StockData) -> dict:
+    def _get_identifier(self) -> str:
+        try:
+            auth_data = self.get_auth_data()
+            token = auth_data.get('token')
+            decoded_token = jwt.decode(token, algorithms=["HS256"], options={"verify_signature": False})
+            sid = decoded_token.get('sid')
+            return str(sid)
+        except Exception:
+            return f'-1'
+
+    def get_stock_request(self, stock_data: StockData) -> StockRequest:
         return {
-            'sku': stock_data['marketplace_product'].third_additional_article,
-            'amount': stock_data['full_stock']
+            'request_data': {
+                'sku': stock_data['marketplace_product'].third_additional_article,
+                'amount': stock_data['full_stock']
+            },
+            'product_id': stock_data['product_id'],
+
+        'full_stock': stock_data['full_stock']
         }