WIP

examples/app.py

@@ -7,7 +7,6 @@
 from justscheduleit import Scheduler, after, every, recurrent
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 scheduler = Scheduler()

examples/cancellation.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+
+import logging
+from asyncio import CancelledError
+from datetime import timedelta
+
+import anyio
+
+from justscheduleit import Scheduler, every
+
+logging.basicConfig()
+logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
+
+scheduler = Scheduler()
+
+
+@scheduler.task(every(timedelta(seconds=5), delay=None))
+async def long_async_task():
+    print(f"{long_async_task.__name__} is triggered!")
+    try:
+        await anyio.sleep(15)
+    except CancelledError:
+        # Won't be cancelled in the normal flow (graceful shutdown)
+        print(f"Forced shutdown!")
+        raise
+    print(f"{long_async_task.__name__} has finished!")
+
+
+if __name__ == "__main__":
+    import justscheduleit
+
+    justscheduleit.run(scheduler)

examples/channel.py

@@ -9,7 +9,6 @@
 from justscheduleit.hosting import Host
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 
@@ -41,4 +40,4 @@ async def scheduled_background_task():
 if __name__ == "__main__":
     from justscheduleit import hosting
 
-    hosting.run(host)
+    exit(hosting.run(host))

examples/cron_cond.py → examples/cron.py

@@ -6,7 +6,6 @@
 from justscheduleit.cond.cron import cron
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 scheduler = Scheduler()

examples/failing_task.py → examples/failing_tasks.py

@@ -3,11 +3,9 @@
 import logging
 from datetime import timedelta
 
-import justscheduleit
 from justscheduleit import Scheduler, every
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 scheduler = Scheduler()
@@ -26,4 +24,6 @@ def a_sync_task():
 
 
 if __name__ == "__main__":
+    import justscheduleit
+
     justscheduleit.run(scheduler)

examples/fastdepends.py

@@ -9,7 +9,6 @@
 from justscheduleit import Scheduler, every
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 scheduler = Scheduler()
@@ -30,4 +29,6 @@ def print_task(
 
 
 if __name__ == "__main__":
-    scheduler.run()
+    import justscheduleit
+
+    justscheduleit.run(scheduler)

examples/finite_task.py

@@ -9,7 +9,6 @@
 from justscheduleit.cond import take_first
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 scheduler = Scheduler()

examples/host/app.py

@@ -10,7 +10,6 @@
 from justscheduleit.hosting import Host, ServiceLifetime
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 host = Host()
@@ -89,4 +88,4 @@ async def an_async_service_tpl(n: int):
 if __name__ == "__main__":
     from justscheduleit import hosting
 
-    hosting.run(host)
+    exit(hosting.run(host))

examples/host/failing_main_service.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+
+import time
+
+from examples.host.failing_service import host
+from justscheduleit.hosting import ServiceLifetime
+
+
+@host.service(main=True)
+def a_main_sync_service(service_lifetime: ServiceLifetime):
+    print(f"{a_main_sync_service.__name__} started")
+    service_lifetime.set_started()
+    crash_countdown = 5
+    while not service_lifetime.shutting_down.is_set():
+        print(f"{a_main_sync_service.__name__} running")
+        time.sleep(1)
+        crash_countdown -= 1
+        if crash_countdown == 0:
+            raise RuntimeError("This is a test error from _sync_")
+    print(f"{a_main_sync_service.__name__} stopped")
+
+
+if __name__ == "__main__":
+    from justscheduleit import hosting
+
+    exit(hosting.run(host))

examples/host/failing_service.py

@@ -8,7 +8,6 @@
 from justscheduleit.hosting import Host, ServiceLifetime
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 host = Host()
@@ -33,7 +32,7 @@ async def an_async_func():
     try:
         while True:
             print(f"{an_async_func.__name__} running")
-            await anyio.sleep(5)
+            await anyio.sleep(1)
             # raise RuntimeError("This is a test error from _async_")
     finally:
         print(f"{an_async_func.__name__} done")
@@ -42,4 +41,4 @@ async def an_async_func():
 if __name__ == "__main__":
     from justscheduleit import hosting
 
-    hosting.run(host)
+    exit(hosting.run(host))

examples/host/finite_service.py

@@ -6,7 +6,6 @@
 from justscheduleit.hosting import Host, ServiceLifetime
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 host = Host()
@@ -25,4 +24,4 @@ def a_sync_service(service_lifetime: ServiceLifetime):
 if __name__ == "__main__":
     from justscheduleit import hosting
 
-    hosting.run(host)
+    exit(hosting.run(host))

examples/host/http_app.py

@@ -4,18 +4,24 @@
 from datetime import timedelta
 
 import anyio
-from fastapi import FastAPI  # noqa
+from fastapi import FastAPI
+from starlette.responses import JSONResponse
 
 from justscheduleit import Scheduler, every
 from justscheduleit.hosting import Host
 from justscheduleit.http import UvicornService
 
 logging.basicConfig()
-
 logging.getLogger("justscheduleit").setLevel(logging.DEBUG)
 
 host = Host()
 
+http_api = FastAPI()
+host.add_service(UvicornService.for_app(http_api), name="http_api")
+
+scheduler = Scheduler()
+host.add_service(scheduler, name="scheduler")
+
 
 @host.service()
 async def background_queue_consumer():
@@ -28,33 +34,22 @@ async def background_queue_consumer():
         print(f"{background_queue_consumer.__name__} done")
 
 
-scheduler = Scheduler()
-host.add_service(scheduler, name="scheduler")
-
-
 @scheduler.task(every(timedelta(seconds=3), delay=(1, 5)))
 async def heavy_background_task():
     print("Some work here")
 
 
-# Or Starlette
-http_api = FastAPI()
-host.add_service(UvicornService.for_app(http_api), name="http_api")
-
-
 @http_api.get("/predict")
 async def predict():
     return {"result": "some prediction"}
 
 
 @http_api.get("/health")
-async def health_check():
-    services = host.lifetime.service_lifetimes
-    statuses = {name: service.status for name, service in services.items()}
-    return {"status": "UP", "services": statuses}
+async def health_check() -> JSONResponse:
+    return JSONResponse(host.lifetime.status)
 
 
 if __name__ == "__main__":
     from justscheduleit import hosting
 
-    hosting.run(host)
+    exit(hosting.run(host))

examples/serve_async.py

@@ -5,17 +5,16 @@
 from examples.app import scheduler
 
 
-async def main():
+async def main(duration: int):
     from justscheduleit.scheduler import aserve
 
     # Scheduler in the same thread, same event loop
     async with aserve(scheduler):
-        for _ in range(10):
+        for _ in range(duration):
             await anyio.sleep(1)
             print("Main thread is running...")
         print("Main thread is done, exiting the app")
 
 
 if __name__ == "__main__":
-    # noinspection PyTypeChecker
-    anyio.run(main)
+    anyio.run(main, 10)

examples/serve_sync.py

@@ -17,7 +17,6 @@ def main():
         except KeyboardInterrupt:
             # Scheduler service is the only one in the host, so after the scheduler is done, the host will stop too
             scheduler_host.shutdown()
-            # scheduler.lifetime.shutdown()
 
     print("Main thread is done, exiting the app")
 

justscheduleit/_utils.py

@@ -76,7 +76,7 @@ def td_str(td: timedelta) -> str:
 DelayFactory = Union[Callable[[], timedelta], tuple[int, int], tuple[float, float], int, float, timedelta, None]
 
 
-@dc.dataclass(frozen=True)  # Enable slots when Python 3.10+
+@dc.dataclass(frozen=True, slots=True)
 class RandomDelay:  # https://en.wikipedia.org/wiki/Jitter#Types
     bounds: tuple[int, int] | tuple[float, float]
 
@@ -92,7 +92,7 @@ def __call__(self) -> timedelta:
         return timedelta(seconds=delay)
 
 
-@dc.dataclass(frozen=True)  # Enable slots when Python 3.10+
+@dc.dataclass(frozen=True, slots=True)
 class FixedDelay:
     value: timedelta
 

justscheduleit/cond/_core.py

@@ -88,7 +88,7 @@ async def _take(scheduler_lifetime: SchedulerLifetime):
     return _take
 
 
-@dc.dataclass(frozen=True)  # Enable slots when Python 3.10+
+@dc.dataclass(frozen=True, slots=True)
 class Every:
     """
     Triggers every `period`, with an (optional) additional `delay` (jitter).
@@ -133,7 +133,7 @@ def every(period: timedelta | str, /, *, delay: DelayFactory = DEFAULT_JITTER, s
     return Every(ensure_td(period), ensure_delay_factory(delay), stop_on_error)
 
 
-@dc.dataclass(frozen=True)  # Enable slots when Python 3.10+
+@dc.dataclass(frozen=True, slots=True)
 class Recurrent:
     """
     Triggers every `default_period` (unless overwritten), with an (optional) additional `delay` (jitter).
@@ -188,7 +188,7 @@ def recurrent(
     return Recurrent(ensure_td(default_interval), ensure_delay_factory(delay), stop_on_error)
 
 
-@dc.dataclass(frozen=True)  # Enable slots when Python 3.10+
+@dc.dataclass(frozen=True, slots=True)
 class After(Generic[T]):
     """
     Triggers every time `task` is completed, with an (optional) additional `delay` (jitter).

justscheduleit/cond/cron.py

@@ -14,7 +14,7 @@
 __all__ = ["cron"]
 
 
-@dc.dataclass(frozen=True)  # Enable slots when Python 3.10+
+@dc.dataclass(frozen=True, slots=True)
 class Cron:
     """
     Triggers according to the cron schedule, with an (optional) additional `delay` (jitter).