97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from yuxi.channels.models import DeliveryResult
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
try:
|
||
|
|
import lark_oapi
|
||
|
|
|
||
|
|
HAS_LARK_SDK = True
|
||
|
|
except ImportError:
|
||
|
|
HAS_LARK_SDK = False
|
||
|
|
lark_oapi = None # type: ignore
|
||
|
|
|
||
|
|
|
||
|
|
async def create_pin(client: Any, message_id: str) -> DeliveryResult:
|
||
|
|
if not HAS_LARK_SDK:
|
||
|
|
return DeliveryResult(success=False, error="lark-oapi SDK not available")
|
||
|
|
|
||
|
|
try:
|
||
|
|
request = (
|
||
|
|
lark_oapi.api.im.v1.CreatePinRequest.builder()
|
||
|
|
.request_body(lark_oapi.api.im.v1.CreatePinRequestBody.builder().message_id(message_id).build())
|
||
|
|
.build()
|
||
|
|
)
|
||
|
|
|
||
|
|
resp = await client.im.v1.pin.create(request)
|
||
|
|
if resp.success():
|
||
|
|
return DeliveryResult(success=True, message_id=message_id)
|
||
|
|
return DeliveryResult(success=False, error=resp.msg)
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"[Feishu] create_pin failed: {e}")
|
||
|
|
return DeliveryResult(success=False, error=str(e))
|
||
|
|
|
||
|
|
|
||
|
|
async def remove_pin(client: Any, message_id: str) -> DeliveryResult:
|
||
|
|
if not HAS_LARK_SDK:
|
||
|
|
return DeliveryResult(success=False, error="lark-oapi SDK not available")
|
||
|
|
|
||
|
|
try:
|
||
|
|
request = lark_oapi.api.im.v1.DeletePinRequest.builder().message_id(message_id).build()
|
||
|
|
|
||
|
|
resp = await client.im.v1.pin.delete(request)
|
||
|
|
if resp.success():
|
||
|
|
return DeliveryResult(success=True)
|
||
|
|
return DeliveryResult(success=False, error=resp.msg)
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"[Feishu] remove_pin failed: {e}")
|
||
|
|
return DeliveryResult(success=False, error=str(e))
|
||
|
|
|
||
|
|
|
||
|
|
async def list_pins(
|
||
|
|
client: Any,
|
||
|
|
chat_id: str,
|
||
|
|
*,
|
||
|
|
start_time: str | None = None,
|
||
|
|
end_time: str | None = None,
|
||
|
|
page_size: int = 20,
|
||
|
|
page_token: str | None = None,
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
if not HAS_LARK_SDK:
|
||
|
|
return {"pins": [], "total": 0}
|
||
|
|
|
||
|
|
try:
|
||
|
|
request = lark_oapi.api.im.v1.ListPinRequest.builder().chat_id(chat_id).page_size(page_size).build()
|
||
|
|
if page_token:
|
||
|
|
request.page_token(page_token)
|
||
|
|
if start_time:
|
||
|
|
request.start_time(start_time)
|
||
|
|
if end_time:
|
||
|
|
request.end_time(end_time)
|
||
|
|
|
||
|
|
resp = await client.im.v1.pin.list(request)
|
||
|
|
if resp.success():
|
||
|
|
items = resp.data.get("items", [])
|
||
|
|
return {
|
||
|
|
"pins": [
|
||
|
|
{
|
||
|
|
"message_id": item.get("message_id", ""),
|
||
|
|
"chat_id": item.get("chat_id", ""),
|
||
|
|
"create_time": item.get("create_time", ""),
|
||
|
|
"operator_id": item.get("operator_id", ""),
|
||
|
|
}
|
||
|
|
for item in items
|
||
|
|
],
|
||
|
|
"total": resp.data.get("total", 0),
|
||
|
|
"page_token": resp.data.get("page_token", ""),
|
||
|
|
"has_more": resp.data.get("has_more", False),
|
||
|
|
}
|
||
|
|
return {"pins": [], "total": 0, "error": resp.msg}
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"[Feishu] list_pins failed: {e}")
|
||
|
|
return {"pins": [], "total": 0, "error": str(e)}
|