From 74bcce477ce4597038242c82fea6badba85cdad0 Mon Sep 17 00:00:00 2001 From: agatha Date: Sat, 14 Mar 2026 16:26:33 -0400 Subject: [PATCH] fix: add field_validator to coerce INET types --- src/proxy_pool/proxy/schemas.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) 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