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

Commit 654601b

Browse files
add user-creation code block while executing extension
1 parent 080a4ca commit 654601b

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

aqueductcore/backend/services/extensions.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,17 @@ async def execute(
157157
)
158158

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

0 commit comments

Comments
 (0)