Skip to content

Commit

Permalink
legacy validator
Browse files Browse the repository at this point in the history
Signed-off-by: Keming <kemingy94@gmail.com>
  • Loading branch information
kemingy committed Dec 31, 2024
1 parent ea432d1 commit 01af788
Showing 1 changed file with 12 additions and 6 deletions.
18 changes: 12 additions & 6 deletions tests/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,8 @@
from enum import Enum, IntEnum
from typing import Any, Dict, List, Optional, Union, cast

from pydantic import BaseModel, Field, field_validator, model_validator
# legacy code of: from pydantic import model_validator, field_validator
from pydantic import BaseModel, Field, root_validator, validator

from spectree import BaseFile, ExternalDocs, SecurityScheme, SecuritySchemeData, Tag
from spectree._pydantic import generate_root_model
Expand Down Expand Up @@ -77,10 +78,14 @@ class Language(str, Enum):
class Headers(BaseModel):
lang: Language

@model_validator(mode="before")
@classmethod
def lower_keys(cls, data: Any):
return {key.lower(): value for key, value in data.items()}
# @model_validator(mode="before")
# @classmethod
# def lower_keys(cls, data: Any):
# return {key.lower(): value for key, value in data.items()}

@root_validator(pre=True)
def lower_keys(cls, values):
return {key.lower(): value for key, value in values.items()}


class Cookies(BaseModel):
Expand All @@ -101,7 +106,8 @@ class DemoQuery(BaseModel):
class CustomError(BaseModel):
foo: str

@field_validator("foo")
# @field_validator("foo")
@validator("foo")
def value_must_be_foo(cls, value):
if value != "foo":
# this is not JSON serializable if included in the error context
Expand Down

0 comments on commit 01af788

Please sign in to comment.