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


class OzonUpdater(BaseMarketplaceUpdater):
    def get_update_for_marketplace(self, data: StockData) -> dict:
        return {
            'offer_id': str(data['article']),
            'stock': data['full_stock'],
            'warehouse_id': self.marketplace.warehouse_id
        }