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


class YandexmarketUpdater(BaseMarketplaceUpdater):
    def get_update_for_marketplace(self, stock_data: StockData) -> dict:
        return {
            'sku': str(stock_data['article']),
            'warehouseId': self.marketplace.warehouse_id,
            'items': [{
                'count': stock_data['full_stock'],
            }]
        }