68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
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 list_peers(client: Any, page_size: int = 50) -> list[dict]:
|
||
|
|
if not HAS_LARK_SDK or not client:
|
||
|
|
return []
|
||
|
|
|
||
|
|
try:
|
||
|
|
request = lark_oapi.api.contact.v3.ListUserRequest.builder().page_size(page_size).build()
|
||
|
|
resp = await client.contact.v3.user.list(request)
|
||
|
|
if resp.success():
|
||
|
|
items = resp.data.get("items", [])
|
||
|
|
return [
|
||
|
|
{
|
||
|
|
"open_id": item.get("open_id", ""),
|
||
|
|
"user_id": item.get("user_id", ""),
|
||
|
|
"name": item.get("name", ""),
|
||
|
|
"nickname": item.get("nickname", ""),
|
||
|
|
"email": item.get("email", ""),
|
||
|
|
"department_ids": item.get("department_ids", []),
|
||
|
|
"avatar": item.get("avatar", {}),
|
||
|
|
}
|
||
|
|
for item in items
|
||
|
|
]
|
||
|
|
return []
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"[FeishuDir] list_peers failed: {e}")
|
||
|
|
return []
|
||
|
|
|
||
|
|
|
||
|
|
async def list_groups(client: Any, page_size: int = 50) -> list[dict]:
|
||
|
|
if not HAS_LARK_SDK or not client:
|
||
|
|
return []
|
||
|
|
|
||
|
|
try:
|
||
|
|
request = lark_oapi.api.im.v1.ListChatRequest.builder().page_size(page_size).build()
|
||
|
|
resp = await client.im.v1.chat.list(request)
|
||
|
|
if resp.success():
|
||
|
|
items = resp.data.get("items", [])
|
||
|
|
return [
|
||
|
|
{
|
||
|
|
"chat_id": item.get("chat_id", ""),
|
||
|
|
"name": item.get("name", ""),
|
||
|
|
"description": item.get("description", ""),
|
||
|
|
"chat_type": item.get("chat_type", ""),
|
||
|
|
"member_count": item.get("member_count", 0),
|
||
|
|
"owner_id": item.get("owner_id", ""),
|
||
|
|
}
|
||
|
|
for item in items
|
||
|
|
]
|
||
|
|
return []
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"[FeishuDir] list_groups failed: {e}")
|
||
|
|
return []
|