from pydantic import BaseModel

import fargate_redeploy.config as Config


class LambdaPayload(BaseModel):
    """Represents payload lambda expects."""

    service: str = Config.DEFAULT_ECS_SERVICE_NAME
    cluster: str = Config.DEFAULT_ECS_SERVICE_NAME
    wait_stable: bool = Config.WAIT_STABLE