|
| 1 | +from typing import Any, Dict, List, Optional |
| 2 | + |
| 3 | +from django.conf import settings as django_settings |
| 4 | +from django.core.signals import setting_changed |
| 5 | +from ninja.pagination import PaginationBase |
| 6 | +from ninja.throttling import BaseThrottle |
| 7 | +from pydantic import BaseModel, Field |
| 8 | +from typing_extensions import Annotated |
| 9 | + |
| 10 | +from ninja_extra.conf.decorator import AllowTypeOfSource |
| 11 | +from ninja_extra.interfaces.ordering import OrderingBase |
| 12 | +from ninja_extra.interfaces.route_context import RouteContextBase |
| 13 | +from ninja_extra.interfaces.searching import SearchingBase |
| 14 | + |
| 15 | +_GenericModelValidator = AllowTypeOfSource( |
| 16 | + validator=lambda source, value: isinstance(value, type) |
| 17 | + and issubclass(value, source), |
| 18 | + error_message=lambda source, |
| 19 | + value: f"Expected type of {source.__name__}, received: {type(value)}", |
| 20 | +) |
| 21 | +PaginationClassHandlerType = Annotated[PaginationBase, _GenericModelValidator] |
| 22 | +ThrottlingClassHandlerType = Annotated[BaseThrottle, _GenericModelValidator] |
| 23 | +SearchingClassHandlerType = Annotated[SearchingBase, _GenericModelValidator] |
| 24 | +OrderingClassHandlerType = Annotated[OrderingBase, _GenericModelValidator] |
| 25 | +RouteContextHandlerType = Annotated[RouteContextBase, _GenericModelValidator] |
| 26 | + |
| 27 | +_InjectorModuleValidator = AllowTypeOfSource( |
| 28 | + validator=lambda source, value: value is not None, |
| 29 | + error_message=lambda source, |
| 30 | + value: f"Expected PaginationBase object, received: {type(value)}", |
| 31 | +) |
| 32 | +InjectorModuleHandlerType = Annotated[Any, _InjectorModuleValidator] |
| 33 | + |
| 34 | + |
| 35 | +class UserDefinedSettingsMapper: |
| 36 | + def __init__(self, data: dict) -> None: |
| 37 | + self.__dict__ = data |
| 38 | + |
| 39 | + |
| 40 | +NinjaEXTRA_SETTINGS_DEFAULTS = { |
| 41 | + "INJECTOR_MODULES": [], |
| 42 | + "PAGINATION_CLASS": "ninja.pagination.LimitOffsetPagination", |
| 43 | + "THROTTLE_CLASSES": [ |
| 44 | + "ninja_extra.throttling.AnonRateThrottle", |
| 45 | + "ninja_extra.throttling.UserRateThrottle", |
| 46 | + ], |
| 47 | + "THROTTLE_RATES": {"user": None, "anon": None}, |
| 48 | + "ORDERING_CLASS": "ninja_extra.ordering.Ordering", |
| 49 | + "SEARCHING_CLASS": "ninja_extra.searching.Searching", |
| 50 | + "ROUTE_CONTEXT_CLASS": "ninja_extra.context.RouteContext", |
| 51 | +} |
| 52 | + |
| 53 | +USER_SETTINGS = UserDefinedSettingsMapper( |
| 54 | + getattr(django_settings, "NINJA_EXTRA", NinjaEXTRA_SETTINGS_DEFAULTS) |
| 55 | +) |
| 56 | + |
| 57 | + |
| 58 | +class NinjaExtraSettings(BaseModel): |
| 59 | + class Config: |
| 60 | + from_attributes = True |
| 61 | + validate_assignment = True |
| 62 | + |
| 63 | + PAGINATION_CLASS: PaginationClassHandlerType = Field( # type: ignore[assignment] |
| 64 | + "ninja.pagination.LimitOffsetPagination", |
| 65 | + ) |
| 66 | + PAGINATION_PER_PAGE: int = Field(100) |
| 67 | + THROTTLE_RATES: Dict[str, Optional[str]] = Field( |
| 68 | + {"user": "1000/day", "anon": "100/day"} |
| 69 | + ) |
| 70 | + THROTTLE_CLASSES: List[ThrottlingClassHandlerType] = Field( |
| 71 | + [ |
| 72 | + "ninja_extra.throttling.AnonRateThrottle", # type: ignore[list-item] |
| 73 | + "ninja_extra.throttling.UserRateThrottle", # type: ignore[list-item] |
| 74 | + ] |
| 75 | + ) |
| 76 | + NUM_PROXIES: Optional[int] = None |
| 77 | + INJECTOR_MODULES: List[InjectorModuleHandlerType] = [] |
| 78 | + ORDERING_CLASS: OrderingClassHandlerType = Field( # type: ignore[assignment] |
| 79 | + "ninja_extra.ordering.Ordering", |
| 80 | + ) |
| 81 | + SEARCHING_CLASS: SearchingClassHandlerType = Field( # type: ignore[assignment] |
| 82 | + "ninja_extra.searching.Searching", |
| 83 | + ) |
| 84 | + ROUTE_CONTEXT_CLASS: RouteContextHandlerType = Field( # type: ignore[assignment] |
| 85 | + "ninja_extra.context.RouteContext", |
| 86 | + ) |
| 87 | + |
| 88 | + |
| 89 | +settings = NinjaExtraSettings.model_validate(USER_SETTINGS) |
| 90 | + |
| 91 | + |
| 92 | +def reload_settings(*args: Any, **kwargs: Any) -> None: # pragma: no cover |
| 93 | + global settings |
| 94 | + |
| 95 | + setting, value = kwargs["setting"], kwargs["value"] |
| 96 | + |
| 97 | + if setting == "NINJA_EXTRA": |
| 98 | + settings = NinjaExtraSettings.model_validate(UserDefinedSettingsMapper(value)) |
| 99 | + |
| 100 | + |
| 101 | +setting_changed.connect(reload_settings) # pragma: no cover |
0 commit comments