92 lines
2.4 KiB
Python
92 lines
2.4 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
import logging
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
|
||
|
|
from yuxi.channel.extensions.ringcentral.sdk import AsyncRingCentralClient
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
async def upload_file(
|
||
|
|
client: AsyncRingCentralClient,
|
||
|
|
group_id: str,
|
||
|
|
file_data: bytes,
|
||
|
|
filename: str,
|
||
|
|
content_type: str = "application/octet-stream",
|
||
|
|
) -> dict:
|
||
|
|
builder = client.platform.create_multipart_builder()
|
||
|
|
builder.set_body({"groupId": group_id})
|
||
|
|
builder.add_file(None, content=file_data, content_type=content_type, file_name=filename)
|
||
|
|
|
||
|
|
def _send():
|
||
|
|
request = builder.request("/restapi/v1.0/glip/files")
|
||
|
|
return client.platform.send_request(request)
|
||
|
|
|
||
|
|
resp = await asyncio.to_thread(_send)
|
||
|
|
return resp.json()
|
||
|
|
|
||
|
|
|
||
|
|
async def download_file_content(
|
||
|
|
client: AsyncRingCentralClient,
|
||
|
|
file_id: str,
|
||
|
|
) -> bytes:
|
||
|
|
def _download():
|
||
|
|
resp = client.platform.get(f"/restapi/v1.0/glip/files/{file_id}/content")
|
||
|
|
return resp.response.content
|
||
|
|
|
||
|
|
return await asyncio.to_thread(_download)
|
||
|
|
|
||
|
|
|
||
|
|
async def send_media_message(
|
||
|
|
client: AsyncRingCentralClient,
|
||
|
|
group_id: str,
|
||
|
|
text: str,
|
||
|
|
file_info: dict,
|
||
|
|
) -> dict:
|
||
|
|
return await client.post(
|
||
|
|
f"/restapi/v1.0/glip/chats/{group_id}/posts",
|
||
|
|
body={
|
||
|
|
"text": text,
|
||
|
|
"attachments": [
|
||
|
|
{
|
||
|
|
"type": "File",
|
||
|
|
"id": file_info.get("id"),
|
||
|
|
"name": file_info.get("name", ""),
|
||
|
|
"contentUri": file_info.get("contentUri", ""),
|
||
|
|
}
|
||
|
|
],
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
async def send_media_from_url(
|
||
|
|
client: AsyncRingCentralClient,
|
||
|
|
group_id: str,
|
||
|
|
media_url: str,
|
||
|
|
media_type: str,
|
||
|
|
) -> None:
|
||
|
|
async with httpx.AsyncClient(timeout=httpx.Timeout(60.0)) as http:
|
||
|
|
resp = await http.get(media_url)
|
||
|
|
resp.raise_for_status()
|
||
|
|
content = resp.content
|
||
|
|
|
||
|
|
filename = _extract_filename(media_url, media_type)
|
||
|
|
upload_result = await upload_file(client, group_id, content, filename, media_type)
|
||
|
|
await send_media_message(client, group_id, "", upload_result)
|
||
|
|
|
||
|
|
|
||
|
|
def _extract_filename(url: str, content_type: str) -> str:
|
||
|
|
ext_map = {
|
||
|
|
"image/jpeg": ".jpg",
|
||
|
|
"image/png": ".png",
|
||
|
|
"image/gif": ".gif",
|
||
|
|
"image/webp": ".webp",
|
||
|
|
"video/mp4": ".mp4",
|
||
|
|
"application/pdf": ".pdf",
|
||
|
|
}
|
||
|
|
ext = ext_map.get(content_type, ".bin")
|
||
|
|
return f"attachment{ext}"
|