add autoload to recordmodel
This commit is contained in:
parent
b4ba3ef4c8
commit
1f23ba5490
1 changed files with 12 additions and 8 deletions
|
|
@ -154,17 +154,21 @@ class ObjectModel(BaseModel):
|
||||||
|
|
||||||
|
|
||||||
class RecordModel(BaseModel):
|
class RecordModel(BaseModel):
|
||||||
record_id: ClassVar[str] = "open_notebook:default_models"
|
record_id: ClassVar[str]
|
||||||
|
|
||||||
@classmethod
|
def __init__(self, **kwargs):
|
||||||
def load(cls):
|
super().__init__(**kwargs)
|
||||||
result = repo_query(f"SELECT * FROM {cls.record_id};")
|
self.load()
|
||||||
|
|
||||||
|
def load(self):
|
||||||
|
result = repo_query(f"SELECT * FROM {self.record_id};")
|
||||||
if result:
|
if result:
|
||||||
result = result[0]
|
result = result[0]
|
||||||
dm = cls(**result)
|
for key, value in result.items():
|
||||||
return dm
|
if hasattr(self, key):
|
||||||
return cls()
|
setattr(self, key, value)
|
||||||
|
return self
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def update(self, data):
|
def update(self, data):
|
||||||
repo_update(self.record_id, data)
|
repo_update(self.record_id, data)
|
||||||
|
return self.load()
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue