|
|
@@ -87,13 +87,24 @@ class LegacySqlAlchemyInputReader:
|
|
|
sessions: async_sessionmaker[AsyncSession],
|
|
|
*,
|
|
|
include_raw_upload: bool = False,
|
|
|
+ uploaded_sessions: async_sessionmaker[AsyncSession] | None = None,
|
|
|
) -> None:
|
|
|
self._sessions = sessions
|
|
|
self._include_raw_upload = include_raw_upload
|
|
|
+ self._uploaded_sessions = uploaded_sessions
|
|
|
|
|
|
async def read_topic_graph(
|
|
|
self, *, execution_id: int, topic_build_id: int, topic_id: int
|
|
|
) -> dict[str, Any]:
|
|
|
+ if execution_id == 0 and self._uploaded_sessions is not None:
|
|
|
+ return await LegacySqlAlchemyInputReader(
|
|
|
+ self._uploaded_sessions,
|
|
|
+ include_raw_upload=self._include_raw_upload,
|
|
|
+ ).read_topic_graph(
|
|
|
+ execution_id=execution_id,
|
|
|
+ topic_build_id=topic_build_id,
|
|
|
+ topic_id=topic_id,
|
|
|
+ )
|
|
|
async with self._sessions() as session:
|
|
|
execution = None
|
|
|
if execution_id != 0:
|