@@ -1488,7 +1488,7 @@ async def fetch_session_creation_data(
14881488 db_sess ,
14891489 spec .user_scope .domain_name ,
14901490 str (spec .user_scope .group_id ),
1491- spec .main_access_key ,
1491+ spec .access_key ,
14921492 )
14931493 image_infos = await self ._resolve_image_info (db_sess , image_refs )
14941494
@@ -1509,7 +1509,7 @@ async def fetch_session_creation_data(
15091509 dotfile_data = await self ._fetch_dotfiles (
15101510 db_sess ,
15111511 spec .user_scope ,
1512- spec .main_access_key ,
1512+ spec .access_key ,
15131513 vfolder_mounts ,
15141514 )
15151515
@@ -1553,7 +1553,7 @@ async def fetch_session_creation_context(
15531553 db_sess ,
15541554 spec .user_scope .domain_name ,
15551555 str (spec .user_scope .group_id ),
1556- spec .main_access_key ,
1556+ spec .access_key ,
15571557 )
15581558 image_infos = await self ._resolve_image_info (db_sess , image_refs )
15591559
@@ -2927,11 +2927,6 @@ async def _get_sessions_by_statuses(
29272927 session_id = session .id ,
29282928 creation_id = session .creation_id or "" ,
29292929 main_access_key = AccessKey (owner_main_ak ) if owner_main_ak else AccessKey ("" ),
2930- else AccessKey ("" ),
2931- >> >> >> > e35caec08 (refactor (BA - 5650 - I ): drop stray non - BA - 5650 changes from slice )
2932- == == == =
2933- main_access_key = AccessKey (owner_main_ak ) if owner_main_ak else AccessKey ("" ),
2934- >> >> >> > 9 fbd63dda (fix (BA - 5650 - I ): resolve cascaded rebase conflicts and align tests )
29352930 reason = "triggered-by-scheduler" ,
29362931 )
29372932 scheduled_sessions .append (scheduled_session )
@@ -2979,11 +2974,6 @@ async def _get_scheduled_sessions(self, db_sess: SASession) -> list[ScheduledSes
29792974 session_id = session .id ,
29802975 creation_id = session .creation_id or "" ,
29812976 main_access_key = AccessKey (owner_main_ak ) if owner_main_ak else AccessKey ("" ),
2982- else AccessKey ("" ),
2983- >> >> >> > e35caec08 (refactor (BA - 5650 - I ): drop stray non - BA - 5650 changes from slice )
2984- == == == =
2985- >> >> >> > c2dad7307 (refactor (BA - 5650 - I ): test and remaining ORM updates )
2986- > >> >> >> f0233454a (refactor (BA - 5650 - I ): test and remaining ORM updates )
29872977 reason = "triggered-by-scheduler" ,
29882978 )
29892979 )
@@ -4827,21 +4817,3 @@ async def _get_db_now_in_session(self, db_sess: SASession) -> datetime:
48274817 """
48284818 result = await db_sess .execute (sa .select (sa .func .now ()))
48294819 return result .scalar_one ()
4830-
4831- async def resolve_main_access_keys (
4832- self , session_ids : Sequence [SessionId ]
4833- ) -> dict [SessionId , AccessKey ]:
4834- """Resolve the main access key for each session's owner.
4835-
4836- Joins sessions → users to look up the owner's main_access_key.
4837- """
4838- if not session_ids :
4839- return {}
4840- async with self ._db .begin_readonly_session () as db_sess :
4841- stmt = (
4842- sa .select (SessionRow .id , UserRow .main_access_key )
4843- .join (UserRow , SessionRow .user_uuid == UserRow .uuid )
4844- .where (SessionRow .id .in_ ([sid for sid in session_ids ]))
4845- )
4846- rows = (await db_sess .execute (stmt )).all ()
4847- return {SessionId (row [0 ]): AccessKey (row [1 ]) for row in rows if row [1 ] is not None }
0 commit comments