Fix reschedule command

ooni · Sep 6, 2024 · 5e1d841 · 5e1d841
1 parent da89f2d
commit 5e1d841
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 6 deletions.
diff --git a/oonipipeline/src/oonipipeline/cli/commands.py b/oonipipeline/src/oonipipeline/cli/commands.py
@@ -206,11 +206,6 @@ def schedule(
 @cli.command()
 @probe_cc_option
 @test_name_option
-@click.option(
-    "--fast-fail",
-    is_flag=True,
-    help="should we fail immediately when we encounter an error?",
-)
 def reschedule(
     probe_cc: List[str],
     test_name: List[str],

diff --git a/oonipipeline/src/oonipipeline/temporal/client_operations.py b/oonipipeline/src/oonipipeline/temporal/client_operations.py
@@ -8,6 +8,7 @@
     ScheduleIdMap,
     schedule_all,
     schedule_backfill,
+    reschedule_all,
 )
 
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
@@ -153,6 +154,26 @@ async def create_schedules(
     )
 
 
+async def reschedule(
+    probe_cc: List[str],
+    test_name: List[str],
+    clickhouse_url: str,
+    data_dir: str,
+    temporal_config: TemporalConfig,
+) -> ScheduleIdMap:
+    log.info(f"rescheduling everything")
+
+    client = await temporal_connect(temporal_config=temporal_config)
+
+    return await reschedule_all(
+        client=client,
+        probe_cc=probe_cc,
+        test_name=test_name,
+        clickhouse_url=clickhouse_url,
+        data_dir=data_dir,
+    )
+
+
 async def get_status(
     temporal_config: TemporalConfig,
 ) -> Tuple[List[WorkflowExecution], List[WorkflowExecution]]:
@@ -243,7 +264,7 @@ def run_reschedule(
 ):
     try:
         asyncio.run(
-            create_schedules(
+            reschedule(
                 probe_cc=probe_cc,
                 test_name=test_name,
                 clickhouse_url=clickhouse_url,