46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
"""Discord REST API 클라이언트."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
|
|
import httpx
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
DISCORD_API_BASE = "https://discord.com/api/v10"
|
|
|
|
|
|
class DiscordClient:
|
|
def __init__(self, token: str):
|
|
self.token = token
|
|
self._client = httpx.AsyncClient(
|
|
base_url=DISCORD_API_BASE,
|
|
headers={"Authorization": f"Bot {self.token}"},
|
|
timeout=15.0,
|
|
)
|
|
|
|
async def send_message(self, channel_id: str, content: str) -> dict:
|
|
resp = await self._client.post(
|
|
f"/channels/{channel_id}/messages",
|
|
json={"content": content},
|
|
)
|
|
resp.raise_for_status()
|
|
return resp.json()
|
|
|
|
async def send_thread_reply(self, channel_id: str, thread_id: str, content: str) -> dict:
|
|
return await self.send_message(channel_id=thread_id, content=content)
|
|
|
|
async def close(self):
|
|
await self._client.aclose()
|
|
|
|
|
|
_client: DiscordClient | None = None
|
|
|
|
|
|
def get_discord_client() -> DiscordClient:
|
|
global _client
|
|
if _client is None:
|
|
_client = DiscordClient(token=os.environ.get("DISCORD_TOKEN", ""))
|
|
return _client
|