Kris
69b7139863
1. 移除多个文件中的空行、冗余导入
2. 修复文件末尾缺少换行符的问题
3. 新增并补全飞书多类工具API实现：
- 多维表格：更新、删除记录，列出视图
- 文档：更新、追加、删除块
- 云文档：上传、下载文件
- 群组：创建、添加成员、更新信息、创建公告
- 目录：重构用户部门缓存逻辑
4. 优化消息发送、回复、转发等API的错误处理和逻辑
5. 新增消息列表查询、已读状态查询等功能
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections import defaultdict
|
|
from collections.abc import Coroutine
|
|
from typing import Any
|
|
|
|
_SEQUENTIAL_TIMEOUT_S = 300
|
|
|
|
|
|
class FeishuSequentialQueue:
|
|
def __init__(self, timeout_s: float = _SEQUENTIAL_TIMEOUT_S):
|
|
self._queues: dict[str, asyncio.Queue] = {}
|
|
self._locks: dict[str, asyncio.Lock] = defaultdict(asyncio.Lock)
|
|
self._timeout_s = timeout_s
|
|
|
|
async def acquire(self, key: str) -> None:
|
|
lock = self._locks[key]
|
|
try:
|
|
await asyncio.wait_for(lock.acquire(), timeout=self._timeout_s)
|
|
except TimeoutError:
|
|
raise RuntimeError(f"[Sequential] Timeout acquiring lock for key '{key}'")
|
|
|
|
def release(self, key: str) -> None:
|
|
lock = self._locks.get(key)
|
|
if lock and lock.locked():
|
|
lock.release()
|
|
|
|
async def run_sequential(self, key: str, coro: Coroutine[Any, Any, Any]) -> None:
|
|
try:
|
|
await asyncio.wait_for(self.acquire(key), timeout=self._timeout_s)
|
|
try:
|
|
await coro
|
|
finally:
|
|
self.release(key)
|
|
except TimeoutError:
|
|
raise RuntimeError(f"[Sequential] Timeout waiting for key '{key}'")
|
|
|
|
def remove(self, key: str) -> None:
|
|
lock = self._locks.pop(key, None)
|
|
if lock and lock.locked():
|
|
lock.release()
|
|
|
|
def clear(self) -> None:
|
|
for lock in self._locks.values():
|
|
if lock.locked():
|
|
lock.release()
|
|
self._locks.clear()
|