fix: make all examples runnable (hatchet-dev#302)

Author: mrkaye97

examples/cancellation/worker.py

@@ -23,8 +23,13 @@ async def step1(self, context: Context) -> None:
             print("Cancelled")
 
 
-workflow = CancelWorkflow()
-worker = hatchet.worker("cancellation-worker", max_runs=4)
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = CancelWorkflow()
+    worker = hatchet.worker("cancellation-worker", max_runs=4)
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/concurrency_limit_rr/worker.py

@@ -33,8 +33,13 @@ def step1(self, context: Context) -> None:
         pass
 
 
-workflow = ConcurrencyDemoWorkflowRR()
-worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10)
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = ConcurrencyDemoWorkflowRR()
+    worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10)
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/delayed/worker.py

@@ -32,8 +32,13 @@ def step1(self, context: Context) -> None:
         print(f"message \t {context.workflow_input()['message']}")
 
 
-worker = hatchet.worker("delayed-worker", max_runs=4)
-worker.register_workflow(PrintSchedule())
-worker.register_workflow(PrintPrinter())
+def main() -> None:
+    worker = hatchet.worker("delayed-worker", max_runs=4)
+    worker.register_workflow(PrintSchedule())
+    worker.register_workflow(PrintPrinter())
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/manual_trigger/worker.py

@@ -47,8 +47,13 @@ def step2(self, context: Context) -> dict[str, str]:
         return {"step2": "data2"}
 
 
-workflow = ManualTriggerWorkflow()
-worker = hatchet.worker("manual-worker", max_runs=4)
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = ManualTriggerWorkflow()
+    worker = hatchet.worker("manual-worker", max_runs=4)
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/overrides/worker.py

@@ -69,8 +69,13 @@ def step4(self, context: Context) -> dict[str, str]:
         }
 
 
-workflow = OverridesWorkflow()
-worker = hatchet.worker("overrides-worker")
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = OverridesWorkflow()
+    worker = hatchet.worker("overrides-worker")
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/rate_limit/worker.py

@@ -23,3 +23,7 @@ def main() -> None:
     worker.register_workflow(RateLimitWorkflow())
 
     worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/sticky_workers/worker.py

@@ -36,7 +36,12 @@ def child(self, context: Context) -> dict[str, str | None]:
         return {"worker": context.worker.id()}
 
 
-worker = hatchet.worker("sticky-worker", max_runs=10)
-worker.register_workflow(StickyWorkflow())
-worker.register_workflow(StickyChildWorkflow())
-worker.start()
+def main() -> None:
+    worker = hatchet.worker("sticky-worker", max_runs=10)
+    worker.register_workflow(StickyWorkflow())
+    worker.register_workflow(StickyChildWorkflow())
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()