rewritten crap

marketplaces/wildberries.py

@@ -3,6 +3,7 @@ import logging
 from typing import Union
 
 import jwt
+from aiohttp import ClientSession, ClientResponse
 
 from database import Marketplace
 from marketplaces.base import BaseMarketplaceApi
@@ -36,7 +37,7 @@ class WildberriesMarketplaceApi(BaseMarketplaceApi):
     def api_url(self):
         return 'https://marketplace-api.wildberries.ru'
 
-    async def update_stocks(self, data: Union[list, dict]):
+    async def update_stocks(self, data: Union[list, dict])-> (ClientSession, ClientResponse):
         warehouse_id = self.marketplace.warehouse_id
         return await self._method('PUT', f'/api/v3/stocks/{warehouse_id}', data={'stocks': data})