rewritten crap

marketplaces/ozon.py

@@ -2,7 +2,7 @@ import json
 import logging
 from typing import Union
 
-from aiohttp import ClientResponse
+from aiohttp import ClientResponse, ClientSession
 
 from database import Marketplace
 from marketplaces.base import BaseMarketplaceApi
@@ -34,5 +34,5 @@ class OzonMarketplaceApi(BaseMarketplaceApi):
     def api_url(self):
         return 'https://api-seller.ozon.ru'
 
-    async def update_stocks(self, data: Union[list, dict]) -> ClientResponse:
+    async def update_stocks(self, data: Union[list, dict]) -> (ClientSession, ClientResponse):
         return await self._method('POST', '/v2/products/stocks', data={'stocks': data})