1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
royalnet/royalpack/events/discord_cv.py

91 lines
3.5 KiB
Python

import discord
from typing import *
from royalnet.serf.discord.discordserf import DiscordSerf
import royalnet.commands as rc
class DiscordCvEvent(rc.HeraldEvent):
name = "discord_cv"
async def run(self, guild_id: Optional[int] = None, **kwargs) -> dict:
if not isinstance(self.parent, DiscordSerf):
raise rc.UnsupportedError()
# noinspection PyTypeChecker
serf: DiscordSerf = self.parent
client: discord.Client = serf.client
if guild_id is None:
guilds: List[discord.Guild] = client.guilds
if len(guilds) == 0:
raise rc.ConfigurationError("Il bot non è in nessun Server.")
elif len(guilds) > 1:
raise rc.UserError("Non hai specificato di quale Server vuoi vedere le informazioni!")
else:
guild = guilds[0]
else:
guild: discord.Guild = client.get_guild(guild_id)
members: List[Union[discord.User, discord.Member]] = guild.members
results = []
for member in members:
data = {
"id": member.id,
"name": member.name,
"discriminator": member.discriminator,
"nick": member.nick,
"bot": member.bot,
"voice": {
"channel": {
"id": member.voice.channel.id,
"name": member.voice.channel.name,
"position": member.voice.channel.position,
"category": {
"id": member.voice.channel.category_id,
"name": member.voice.channel.category.name,
"position": member.voice.channel.category.position,
} if member.voice.channel.category is not None else {
"id": None,
"name": None,
"position": -1,
},
"bitrate": member.voice.channel.bitrate,
"user_limit": member.voice.channel.user_limit,
},
"server_mute": member.voice.mute,
"server_deaf": member.voice.deaf,
"self_mute": member.voice.self_mute,
"self_deaf": member.voice.self_deaf,
"video": member.voice.self_video,
"golive": member.voice.self_stream,
"afk": member.voice.afk,
} if member.voice is not None else None,
"status": {
"main": member.status.value,
"desktop": member.desktop_status.value,
"mobile": member.mobile_status.value,
"web": member.web_status.value,
},
"activities": [activity.to_dict() for activity in member.activities if activity is not None],
"roles": [{
"id": role.id,
"name": role.name,
"hoist": role.hoist,
"position": role.position,
"managed": role.managed,
"mentionable": role.mentionable,
} for role in member.roles]
}
results.append(data)
return {
"guild": {
"id": guild.id,
"name": guild.name,
"members": results,
}
}