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


class WildberriesUpdater(BaseMarketplaceUpdater):

    def get_update_for_marketplace(self, stock_data: StockData) -> dict:
        return {
            'sku': stock_data['marketplace_product'].third_additional_article,
            'amount': stock_data['full_stock']
        }