From a3d9862c4e063b8e88879b10f83e17782b3a9de4 Mon Sep 17 00:00:00 2001 From: Brian Wiborg Date: Thu, 2 Oct 2025 00:49:12 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20Add=20Schema.get=20method?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/ohmyapi/db/model/model.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) 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