mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-07-02 07:01:30 +00:00
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
import contextvars
|
|
import re
|
|
from typing import Any, Dict
|
|
|
|
|
|
class AsyncLocal:
|
|
"""
|
|
一个异步环境下安全的上下文存储对象，用于替代 werkzeug.local.Local。
|
|
内部使用一个 ContextVar 来存储一个字典。
|
|
"""
|
|
|
|
def __init__(self, context_var_name: str = "_async_local_storage"):
|
|
object.__setattr__(self, "_storage", contextvars.ContextVar(
|
|
context_var_name,
|
|
default={}
|
|
))
|
|
|
|
def __setattr__(self, key: str, value: Any) -> None:
|
|
if key.startswith("_"):
|
|
object.__setattr__(self, key, value)
|
|
return
|
|
self.set(key, value)
|
|
|
|
def __getattr__(self, key: str) -> Any:
|
|
value = self.get(key, default=None)
|
|
if value is None:
|
|
raise AttributeError(f"{self.__class__.__name__!s} has no attribute {key!r}")
|
|
return value
|
|
|
|
def __delattr__(self, key: str) -> None:
|
|
if key.startswith("_"):
|
|
object.__delattr__(self, key)
|
|
return
|
|
if key not in self._storage.get():
|
|
raise AttributeError(f"{self.__class__.__name__!s} has no attribute {key!r}")
|
|
self.delete(key)
|
|
|
|
def set(self, key: str, value: Any) -> None:
|
|
current_data = self._storage.get().copy()
|
|
current_data[key] = value
|
|
self._storage.set(current_data)
|
|
|
|
def get(self, key: str, default: Any = None) -> Any:
|
|
return self._storage.get().get(key, default)
|
|
|
|
def delete(self, key: str) -> None:
|
|
current_data = self._storage.get().copy()
|
|
if key in current_data:
|
|
del current_data[key]
|
|
self._storage.set(current_data)
|
|
|
|
def clear(self) -> None:
|
|
self._storage.set({})
|
|
|
|
|
|
thread_local = AsyncLocal()
|
|
exclude_encrypted_fields = ('secret_type', 'secret_strategy', 'password_rules')
|
|
similar_encrypted_pattern = re.compile(
|
|
'password|secret|token|passphrase|private|key|cert', re.IGNORECASE
|
|
)
|
|
|
|
|
|
def _find(attr):
|
|
return getattr(thread_local, attr, None)
|