diff --git a/src/ohmyapi/db/model/model.py b/src/ohmyapi/db/model/model.py index 20359d6..929ff60 100644 --- a/src/ohmyapi/db/model/model.py +++ b/src/ohmyapi/db/model/model.py @@ -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