style: format all Python files with Black
Apply Black formatter to the entire codebase for consistent code style. Co-Authored-By: Claude (glm-5) <noreply@anthropic.com>
This commit is contained in:
22
gui/utils.py
22
gui/utils.py
@@ -61,24 +61,25 @@ def admin_only(func):
|
||||
Returns:
|
||||
如果是管理员,返回原函数的结果;否则返回 None
|
||||
"""
|
||||
|
||||
@functools.wraps(func)
|
||||
def wrapper(self, *args, **kwargs):
|
||||
# 尝试从 self 获取 session_manager
|
||||
session_manager = getattr(self, 'session_manager', None)
|
||||
session_manager = getattr(self, "session_manager", None)
|
||||
|
||||
# 如果没有 session_manager,尝试从 main_window 获取
|
||||
if session_manager is None:
|
||||
main_window = getattr(self, 'main_window', None)
|
||||
main_window = getattr(self, "main_window", None)
|
||||
if main_window:
|
||||
session_manager = getattr(main_window, 'session_manager', None)
|
||||
session_manager = getattr(main_window, "session_manager", None)
|
||||
|
||||
# 检查是否为管理员
|
||||
if session_manager and hasattr(session_manager, 'is_admin'):
|
||||
if session_manager and hasattr(session_manager, "is_admin"):
|
||||
if session_manager.is_admin():
|
||||
return func(self, *args, **kwargs)
|
||||
|
||||
# 非管理员,记录日志并返回 None
|
||||
logger = getattr(self, 'logger', None)
|
||||
logger = getattr(self, "logger", None)
|
||||
if logger:
|
||||
logger.debug(f"权限拒绝: {func.__name__} 需要管理员权限")
|
||||
|
||||
@@ -103,24 +104,25 @@ def require_session(func):
|
||||
Returns:
|
||||
如果有有效会话,返回原函数的结果;否则返回 None
|
||||
"""
|
||||
|
||||
@functools.wraps(func)
|
||||
def wrapper(self, *args, **kwargs):
|
||||
# 尝试从 self 获取 session_manager
|
||||
session_manager = getattr(self, 'session_manager', None)
|
||||
session_manager = getattr(self, "session_manager", None)
|
||||
|
||||
# 如果没有 session_manager,尝试从 main_window 获取
|
||||
if session_manager is None:
|
||||
main_window = getattr(self, 'main_window', None)
|
||||
main_window = getattr(self, "main_window", None)
|
||||
if main_window:
|
||||
session_manager = getattr(main_window, 'session_manager', None)
|
||||
session_manager = getattr(main_window, "session_manager", None)
|
||||
|
||||
# 检查会话是否有效
|
||||
if session_manager and hasattr(session_manager, 'is_authenticated'):
|
||||
if session_manager and hasattr(session_manager, "is_authenticated"):
|
||||
if session_manager.is_authenticated():
|
||||
return func(self, *args, **kwargs)
|
||||
|
||||
# 会话无效,记录日志并返回 None
|
||||
logger = getattr(self, 'logger', None)
|
||||
logger = getattr(self, "logger", None)
|
||||
if logger:
|
||||
logger.warning(f"会话无效: {func.__name__} 需要有效会话")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user