import json
import logging
from typing import Union

import jwt

from database import Marketplace
from marketplaces.base import BaseMarketplaceApi


class WildberriesMarketplaceApi(BaseMarketplaceApi):

    def __init__(self, marketplace: Marketplace):
        self.marketplace = marketplace
        auth_data = json.loads(marketplace.auth_data)
        token = auth_data.get('token')
        self.is_valid = True
        try:
            decoded_token = jwt.decode(token, algorithms=["HS256"], options={"verify_signature": False})
        except Exception:
            logging.error(f"Couldn't decode token for {marketplace.id}")
            self.is_valid = False
            return
        self.limiter_key = str(marketplace.company_id) + str(decoded_token.get('sid'))

        self.headers = {
            'Authorization': token,
            'Content-Type': 'application/json',
            'User-Agent': 'wbas_seller.denco.store3547'
        }

    def get_headers(self):
        return self.headers

    @property
    def api_url(self):
        return 'https://marketplace-api.wildberries.ru'

    async def update_stocks(self, data: Union[list, dict]):
        warehouse_id = self.marketplace.warehouse_id
        return await self._method('PUT', f'/api/v3/stocks/{warehouse_id}', data={'stocks': data})