From 654601bda60262252ebc45bed18e242094d587d9 Mon Sep 17 00:00:00 2001
From: Jatin <140497406+jatinriverlane@users.noreply.github.com>
Date: Tue, 12 Nov 2024 15:48:38 +0000
Subject: [PATCH 1/2] add user-creation code block while executing extension

---
 aqueductcore/backend/services/extensions.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/aqueductcore/backend/services/extensions.py b/aqueductcore/backend/services/extensions.py
index df5e00b7..1ca71735 100644
--- a/aqueductcore/backend/services/extensions.py
+++ b/aqueductcore/backend/services/extensions.py
@@ -157,6 +157,17 @@ async def execute(
             )
 
         start_time = datetime.now().astimezone(timezone.utc)
+
+        db_user_statement = select(orm.User).where(orm.User.uuid == user_info.uuid)
+        db_user = (await db_session.execute(db_user_statement)).scalars().first()
+
+        if not db_user:
+            db_user = orm.User(
+                uuid=user_info.uuid,
+                username=user_info.username,
+            )
+        db_session.add(db_user)
+
         task = await _execute_task(
             extension_directory_name=cwd.name,
             shell_script=rich_script,

From b48f00b08d5047f4b72683c66b30d7fc0ef4e7f2 Mon Sep 17 00:00:00 2001
From: Jatin <140497406+jatinriverlane@users.noreply.github.com>
Date: Tue, 12 Nov 2024 16:14:47 +0000
Subject: [PATCH 2/2] fix too-many-locals pylint warning

---
 aqueductcore/backend/services/extensions.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aqueductcore/backend/services/extensions.py b/aqueductcore/backend/services/extensions.py
index 1ca71735..646ec004 100644
--- a/aqueductcore/backend/services/extensions.py
+++ b/aqueductcore/backend/services/extensions.py
@@ -104,6 +104,7 @@ class ExtensionAction(BaseModel):
     script: str
     parameters: List[ExtensionParameter]
 
+    # pylint: disable=too-many-locals
     async def execute(
         self,
         user_info: UserInfo,