from queries.general import StockData
from sender.base import StockRequest
from updaters.base import BaseMarketplaceUpdater


class OzonUpdater(BaseMarketplaceUpdater):

    def _get_identifier(self) -> str:
        auth_data = self.get_auth_data()
        client_id = auth_data.get('clientId', -1)
        return str(client_id)

    def get_stock_request(self, stock_data: StockData) -> StockRequest:
        return {
            'request_data': {
                'offer_id': str(stock_data['article']),
                'stock': stock_data['full_stock'],
                'warehouse_id': self.marketplace.warehouse_id
            },
            'product_id': stock_data['product_id'],
            'full_stock': stock_data['full_stock']
        }