from dataclasses import dataclass
from typing import Union, List

from pydantic import BaseModel


@dataclass
class StockUpdate:
    product_id: int


class BaseSchema(BaseModel):
    pass


class UpdateRequest(BaseSchema):
    product_ids: List[int]


class UpdateMarketplaceRequest(BaseSchema):
    marketplace_id: int


class UpdateMarketplaceProductsRequest(UpdateMarketplaceRequest):
    product_ids: List[int]


class UpdateMarketplacesRequest(BaseSchema):
    marketplace_ids: Union[List[int], None] = None


class UpdateResponse(BaseSchema):
    task_id: str