49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from pypresence import Presence
|
|
from settings import settings
|
|
from discord.models import DiscordRPCUpdatePayload
|
|
import logging
|
|
|
|
|
|
class DiscordRPC:
|
|
"""
|
|
Client for interacting with Discord Rich Presence (RPC).
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""
|
|
Initializes the Discord RPC client and connects to Discord via IPC.
|
|
"""
|
|
self.logger = logging.getLogger('DiscordRPC')
|
|
|
|
self.logger.info("Connecting to Discord RPC...")
|
|
self.rpc = Presence(settings.discord_app_id)
|
|
self.rpc.connect()
|
|
self.logger.info("Connected to Discord RPC.")
|
|
|
|
def update(self, payload: DiscordRPCUpdatePayload):
|
|
"""
|
|
Updates the Discord RPC presence with the provided payload.
|
|
|
|
Args:
|
|
payload (DiscordRPCUpdatePayload): The payload containing presence information.
|
|
"""
|
|
self.logger.info("Updating Discord RPC presence...")
|
|
self.rpc.update(
|
|
activity_type=payload.activity_type,
|
|
details=payload.title,
|
|
state=payload.subtitle,
|
|
large_image=payload.image_url,
|
|
large_text=payload.details,
|
|
start=payload.start,
|
|
end=payload.end
|
|
)
|
|
self.logger.info("Discord RPC presence updated.")
|
|
|
|
def clear(self):
|
|
"""
|
|
Clears the Discord RPC presence.
|
|
"""
|
|
self.logger.info("Clearing Discord RPC presence...")
|
|
self.rpc.clear()
|
|
self.logger.info("Discord RPC presence cleared.")
|