ソースを参照

fix: add properties to retrieve created by account and end user in Wo… (#12129)

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 ヶ月 前
コミット
4e76f2fc44
共有1 個のファイルを変更した12 個の追加0 個の削除を含む
  1. 12 0
      api/models/workflow.py

+ 12 - 0
api/models/workflow.py

@@ -415,6 +415,18 @@ class WorkflowRun(db.Model):  # type: ignore[name-defined]
     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 {}