{{cookiecutter.project_slug}}.app.context

Complexity

Total Complexity

6

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	_ContextLocalData.get()	0	2	1
A	_ContextProperty.request_payload()	0	3	1
A	_ContextProperty.secret_key()	0	3	1
A	_ContextLocalData.set()	0	2	1
A	_ContextLocalData.__init__()	0	3	1

from typing import Optional, Type

from flask import current_app, g
from pydantic import BaseModel


class _ContextLocalData:
    def __init__(self, key_name, default):
        self.key_name = key_name
        self.default = default

    def get(self, _g):
        return getattr(_g, self.key_name, self.default)

    def set(self, _g, value):
        setattr(_g, self.key_name, value)


class _ContextProperty:
    class ContextLocalData:
        request_payload = _ContextLocalData("request_payload", None)

    @property
    def secret_key(self) -> str:
        return current_app.secret_key

    # - request payload -

    @property
    def request_payload(self) -> Optional[BaseModel]:
        return self.ContextLocalData.request_payload.get(g)

    @request_payload.setter
    def request_payload(self, value: Type[BaseModel]):
        self.ContextLocalData.request_payload.set(g, value)


context_property = _ContextProperty()