from pydantic import BaseModel

class GenerateRequest(BaseModel):
    """
    Base class for all generate requests.
    
    Attributes:
        order_ids: List of TCGPlayer order numbers
    """
    order_ids: list[str]

class GenerateAddressLabelsRequest(GenerateRequest):
    """
    Request for generating address labels.
    
    Attributes:
        label_type: Type of label to generate (dk1201 or dk1241)
    """
    label_type: str

class GeneratePackingSlipsRequest(GenerateRequest):
    pass

class GeneratePullSheetsRequest(GenerateRequest):
    pass

class GenerateResponse(BaseModel):
    message: str
    success: bool

class GenerateReturnLabelsRequest(BaseModel):
    number_of_labels: int