baccenfutter/ohmyapi
1
2
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

🎨 pydantic model and readonly schema

Browse source
This commit is contained in:
Brian Wiborg 2025-09-28 02:12:14 +02:00
parent d509b58282
commit 09648fa292
No known key found for this signature in database
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/ohmyapi/db/model/model.py
View file

@ -12,7 +12,7 @@ class ModelMeta(type(TortoiseModel)):
class BoundSchema:
@property
def model(self):
"""Return a Pydantic model class for 'one' results."""
"""Return a Pydantic model class for serializing results."""
include = getattr(schema_opts, "include", None)
exclude = getattr(schema_opts, "exclude", None)
return pydantic_model_creator(
@ -20,6 +20,18 @@ class ModelMeta(type(TortoiseModel)):
name=f"{new_cls.__name__}Schema",
include=include,
exclude=exclude,
)
@property
def readonly(self):
"""Return a Pydantic model class for serializing readonly results."""
include = getattr(schema_opts, "include", None)
exclude = getattr(schema_opts, "exclude", None)
return pydantic_model_creator(
new_cls,
name=f"{new_cls.__name__}SchemaReadonly",
include=include,
exclude=exclude,
exclude_readonly=True,
)