Skip to content

Commit

Permalink
oops/
Browse files Browse the repository at this point in the history
  • Loading branch information
jonaraphael committed Nov 22, 2024
1 parent f80d4e7 commit fb3f849
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
5 changes: 3 additions & 2 deletions cerulean_cloud/cloud_function_ais_analysis/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,8 @@ async def handle_asa_request(request):
print(f"Deactivating sources for slick {slick.id}")
await db_client.deactivate_sources_for_slick(slick.id)
previous_asa = {
slick: await db_client.get_previous_asa(slick) for slick in slicks
slick.id: await db_client.get_previous_asa(slick.id)
for slick in slicks
}

print(f"Running ASA ({run_flags}) on scene_id: {scene_id}")
Expand All @@ -107,7 +108,7 @@ async def handle_asa_request(request):
analyzers_to_run = [
analyzer
for analyzer in analyzers
if analyzer.source_type not in previous_asa[slick]
if analyzer.source_type not in previous_asa[slick.id]
]
if len(analyzers_to_run) == 0:
continue
Expand Down
4 changes: 2 additions & 2 deletions cerulean_cloud/database_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -357,14 +357,14 @@ async def deactivate_sources_for_slick(self, slick_id):
.values(active=False)
)

async def get_previous_asa(self, slick):
async def get_previous_asa(self, slick_id):
"""Return a list of ASA types that have been run for a slick."""
return (
(
await self.session.execute(
select(db.Source.type)
.join(db.SlickToSource.source1)
.where(db.SlickToSource.slick == slick)
.where(db.SlickToSource.slick.id == slick_id)
)
)
.scalars()
Expand Down

0 comments on commit fb3f849

Please sign in to comment.