46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
from pydantic import BaseModel, Field, field_validator
|
|
from typing import Optional
|
|
|
|
|
|
class ServiceBaseSchema(BaseModel):
|
|
"""Esquema base para servicios con campos comunes"""
|
|
estado: int = Field(..., description="ID único del servicio")
|
|
tipo_procesamiento: int = Field(..., description="ID de la organización")
|
|
pedimento: str = Field(..., description="ID del estado del servicio")
|
|
servicio: int = Field(..., description="ID del tipo de servicio")
|
|
organizacion: str = Field(..., description="ID de la organización")
|
|
|
|
|
|
|
|
@field_validator('pedimento', 'organizacion')
|
|
def validate_string_fields(cls, v):
|
|
if not v or not v.strip():
|
|
raise ValueError('Los campos de texto no pueden estar vacíos')
|
|
return v.strip()
|
|
|
|
@field_validator('estado', 'tipo_procesamiento', 'servicio')
|
|
def validate_numeric_ids(cls, v):
|
|
if v is None or v < 0:
|
|
raise ValueError('Los IDs numéricos deben ser números positivos')
|
|
return v
|
|
|
|
class ServiceUpdateRequest(ServiceBaseSchema):
|
|
"""Esquema para actualizar un servicio"""
|
|
id: int = Field(..., description="ID del servicio a actualizar")
|
|
|
|
@field_validator('id')
|
|
def validate_id(cls, v):
|
|
if v is None or v < 0:
|
|
raise ValueError('El ID debe ser un número positivo')
|
|
return v
|
|
|
|
class ServiceRemesaSchema(BaseModel):
|
|
"""Esquema para remesas de servicios"""
|
|
organizacion: str = Field(..., description="ID de la organización")
|
|
pedimento: str = Field(..., description="ID del pedimento")
|
|
|
|
@field_validator('organizacion', 'pedimento')
|
|
def validate_string_fields(cls, v):
|
|
if not v or not v.strip():
|
|
raise ValueError('Los campos de texto no pueden estar vacíos')
|
|
return v.strip() |