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


class YandexmarketUpdater(BaseMarketplaceUpdater):

    def _get_identifier(self) -> str:
        return str(self.marketplace.campaign_id)

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

        }