18 lines
420 B
Python
18 lines
420 B
Python
|
from typing import Any
|
||
|
|
|
||
|
|
from pydantic import GetCoreSchemaHandler
|
||
|
|
from pydantic_core import CoreSchema, core_schema
|
||
|
|
|
||
|
|
|
||
|
|
class LengthUnit(int):
|
||
|
|
""""""
|
||
|
|
|
||
|
|
def __int__(self) -> int:
|
||
|
|
return self
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def __get_pydantic_core_schema__(
|
||
|
|
cls, source_type: Any, handler: GetCoreSchemaHandler
|
||
|
|
) -> CoreSchema:
|
||
|
|
return core_schema.no_info_after_validator_function(cls, handler(int))
|