From a1a29aee950be3be326f982e0b8161da8ef53800 Mon Sep 17 00:00:00 2001 From: Lemon4ksan Date: Mon, 20 Jan 2025 20:36:56 +0300 Subject: [PATCH] feat: Add bot settings commands. --- MusicBot/cogs/general.py | 9 ++- MusicBot/cogs/settings.py | 114 +++++++++++++++++++++++++++++++++++++ MusicBot/database/base.py | 50 +++++++++++++++- MusicBot/database/guild.py | 14 ++++- MusicBot/main.py | 3 +- 5 files changed, 183 insertions(+), 7 deletions(-) create mode 100644 MusicBot/cogs/settings.py diff --git a/MusicBot/cogs/general.py b/MusicBot/cogs/general.py index 6fd1f9a..f2f8c5d 100644 --- a/MusicBot/cogs/general.py +++ b/MusicBot/cogs/general.py @@ -56,6 +56,7 @@ class General(Cog): `help` `like` `queue` + `settings` `track` `voice` """ @@ -76,11 +77,17 @@ class General(Cog): embed.description += ("Вывести список всех команд.\n```/help```\n" "Получить информацию о конкретной команде.\n```/help <команда>```") elif command == 'like': - embed.description += "Добавить трек в плейлист «Мне нравится».\n```/like```" + embed.description += "Добавить трек в плейлист «Мне нравится». Пользовательские треки из этого плейлиста игнорируются.\n```/like```" elif command == 'queue': embed.description += ("Получить очередь треков. По 15 элементов на страницу.\n```/queue get```\n" "Очистить очередь треков и историю прослушивания. Доступно только если вы единственный в голосовом канале" "или имеете разрешение управления каналом.\n```/queue clear```\n") + elif command == 'settings': + embed.description += ("Получить текущие настройки.\n```/settings show```\n" + "Разрешить или запретить воспроизведение Explicit треков.\n```/settings explicit```\n" + "Разрешить или запретить создание меню проигрывателя, даже если в канале больше одного человека.\n```/settings menu```\n" + "Разрешить или запретить голосование.\n```/settings vote <тип голосования>```\n" + "`Примечание`: Только пользователи с разрешением управления каналом могут менять настройки.") elif command == 'track': embed.description += ("`Примечание`: Если вы один в голосовом канале или имеете разрешение управления каналом, голосование не начинается.\n\n" "Переключиться на следующий трек в очереди. \n```/track next```\n" diff --git a/MusicBot/cogs/settings.py b/MusicBot/cogs/settings.py new file mode 100644 index 0000000..54832b0 --- /dev/null +++ b/MusicBot/cogs/settings.py @@ -0,0 +1,114 @@ +from typing import Literal, cast + +import discord +from discord.ext.commands import Cog + +from MusicBot.database import BaseUsersDatabase, BaseGuildsDatabase + +def setup(bot): + bot.add_cog(Settings(bot)) + +class Settings(Cog): + + settings = discord.SlashCommandGroup("settings", "Команды для изменения настроек бота.", guild_ids=[1247100229535141899]) + + def __init__(self, bot: discord.Bot): + self.db = BaseGuildsDatabase() + self.users_db = BaseUsersDatabase() + self.bot = bot + + @settings.command(name="show", description="Показать текущие настройки бота.") + async def show(self, ctx: discord.ApplicationContext) -> None: + guild = self.db.get_guild(ctx.guild.id) + embed = discord.Embed(title="Настройки бота", color=0xfed42b) + + explicit = "✅ - Разрешены" if guild['allow_explicit'] else "❌ - Запрещены" + menu = "✅ - Всегда доступно" if guild['always_allow_menu'] else "❌ - Если в канале 1 пользователь." + + vote = "✅ - Переключение" if guild['vote_next_track'] else "❌ - Переключение" + vote += "\n✅ - Добавление треков" if guild['vote_add_track'] else "\n❌ - Добавление треков" + vote += "\n✅ - Добавление альбомов" if guild['vote_add_album'] else "\n❌ - Добавление альбомов" + vote += "\n✅ - Добавление артистов" if guild['vote_add_artist'] else "\n❌ - Добавление артистов" + vote += "\n✅ - Добавление плейлистов" if guild['vote_add_playlist'] else "\n❌ - Добавление плейлистов" + + embed.add_field(name="__Explicit треки__", value=explicit, inline=False) + embed.add_field(name="__Проигрыватель__", value=menu, inline=False) + embed.add_field(name="__Голосование__", value=vote, inline=False) + + await ctx.respond(embed=embed, ephemeral=True) + + @settings.command(name="explicit", description="Разрешить или запретить воспроизведение Explicit треков.") + async def explicit(self, ctx: discord.ApplicationContext) -> None: + member = cast(discord.Member, ctx.author) + if not member.guild_permissions.manage_channels: + await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True) + return + + guild = self.db.get_guild(ctx.guild.id) + self.db.update(ctx.guild.id, {'allow_explicit': not guild['allow_explicit']}) + await ctx.respond(f"Треки с содержанием не для детей теперь {'разрешены' if not guild['allow_explicit'] else 'запрещены'}.", delete_after=15, ephemeral=True) + + @settings.command(name="menu", description="Разрешить или запретить создание меню проигрывателя, даже если в канале больше одного человека.") + async def menu(self, ctx: discord.ApplicationContext) -> None: + member = cast(discord.Member, ctx.author) + if not member.guild_permissions.manage_channels: + await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True) + return + + guild = self.db.get_guild(ctx.guild.id) + self.db.update(ctx.guild.id, {'always_allow_menu': not guild['always_allow_menu']}) + await ctx.respond(f"Меню проигрывателя теперь {'можно' if not guild['always_allow_menu'] else 'нельзя'} создавать в каналах с несколькими людьми.", delete_after=15, ephemeral=True) + + @settings.command(name="vote", description="Настроить голосование.") + @discord.option( + "vote_type", + description="Тип голосования.", + type=discord.SlashCommandOptionType.string, + choices=['+Всё', '-Всё', 'Переключение', '+Трек', '+Альбом', '+Плейлист'], + default='Всё' + ) + async def vote(self, ctx: discord.ApplicationContext, vote_type: Literal['+Всё', '-Всё', 'Переключение', 'Трек', 'Альбом', 'Плейлист']) -> None: + member = cast(discord.Member, ctx.author) + if not member.guild_permissions.manage_channels: + await ctx.respond("❌ У вас нет прав для выполнения этой команды.", delete_after=15, ephemeral=True) + return + + guild = self.db.get_guild(ctx.guild.id) + + if vote_type == '-Всё': + self.db.update(ctx.guild.id, { + 'vote_next_track': False, + 'vote_add_track': False, + 'vote_add_album': False, + 'vote_add_artist': False, + 'vote_add_playlist': False + } + ) + response_message = "Голосование выключено." + elif vote_type == '+Всё': + self.db.update(ctx.guild.id, { + 'vote_next_track': True, + 'vote_add_track': True, + 'vote_add_album': True, + 'vote_add_artist': True, + 'vote_add_playlist': True + } + ) + response_message = "Голосование включено." + elif vote_type == 'Переключение': + self.db.update(ctx.guild.id, {'vote_next_track': not guild['vote_next_track']}) + response_message = "Голосование за переключение трека " + ("включено." if guild['vote_next_track'] else "выключено.") + elif vote_type == 'Трек': + self.db.update(ctx.guild.id, {'vote_add_track': not guild['vote_add_track']}) + response_message = "Голосование за добавление трека " + ("включено." if guild['vote_add_track'] else "выключено.") + elif vote_type == 'Альбом': + self.db.update(ctx.guild.id, {'vote_add_album': not guild['vote_add_album']}) + response_message = "Голосование за добавление альбома " + ("включено." if guild['vote_add_album'] else "выключено.") + elif vote_type == 'Артист': + self.db.update(ctx.guild.id, {'vote_add_artist': not guild['vote_add_artist']}) + response_message = "Голосование за добавление артиста " + ("включено." if guild['vote_add_artist'] else "выключено.") + elif vote_type == 'Плейлист': + self.db.update(ctx.guild.id, {'vote_add_playlist': not guild['vote_add_playlist']}) + response_message = "Голосование за добавление плейлиста " + ("включено." if guild['vote_add_playlist'] else "выключено.") + + await ctx.respond(response_message, delete_after=15, ephemeral=True) diff --git a/MusicBot/database/base.py b/MusicBot/database/base.py index e416138..1730de8 100644 --- a/MusicBot/database/base.py +++ b/MusicBot/database/base.py @@ -52,7 +52,20 @@ class BaseUsersDatabase: if not user: self.create_record(uid) user = users.find_one({'_id': uid}) - return cast(ExplicitUser, user) + user = cast(ExplicitUser, user) + existing_fields = user.keys() + fields: User = User( + ym_token=None, + playlists=[], + playlists_page=0, + queue_page=0 + ) + for field, default_value in fields.items(): + if field not in existing_fields and field != '_id': + user[field] = default_value + users.update_one({'_id': uid}, {"$set": {field: default_value}}) + + return user def get_ym_token(self, uid: int) -> str | None: user = users.find_one({'_id': uid}) @@ -78,7 +91,12 @@ class BaseGuildsDatabase: is_stopped=True, allow_explicit=True, always_allow_menu=False, - disable_vote=False, + vote_add=True, + vote_next_track=True, + vote_add_track=True, + vote_add_album=True, + vote_add_artist=True, + vote_add_playlist=True, shuffle=False, repeat=False )) @@ -106,4 +124,30 @@ class BaseGuildsDatabase: if not guild: self.create_record(gid) guild = guilds.find_one({'_id': gid}) - return cast(ExplicitGuild, guild) + + guild = cast(ExplicitGuild, guild) + existing_fields = guild.keys() + fields = Guild( + next_tracks=[], + previous_tracks=[], + current_track=None, + current_player=None, + is_stopped=True, + allow_explicit=True, + always_allow_menu=False, + vote_add=True, + vote_next_track=True, + vote_add_track=True, + vote_add_album=True, + vote_add_artist=True, + vote_add_playlist=True, + shuffle=False, + repeat=False + ) + for field, default_value in fields.items(): + if field not in existing_fields and field != '_id': + guild[field] = default_value + guilds.update_one({'_id': gid}, {"$set": {field: default_value}}) + + return guild + diff --git a/MusicBot/database/guild.py b/MusicBot/database/guild.py index f09b8a8..3ae9eb4 100644 --- a/MusicBot/database/guild.py +++ b/MusicBot/database/guild.py @@ -8,7 +8,12 @@ class Guild(TypedDict, total=False): is_stopped: bool allow_explicit: bool always_allow_menu: bool - disable_vote: bool + vote_add: bool + vote_next_track: bool + vote_add_track: bool + vote_add_album: bool + vote_add_artist: bool + vote_add_playlist: bool shuffle: bool repeat: bool @@ -21,6 +26,11 @@ class ExplicitGuild(TypedDict): is_stopped: bool # Prevents the `after` callback of play_track allow_explicit: bool always_allow_menu: bool - disable_vote: bool + vote_add: bool + vote_next_track: bool + vote_add_track: bool + vote_add_album: bool + vote_add_artist: bool + vote_add_playlist: bool shuffle: bool repeat: bool \ No newline at end of file diff --git a/MusicBot/main.py b/MusicBot/main.py index 13dd6dd..1be1566 100644 --- a/MusicBot/main.py +++ b/MusicBot/main.py @@ -16,7 +16,8 @@ bot = Bot(intents=intents) cogs_list = [ 'general', - 'voice' + 'voice', + 'settings' ] @bot.event