create record if doesnt exist

2024-11-10 13:27:30 -03:00 · 2024-11-10 13:27:30 -03:00 · 66edfc1e2b
commit 66edfc1e2b
parent d5be2b0d5b
1 changed files with 7 additions and 3 deletions
--- a/open_notebook/domain/base.py
+++ b/open_notebook/domain/base.py
@ -204,9 +204,13 @@ class RecordModel(BaseModel):
        result = repo_query(f"SELECT * FROM {self.record_id};")
        if result:
            result = result[0]
-            for key, value in result.items():
-                if hasattr(self, key):
-                    setattr(self, key, value)
+        else:
+            repo_create(self.record_id, {})
+            result = {}
+        for key, value in result.items():
+            if hasattr(self, key):
+                setattr(self, key, value)
+
        return self

    def update(self, data):