From 4e76f2fc4411b1e711d0792b0f6f1f5e0fd52fd3 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Thu, 26 Dec 2024 19:50:45 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20add=20properties=20to=20retrieve=20creat?= =?UTF-8?q?ed=20by=20account=20and=20end=20user=20in=20Wo=E2=80=A6=20(#121?= =?UTF-8?q?29)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: -LAN- --- api/models/workflow.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/api/models/workflow.py b/api/models/workflow.py index 32a0860b77..8642df8adb 100644 --- a/api/models/workflow.py +++ b/api/models/workflow.py @@ -414,6 +414,18 @@ class WorkflowRun(db.Model): # type: ignore[name-defined] finished_at = db.Column(db.DateTime) exceptions_count = db.Column(db.Integer, server_default=db.text("0")) + @property + def created_by_account(self): + created_by_role = CreatedByRole(self.created_by_role) + return db.session.get(Account, self.created_by) if created_by_role == CreatedByRole.ACCOUNT else None + + @property + def created_by_end_user(self): + from models.model import EndUser + + created_by_role = CreatedByRole(self.created_by_role) + return db.session.get(EndUser, self.created_by) if created_by_role == CreatedByRole.END_USER else None + @property def graph_dict(self): return json.loads(self.graph) if self.graph else {}