From 549a87bed0dfb85c2c2461dc8c1da217de38cb9b Mon Sep 17 00:00:00 2001 From: Logan Cusano Date: Sun, 5 May 2024 18:28:00 -0400 Subject: [PATCH] Add presense endpoint and client function --- main.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index b5f435a..0d30a02 100644 --- a/main.py +++ b/main.py @@ -4,7 +4,7 @@ import os import platform import socketio import asyncio -from discord import Intents, opus +from discord import Intents, opus, Activity, ActivityType from discord.ext import commands from NoiseGatev2 import NoiseGate from debugger import setup_logger, running_dir @@ -46,6 +46,11 @@ def load_opus(): return False +async def set_discord_presense(presense): + # Set the presence of the bot (what it's listening to) + await client.change_presence(activity=Activity(type=ActivityType.listening, name=presense)) + + async def join_voice_channel(channel_id): global device_id, ng_threshold channel = client.get_channel(int(channel_id)) @@ -144,6 +149,12 @@ async def connect_error(): logger.error("Connection to WebSocket server failed") +@sio.event +async def set_system(data): + logger.info(f"Received command to set system name (presense): {data['system']}") + return await set_discord_presense(data['system']) + + @sio.event async def join_server(data): logger.info(f"Received command to join server: {data['channelId']}")