distro-tools/apollo/rpmworker/rh_matcher_workflows.py
2023-02-01 22:37:16 +01:00

40 lines
1.0 KiB
Python

import datetime
from dataclasses import dataclass
from temporalio import workflow
@dataclass
class RhDefunctWorkflowInput:
supported_product_id: int
@workflow.defn
class RhMatcherWorkflow:
@workflow.run
async def run(self) -> list[int]:
supported_product_ids = await workflow.execute_activity(
"get_supported_products_with_rh_mirrors",
start_to_close_timeout=datetime.timedelta(seconds=20),
)
for supported_product_id in supported_product_ids:
await workflow.execute_activity(
"match_rh_repos",
supported_product_id,
start_to_close_timeout=datetime.timedelta(hours=12),
)
return supported_product_ids
@workflow.defn
class RhDefunctWorkflow:
@workflow.run
async def run(self, wf_in: RhDefunctWorkflowInput) -> None:
await workflow.execute_activity(
"block_remaining_rh_advisories",
wf_in.supported_product_id,
start_to_close_timeout=datetime.timedelta(hours=12),
)