diff --git a/src/proxy_pool/proxy/schemas.py b/src/proxy_pool/proxy/schemas.py index 81de9b3..fd59607 100644 --- a/src/proxy_pool/proxy/schemas.py +++ b/src/proxy_pool/proxy/schemas.py @@ -1,7 +1,7 @@ from datetime import datetime from uuid import UUID -from pydantic import BaseModel, Field +from pydantic import BaseModel, Field, field_validator from proxy_pool.proxy.models import AnonymityLevel, ProxyProtocol, ProxyStatus @@ -52,6 +52,13 @@ class ProxyResponse(BaseModel): first_seen_at: datetime created_at: datetime + @field_validator("ip", "exit_ip", mode="before") + @classmethod + def coerce_inet(cls, value: object) -> str | None: + if value is None: + return None + return str(value) + class ProxyListParams(BaseModel): status: ProxyStatus | None = None