b8e9a80d93
* wait I forgot to comit locally * cp the entire core directory and then rm the .git subdir
26 lines
789 B
Python
26 lines
789 B
Python
from contextvars import ContextVar
|
|
from typing import Any, Dict
|
|
|
|
# Create context var at module level (outside middleware)
|
|
request_attributes: ContextVar[Dict[str, Any]] = ContextVar("request_attributes", default={})
|
|
|
|
|
|
# Helper functions
|
|
def set_ctx_attributes(attrs: Dict[str, Any]):
|
|
"""Set attributes in current context"""
|
|
current = request_attributes.get()
|
|
new_attrs = {**current, **attrs}
|
|
request_attributes.set(new_attrs)
|
|
|
|
|
|
def add_ctx_attribute(key: str, value: Any):
|
|
"""Add single attribute to current context"""
|
|
current = request_attributes.get()
|
|
new_attrs = {**current, key: value}
|
|
request_attributes.set(new_attrs)
|
|
|
|
|
|
def get_ctx_attributes() -> Dict[str, Any]:
|
|
"""Get all attributes from current context"""
|
|
return request_attributes.get()
|