mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 21:52:56 +08:00
3deffcb46c
Some checks failed
检查bot是否运行正常 / bot check (push) Has been cancelled
CodeQL Code Security Analysis / Analyze (${{ matrix.language }}) (none, javascript-typescript) (push) Has been cancelled
CodeQL Code Security Analysis / Analyze (${{ matrix.language }}) (none, python) (push) Has been cancelled
Sequential Lint and Type Check / ruff-call (push) Has been cancelled
Release Drafter / Update Release Draft (push) Has been cancelled
Force Sync to Aliyun / sync (push) Has been cancelled
Update Version / update-version (push) Has been cancelled
Sequential Lint and Type Check / pyright-call (push) Has been cancelled
- 在 `record_request.py` 和 `group_handle/__init__.py` 中引入了 `CacheRoot`，实现请求缓存，避免重复处理相同请求。 - 在 `exception.py` 中更新 `ForceAddGroupError` 类，新增 `group_id` 属性以便于错误处理。 - 在 `data_source.py` 中修改 `ForceAddGroupError` 的抛出逻辑，包含 `group_id` 信息。 - 更新 `cache` 类，支持类型化缓存字典和列表，增强缓存的类型安全性。 此更新提升了请求处理的效率和准确性，同时增强了错误信息的可追溯性。
450 lines
12 KiB
Python
450 lines
12 KiB
Python
from dataclasses import dataclass
|
||
import time
|
||
from typing import Any, Generic, TypeVar
|
||
|
||
T = TypeVar("T")
|
||
|
||
|
||
@dataclass
|
||
class CacheData(Generic[T]):
|
||
"""缓存数据类，存储数据和过期时间"""
|
||
|
||
value: T
|
||
expire_time: float = 0 # 0表示永不过期
|
||
|
||
|
||
class CacheDict(Generic[T]):
|
||
"""缓存字典类，提供类似普通字典的接口，数据只存储在内存中"""
|
||
|
||
def __init__(self, name: str, expire: int = 0):
|
||
"""初始化缓存字典
|
||
|
||
参数:
|
||
name: 字典名称
|
||
expire: 过期时间（秒），默认为0表示永不过期
|
||
"""
|
||
self.name = name.upper()
|
||
self.expire = expire
|
||
self._data: dict[str, CacheData[T]] = {}
|
||
|
||
def expire_time(self, key: str) -> float:
|
||
"""获取字典项的过期时间"""
|
||
data = self._data.get(key)
|
||
if data is None:
|
||
return 0
|
||
if data.expire_time > 0 and data.expire_time < time.time():
|
||
del self._data[key]
|
||
return 0
|
||
return data.expire_time
|
||
|
||
def __getitem__(self, key: str) -> T | None:
|
||
"""获取字典项
|
||
|
||
参数:
|
||
key: 字典键
|
||
|
||
返回:
|
||
T: 字典值
|
||
"""
|
||
if value := self._data.get(key):
|
||
return value.value if self.expire_time(key) else None
|
||
return None
|
||
|
||
def __setitem__(self, key: str, value: T) -> None:
|
||
"""设置字典项
|
||
|
||
参数:
|
||
key: 字典键
|
||
value: 字典值
|
||
"""
|
||
expire_time = time.time() + self.expire if self.expire > 0 else 0
|
||
self._data[key] = CacheData(value=value, expire_time=expire_time)
|
||
|
||
def __delitem__(self, key: str) -> None:
|
||
"""删除字典项
|
||
|
||
参数:
|
||
key: 字典键
|
||
"""
|
||
if key in self._data:
|
||
del self._data[key]
|
||
|
||
def __contains__(self, key: str) -> bool:
|
||
"""检查键是否存在
|
||
|
||
参数:
|
||
key: 字典键
|
||
|
||
返回:
|
||
bool: 是否存在
|
||
"""
|
||
if key not in self._data:
|
||
return False
|
||
|
||
# 检查是否过期
|
||
data = self._data[key]
|
||
if data.expire_time > 0 and data.expire_time < time.time():
|
||
del self._data[key]
|
||
return False
|
||
|
||
return True
|
||
|
||
def get(self, key: str, default: Any = None) -> T | None:
|
||
"""获取字典项，如果不存在返回默认值
|
||
|
||
参数:
|
||
key: 字典键
|
||
default: 默认值
|
||
|
||
返回:
|
||
Any: 字典值或默认值
|
||
"""
|
||
value = self[key]
|
||
return default if value is None else value
|
||
|
||
def set(self, key: str, value: Any, expire: int | None = None):
|
||
"""设置字典项
|
||
|
||
参数:
|
||
key: 字典键
|
||
value: 字典值
|
||
expire: 过期时间（秒），为None时使用默认值
|
||
"""
|
||
# 计算过期时间
|
||
expire_time = 0
|
||
if expire is not None and expire > 0:
|
||
expire_time = time.time() + expire
|
||
elif self.expire > 0:
|
||
expire_time = time.time() + self.expire
|
||
|
||
self._data[key] = CacheData(value=value, expire_time=expire_time)
|
||
|
||
def pop(self, key: str, default: Any = None) -> T:
|
||
"""删除并返回字典项
|
||
|
||
参数:
|
||
key: 字典键
|
||
default: 默认值
|
||
|
||
返回:
|
||
Any: 字典值或默认值
|
||
"""
|
||
if key not in self._data:
|
||
return default
|
||
|
||
data = self._data.pop(key)
|
||
|
||
# 检查是否过期
|
||
if data.expire_time > 0 and data.expire_time < time.time():
|
||
del self._data[key]
|
||
return default
|
||
|
||
return data.value
|
||
|
||
def clear(self) -> None:
|
||
"""清空字典"""
|
||
self._data.clear()
|
||
|
||
def keys(self) -> list[str]:
|
||
"""获取所有键
|
||
|
||
返回:
|
||
list[str]: 键列表
|
||
"""
|
||
# 清理过期的键
|
||
self._clean_expired()
|
||
return list(self._data.keys())
|
||
|
||
def values(self) -> list[Any]:
|
||
"""获取所有值
|
||
|
||
返回:
|
||
list[Any]: 值列表
|
||
"""
|
||
# 清理过期的键
|
||
self._clean_expired()
|
||
return [data.value for data in self._data.values()]
|
||
|
||
def items(self) -> list[tuple[str, T]]:
|
||
"""获取所有键值对
|
||
|
||
返回:
|
||
list[tuple[str, Any]]: 键值对列表
|
||
"""
|
||
# 清理过期的键
|
||
self._clean_expired()
|
||
return [(key, data.value) for key, data in self._data.items()]
|
||
|
||
def _clean_expired(self):
|
||
"""清理过期的键"""
|
||
now = time.time()
|
||
expired_keys = [
|
||
key
|
||
for key, data in self._data.items()
|
||
if data.expire_time > 0 and data.expire_time < now
|
||
]
|
||
for key in expired_keys:
|
||
del self._data[key]
|
||
|
||
def __len__(self) -> int:
|
||
"""获取字典长度
|
||
|
||
返回:
|
||
int: 字典长度
|
||
"""
|
||
# 清理过期的键
|
||
self._clean_expired()
|
||
return len(self._data)
|
||
|
||
def __str__(self) -> str:
|
||
"""字符串表示
|
||
|
||
返回:
|
||
str: 字符串表示
|
||
"""
|
||
# 清理过期的键
|
||
self._clean_expired()
|
||
return f"CacheDict({self.name}, {len(self._data)} items)"
|
||
|
||
|
||
class CacheList(Generic[T]):
|
||
"""缓存列表类，提供类似普通列表的接口，数据只存储在内存中"""
|
||
|
||
def __init__(self, name: str, expire: int = 0):
|
||
"""初始化缓存列表
|
||
|
||
参数:
|
||
name: 列表名称
|
||
expire: 过期时间（秒），默认为0表示永不过期
|
||
"""
|
||
self.name = name.upper()
|
||
self.expire = expire
|
||
self._data: list[CacheData[T]] = []
|
||
self._expire_time = 0
|
||
|
||
# 如果设置了过期时间，计算整个列表的过期时间
|
||
if self.expire > 0:
|
||
self._expire_time = time.time() + self.expire
|
||
|
||
def __getitem__(self, index: int) -> T:
|
||
"""获取列表项
|
||
|
||
参数:
|
||
index: 列表索引
|
||
|
||
返回:
|
||
T: 列表值
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
raise IndexError(f"列表索引 {index} 超出范围")
|
||
|
||
if 0 <= index < len(self._data):
|
||
return self._data[index].value
|
||
raise IndexError(f"列表索引 {index} 超出范围")
|
||
|
||
def __setitem__(self, index: int, value: T):
|
||
"""设置列表项
|
||
|
||
参数:
|
||
index: 列表索引
|
||
value: 列表值
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
|
||
# 确保索引有效
|
||
while len(self._data) <= index:
|
||
raise IndexError(f"列表索引 {index} 超出范围")
|
||
self._data[index] = CacheData(value=value)
|
||
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
|
||
def __delitem__(self, index: int):
|
||
"""删除列表项
|
||
|
||
参数:
|
||
index: 列表索引
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
raise IndexError(f"列表索引 {index} 超出范围")
|
||
|
||
if not 0 <= index < len(self._data):
|
||
raise IndexError(f"列表索引 {index} 超出范围")
|
||
del self._data[index]
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
|
||
def __len__(self) -> int:
|
||
"""获取列表长度
|
||
|
||
返回:
|
||
int: 列表长度
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
return len(self._data)
|
||
|
||
def append(self, value: T):
|
||
"""添加列表项
|
||
|
||
参数:
|
||
value: 列表值
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
|
||
self._data.append(CacheData(value=value))
|
||
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
|
||
def extend(self, values: list[T]):
|
||
"""扩展列表
|
||
|
||
参数:
|
||
values: 要添加的值列表
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
|
||
self._data.extend([CacheData(value=v) for v in values])
|
||
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
|
||
def insert(self, index: int, value: T):
|
||
"""插入列表项
|
||
|
||
参数:
|
||
index: 插入位置
|
||
value: 列表值
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
|
||
self._data.insert(index, CacheData(value=value))
|
||
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
|
||
def pop(self, index: int = -1) -> T:
|
||
"""删除并返回列表项
|
||
|
||
参数:
|
||
index: 列表索引，默认为最后一项
|
||
|
||
返回:
|
||
Any: 列表值
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
raise IndexError("从空列表中弹出")
|
||
|
||
if not self._data:
|
||
raise IndexError("从空列表中弹出")
|
||
|
||
item = self._data.pop(index)
|
||
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
|
||
return item.value
|
||
|
||
def remove(self, value: T):
|
||
"""删除第一个匹配的列表项
|
||
|
||
参数:
|
||
value: 要删除的值
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
raise ValueError(f"{value} 不在列表中")
|
||
|
||
# 查找匹配的项
|
||
for i, item in enumerate(self._data):
|
||
if item.value == value:
|
||
del self._data[i]
|
||
# 更新过期时间
|
||
self._update_expire_time()
|
||
return
|
||
|
||
raise ValueError(f"{value} 不在列表中")
|
||
|
||
def clear(self) -> None:
|
||
"""清空列表"""
|
||
self._data.clear()
|
||
# 重置过期时间
|
||
self._update_expire_time()
|
||
|
||
def index(self, value: T, start: int = 0, end: int | None = None) -> int:
|
||
"""查找值的索引
|
||
|
||
参数:
|
||
value: 要查找的值
|
||
start: 起始索引
|
||
end: 结束索引
|
||
|
||
返回:
|
||
int: 索引位置
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
raise ValueError(f"{value} 不在列表中")
|
||
|
||
end = end if end is not None else len(self._data)
|
||
|
||
for i in range(start, min(end, len(self._data))):
|
||
if self._data[i].value == value:
|
||
return i
|
||
|
||
raise ValueError(f"{value} 不在列表中")
|
||
|
||
def count(self, value: T) -> int:
|
||
"""计算值出现的次数
|
||
|
||
参数:
|
||
value: 要计数的值
|
||
|
||
返回:
|
||
int: 出现次数
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
return 0
|
||
|
||
# sourcery skip: simplify-constant-sum
|
||
return sum(1 for item in self._data if item.value == value)
|
||
|
||
def _is_expired(self) -> bool:
|
||
"""检查整个列表是否过期"""
|
||
return self._expire_time > 0 and self._expire_time < time.time()
|
||
|
||
def _update_expire_time(self):
|
||
"""更新过期时间"""
|
||
self._expire_time = time.time() + self.expire if self.expire > 0 else 0
|
||
|
||
def __str__(self) -> str:
|
||
"""字符串表示
|
||
|
||
返回:
|
||
str: 字符串表示
|
||
"""
|
||
# 检查整个列表是否过期
|
||
if self._is_expired():
|
||
self.clear()
|
||
return f"CacheList({self.name}, {len(self._data)} items)"
|