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

🎨 Add Schema.get method

Browse source
This commit is contained in:
Brian Wiborg 2025-10-02 00:49:12 +02:00
parent ed55c3708f
commit a3d9862c4e
No known key found for this signature in database
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -39,9 +39,7 @@ class ModelMeta(type(TortoiseModel)):
class BoundSchema:
def __call__(self, readonly: bool = False):
if readonly:
return self.readonly
return self.model
return self.get(readonly)
@property
def model(self):
@ -68,6 +66,11 @@ class ModelMeta(type(TortoiseModel)):
exclude_readonly=True,
)
def get(self, readonly: bool = False):
if readonly:
return self.readonly
return self.model
new_cls.Schema = BoundSchema()
return new_cls