Skip to content
This repository was archived by the owner on Jun 19, 2025. It is now read-only.

Commit d16fa40

Browse files
feat(fix-failing-tasks): Update execute method to create non existing user (#236)
* add user-creation code block while executing extension * fix too-many-locals pylint warning
1 parent 080a4ca commit d16fa40

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

aqueductcore/backend/services/extensions.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ class ExtensionAction(BaseModel):
104104
script: str
105105
parameters: List[ExtensionParameter]
106106

107+
# pylint: disable=too-many-locals
107108
async def execute(
108109
self,
109110
user_info: UserInfo,
@@ -157,6 +158,17 @@ async def execute(
157158
)
158159

159160
start_time = datetime.now().astimezone(timezone.utc)
161+
162+
db_user_statement = select(orm.User).where(orm.User.uuid == user_info.uuid)
163+
db_user = (await db_session.execute(db_user_statement)).scalars().first()
164+
165+
if not db_user:
166+
db_user = orm.User(
167+
uuid=user_info.uuid,
168+
username=user_info.username,
169+
)
170+
db_session.add(db_user)
171+
160172
task = await _execute_task(
161173
extension_directory_name=cwd.name,
162174
shell_script=rich_script,

0 commit comments

Comments
 (0)