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

🔥 Remove Model.Schema.many

Browse source
This commit is contained in:
Brian Wiborg 2025-09-28 01:48:03 +02:00
parent 1dcbab06b1
commit ac60c19551
No known key found for this signature in database
1 changed files with 2 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -11,30 +11,18 @@ class ModelMeta(type(TortoiseModel)):
class BoundSchema: class BoundSchema:
@property @property
def one(self): def model(self):
"""Return a Pydantic model class for 'one' results.""" """Return a Pydantic model class for 'one' results."""
include = getattr(schema_opts, "include", None) include = getattr(schema_opts, "include", None)
exclude = getattr(schema_opts, "exclude", None) exclude = getattr(schema_opts, "exclude", None)
return pydantic_model_creator( return pydantic_model_creator(
new_cls, new_cls,
name=f"{new_cls.__name__}SchemaOne", name=f"{new_cls.__name__}Schema",
include=include, include=include,
exclude=exclude, exclude=exclude,
exclude_readonly=True, exclude_readonly=True,
) )
@property
def many(self):
"""Return a Pydantic queryset class for 'many' results."""
include = getattr(schema_opts, "include", None)
exclude = getattr(schema_opts, "exclude", None)
return pydantic_queryset_creator(
new_cls,
name=f"{new_cls.__name__}SchemaMany",
include=include,
exclude=exclude,
)
new_cls.Schema = BoundSchema() new_cls.Schema = BoundSchema()
return new_cls return new_cls