Allow DI args into an app handlers
This commit is contained in:
@@ -12,7 +12,8 @@ from turbosloth.internal_types import QTYPE, BTYPE, PTYPE, HTYPE
|
||||
from turbosloth.req_schema import UnwrappedRequest
|
||||
from turbosloth.schema import RequestBody, HeaderParam
|
||||
|
||||
app = SlothApp(di_autodoc_prefix='/didoc', serialize_selector=SerializeSelector(default_content_type='application/json'))
|
||||
app = SlothApp(di_autodoc_prefix='/didoc',
|
||||
serialize_selector=SerializeSelector(default_content_type='application/json'))
|
||||
|
||||
|
||||
# @app.get("/")
|
||||
@@ -75,17 +76,31 @@ class PTYPESchema:
|
||||
# return SerializedResponse(200, {}, resp)
|
||||
|
||||
|
||||
class DummyDbConnection:
|
||||
constructions = [0]
|
||||
|
||||
def __init__(self):
|
||||
self.constructions[0] += 1
|
||||
|
||||
|
||||
@app.inj_repo.mark_injector()
|
||||
def create_db_connection() -> DummyDbConnection:
|
||||
return DummyDbConnection()
|
||||
|
||||
|
||||
@app.post("/test/body/{a}")
|
||||
async def test_body(r: RequestBody(UserPostSchema),
|
||||
q1: str,
|
||||
a: str,
|
||||
h1: HeaderParam(str, 'header1')) -> SerializedResponse:
|
||||
h1: HeaderParam(str, 'header1'),
|
||||
db: DummyDbConnection) -> SerializedResponse:
|
||||
print(r.user_id)
|
||||
resp = {
|
||||
'req': r,
|
||||
'q1': q1,
|
||||
'h1': h1,
|
||||
'a': a
|
||||
'a': a,
|
||||
'db': db.constructions
|
||||
}
|
||||
return SerializedResponse(200, {}, resp)
|
||||
|
||||
|
||||
@@ -264,7 +264,11 @@ class SlothApp(HTTPApp, WSApp, LifespanApp, MethodRoutersApp):
|
||||
def decorator(fn: HandlerType):
|
||||
|
||||
path_substs = self.router.find_pattern_substs(path_pattern)
|
||||
config = EndpointConfig.from_handler(fn, path_substs)
|
||||
|
||||
injectors = self.inj_repo.filtered_injectors(True, True)
|
||||
injected_types = list(map(lambda x: x.injects, injectors))
|
||||
|
||||
config = EndpointConfig.from_handler(fn, path_substs, injected_types)
|
||||
|
||||
fork_with = set()
|
||||
|
||||
|
||||
@@ -1,7 +1,8 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import inspect
|
||||
from typing import TYPE_CHECKING, overload, Any, TypeAlias, Optional, get_origin, get_args, Callable, get_type_hints
|
||||
from typing import TYPE_CHECKING, overload, Any, TypeAlias, Optional, get_origin, get_args, Callable, get_type_hints, \
|
||||
Iterable
|
||||
|
||||
from dataclasses import dataclass
|
||||
from typing import TypeVar, Generic, Annotated
|
||||
@@ -72,7 +73,7 @@ class EndpointConfig:
|
||||
type_replacement: dict[str, type]
|
||||
|
||||
@classmethod
|
||||
def from_handler(cls, h: Callable, path_substituts: set[str]) -> EndpointConfig:
|
||||
def from_handler(cls, h: Callable, path_substituts: set[str], ignore_types: Iterable[type]) -> EndpointConfig:
|
||||
body_schema = None
|
||||
query_schemas = {}
|
||||
path_schemas = {}
|
||||
@@ -82,6 +83,8 @@ class EndpointConfig:
|
||||
handle_hints = get_endpoint_params_info(h)
|
||||
for argname, s in handle_hints.items():
|
||||
tp = s.schema
|
||||
if tp in ignore_types:
|
||||
continue
|
||||
type_replacement[argname] = s.replacement_type
|
||||
|
||||
if argname == 'return':
|
||||
|
||||
Reference in New Issue
Block a user