From 15a5fff7a439de9d3594609ed823c54eafe7d4e0 Mon Sep 17 00:00:00 2001 From: MystyPy Date: Sat, 13 Apr 2024 18:05:24 +1000 Subject: [PATCH] Add startup/shutdown to options --- starlette_plus/core.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/starlette_plus/core.py b/starlette_plus/core.py index 0118fe7..269ec69 100644 --- a/starlette_plus/core.py +++ b/starlette_plus/core.py @@ -18,7 +18,7 @@ import asyncio import inspect import logging -from collections.abc import Callable, Iterator +from collections.abc import Callable, Coroutine, Iterator from typing import TYPE_CHECKING, Any, Self, TypeAlias, TypedDict, Unpack from starlette.applications import Starlette @@ -47,6 +47,8 @@ class ApplicationOptions(TypedDict, total=False): views: list[View] access_log: bool middleware: list[Middleware] + on_startup: list[Callable[[], Coroutine[Any, Any, None]]] + on_shutdown: list[Callable[[], Coroutine[Any, Any, None]]] __all__ = ("Application", "View", "route", "limit")