Migrating to discord.py 2

TiDurak · TiDurak · commit 34e3e2b4ee8c · 2023-03-25T22:25:22.000+01:00
diff --git a/Procfile b/Procfile
diff --git a/cogs/converters.py b/cogs/converters.py
@@ -41,5 +41,5 @@ async def decode_binary(self, ctx, *, arg):
         embed = discord.Embed(color = 0xff8080, title = 'Binary Decoder', description = json_data['text'])
         await ctx.send(embed = embed)
 
-def setup(bot):
-        bot.add_cog(Convert(bot))
+async def setup(bot):
+        await bot.add_cog(Convert(bot))
diff --git a/cogs/games.py b/cogs/games.py
@@ -1,47 +1,37 @@
 import random
 import discord
 from discord.ext import commands
-from discord_components import DiscordComponents, Button, ButtonStyle
 
 class Games(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
-        
+
+    # noinspection PyUnresolvedReferences
+    class JankenButtons(discord.ui.View):
+        def __init__(self):
+            super().__init__()
+
+        @discord.ui.button(label="Камень", emoji="🗿", style=discord.ButtonStyle.blurple)
+        async def rock_button(self, interaction: discord.Interaction, button: discord.ui.Button):
+            await interaction.response.send_message("Камень")
+
+        @discord.ui.button(label="Ножницы", emoji="✂️", style=discord.ButtonStyle.red)
+        async def scissors_button(self, interaction: discord.Interaction, button: discord.ui.Button):
+            await interaction.response.send_message("Ножницы")
+
+        @discord.ui.button(label="Бумага", emoji="📄", style=discord.ButtonStyle.gray)
+        async def paper_button(self, interaction: discord.Interaction, button: discord.ui.Button):
+            await interaction.response.send_message("Бумага")
+
     @commands.command(aliases=['rps', 'rockpaperscissors'])
     async def janken(self, ctx):
         description = 'Сыграй со мной в камень ножницы бумагу! выбери один из вариантов ниже:'
         embed = discord.Embed(color = 0xffcd4c, title = f'{ctx.message.author}: Камень Ножницы Бумага', description = description)
-        gamebar = await ctx.send(
-            embed = embed,
-            components = [
-                [
-                    Button(style = ButtonStyle.blue, label = 'Камень', emoji = '🗿'),
-                    Button(style = ButtonStyle.red, label = 'Ножницы', emoji = '✂️'),
-                    Button(style = ButtonStyle.gray, label = 'Бумага', emoji = '📄'),
-                ]
-        ])
+        await ctx.send(embed = embed)
         answers = ['Камень', 'Ножницы', 'Бумага']
         choice = random.choice(answers)
 
-        responce = await self.bot.wait_for('button_click', check = lambda message: message.author == ctx.author)
-           
-        async def victory():
-            await gamebar.edit(embed=discord.Embed(color = embed.color, title = embed.title, description = f'{embed.description} \n Ты выбрал `{responce.component.label}`, а я выбрал `{choice}` \n Победа!!!'), components=[])
-             
-        if responce.component.label == choice:
-            await gamebar.edit(embed=discord.Embed(color = embed.color, title = embed.title, description = f'{embed.description} \n Ты выбрал `{responce.component.label}`, а я выбрал `{choice}` \n Ничья!'), components=[])
-
-        elif responce.component.label == 'Камень' and choice == 'Ножницы':
-            await victory()
-
-        elif responce.component.label == 'Ножницы' and choice == 'Бумага':
-            await victory()
-
-        elif responce.component.label == 'Бумага' and choice == 'Камень':
-            await victory()
-            
-        else:
-            await gamebar.edit(embed=discord.Embed(color = embed.color, title = embed.title, description = f'{embed.description} \n Ты выбрал `{responce.component.label}`, а я выбрал `{choice}` \n Проигрыш :('), components=[])
+        #await gamebar.edit(embed=discord.Embed(color = embed.color, title = embed.title, description = f'{embed.description} \n Ты выбрал `{responce.component.label}`, а я выбрал `{choice}` \n Проигрыш :('), components=[])
 
 
     @commands.command()
@@ -77,5 +67,5 @@ async def slots(self, ctx):
         embed.set_footer(text = footer, icon_url = "https://i.imgur.com/uZIlRnK.png")
         await ctx.send(embed = embed)
 
-def setup(bot):
-    bot.add_cog(Games(bot))
+async def setup(bot):
+    await bot.add_cog(Games(bot))
diff --git a/cogs/help.py b/cogs/help.py
@@ -1,6 +1,5 @@
 import discord
 from discord.ext import commands
-from discord_components import DiscordComponents, Button, ButtonStyle
 from config import settings
 
 class Help(commands.Cog):
@@ -139,18 +138,20 @@ async def server_info(self, ctx):
         embed = discord.Embed(color = 0xffcd4c , title = 'server_info', description = help_text)
         await ctx.send(embed = embed)
 
+    class ISOCodes(discord.ui.View):
+        def __init__(self):
+            super().__init__()
+            self.__url = "https://snipp.ru/handbk/iso-639-1"
+            self.add_item(discord.ui.Button(label='Коды ISO 639-1', url=self.__url))
+
     @help.command()
     async def translate(self, ctx):
-        help_text = (f'```{settings.get("prefix")}translate <язык> <текст>```\n'
-                      'Переводит ваш текст, язык указывается по стандарту ISO 639-1\n'
-                      'Полный список наименований будет доступен после нажатия на кнопку ниже')
+        help_text = (f"```{settings.get('prefix')}translate <язык> <текст>```\n"
+                      "Переводит ваш текст, язык указывается по стандарту ISO 639-1\n"
+                      "Полный список наименований будет доступен после нажатия на кнопку ниже")
         embed = discord.Embed(color = 0xffcd4c , title = 'translate', description = help_text)
-        await ctx.send(
-        embed = embed,
-        components = [
-        Button(style = ButtonStyle.URL, url = 'https://snipp.ru/handbk/iso-639-1', label='Коды ISO 639-1')
-        ])
-        
+        await ctx.send(embed=embed, view=self.ISOCodes())
+
     @help.command()
     async def poll(self, ctx):
         help_text = (f'```{settings.get("prefix")}poll "вопрос" "вариант 1" "вариант 2"```\n'
@@ -160,10 +161,9 @@ async def poll(self, ctx):
                       '```{settings.get("prefix")}poll "Какие чипсы вы предпочитаете" "Lais" "Prongls" "2 корочки"```\n'
                       'Плохой пример:\n'
                       '```{settings.get("prefix")}poll Какие чипсы вы предпочитаете Lais Prongls 2 корочки```\n'
-                      'В плохом примере нету кавычек, соответственно, за вопрос будет считыватся только `Какие`, остальное будет вариантами ответа\n'
-                      '**Для вывода результата плохого примера, нажмите на кнопку ниже**')
+                      'В плохом примере нету кавычек, соответственно, за вопрос будет считыватся только `Какие`, остальное будет вариантами ответа\n')
         embed = discord.Embed(color = 0xffcd4c , title = 'poll', description = help_text)
-        await ctx.send(embed = embed, components = [[Button(style = ButtonStyle.blue, label = 'Результат Плохого Примера')]])
+        await ctx.send(embed=embed)
                 
     @help.command()
     async def echo(self, ctx):
@@ -217,5 +217,5 @@ async def janken(self, ctx):
         embed = discord.Embed(color = 0xffcd4c , title = 'Камень-Ножницы-Бумага', description = help_text)
         await ctx.send(embed = embed)
 
-def setup(bot):
-        bot.add_cog(Help(bot))
+async def setup(bot):
+    await bot.add_cog(Help(bot))
diff --git a/cogs/information.py b/cogs/information.py
@@ -42,5 +42,5 @@ async def server_info(self, ctx):
         embed.set_footer(text = f"Запросил {ctx.author}", icon_url = ctx.author.avatar_url)
         await ctx.send(embed = embed)
 
-def setup(bot):
-    bot.add_cog(Information(bot))
+async def setup(bot):
+    await bot.add_cog(Information(bot))
diff --git a/cogs/listeners.py b/cogs/listeners.py
@@ -37,6 +37,6 @@ async def on_ready(self):
             await self.bot.change_presence(status=discord.Status.online, activity=discord.Game(f"{settings.get('prefix')}help"))
             await asyncio.sleep(5)
 
-def setup(bot):
-    bot.add_cog(ErrorListener(bot))
-    bot.add_cog(OnReady(bot))
+async def setup(bot):
+    await bot.add_cog(ErrorListener(bot))
+    await bot.add_cog(OnReady(bot))
diff --git a/cogs/moderation.py b/cogs/moderation.py
@@ -14,7 +14,7 @@ async def clear(self, ctx, arg1):
         amount = int(arg1)
         await ctx.message.delete()
         await ctx.channel.purge(limit = amount)
-        await ctx.send(f'{self.bot.get_emoji(settings["emojis"]["squid_cleaning"])} Очищено {amount} сообщений! Запрос на очистку от {ctx.author.mention}')
+        await ctx.send(f'{self.bot.get_emoji(settings["emojis"]["squid_cleaning"])} {ctx.author.mention}: Очищено {amount} сообщений!')
 
     @has_permissions(kick_members = True)
     @commands.command()
@@ -57,5 +57,5 @@ async def idclear_error(self, ctx, error):
         elif isinstance(error, CommandInvokeError):
             await ctx.send('❌ У бота нету привилегий управления сообщениями! Пожалуйста, удалите бота из сервера, и добавьте его снова по следующей ссылке: https://discord.com/api/oauth2/authorize?client_id=699912361481470032&permissions=8&scope=bot')
 
-def setup(bot):
-    bot.add_cog(Moderation(bot))
+async def setup(bot):
+    await bot.add_cog(Moderation(bot))
diff --git a/cogs/music.py b/cogs/music.py
@@ -2,7 +2,6 @@
 import datetime
 import discord
 from discord.ext import commands
-from discord_components import DiscordComponents, Button, ButtonStyle
 from youtube_dl import YoutubeDL
 from config import settings
 
@@ -55,9 +54,9 @@ def get_by_id(self, id):
 
 
 class Music(commands.Cog):
-    def __init__(self, bot):
+    def __init__(self, bot, intents):
         self.bot = bot
-        self.__client = discord.Client()
+        self.__client = discord.Client(intents=intents)
         self.__queue = Queue()
 
     async def __connect(self, ctx):
@@ -87,6 +86,41 @@ def __get_url(self, extracted):
             url = extracted['formats'][0]['url']
             return url
 
+    # noinspection PyUnresolvedReferences
+    class PlayerButtons(discord.ui.View):
+        def __init__(self, voice_chat, context, leave, stop, pause, resume, skip):
+            super().__init__()
+            self.__vc = voice_chat
+            self.__ctx = context
+            self.__leave = leave
+            self.__stop = stop
+            self.__pause = pause
+            self.__resume = resume
+            self.__skip = skip
+
+        @discord.ui.button(style=discord.ButtonStyle.red, label='Выход', emoji='🚪')
+        async def button_leave(self, interaction: discord.Interaction, button: discord.ui.Button):
+            await interaction.response.send_message("Хорошо блин, ухожу, тупой ты дебил", tts=True, delete_after=8)
+            self.__leave(self.__ctx)
+
+        @discord.ui.button(style=discord.ButtonStyle.red, label='Стоп', emoji='🛑')
+        async def button_stop(self, interaction: discord.Interaction, button: discord.ui.Button):
+            await interaction.response.send_message("Хорошо, тормознул", tts=True, delete_after=8)
+            self.__stop(self.__ctx)
+
+        @discord.ui.button(style=discord.ButtonStyle.blurple, label='Пауза / Продолжить', emoji='⏯️')
+        async def button_pause_resume(self, interaction: discord.Interaction, button: discord.ui.Button):
+            if self.__vc.is_playing():
+                await interaction.response.send_message("Усё, усё, пауза", tts=True, delete_after=8)
+                self.__vc.pause()
+            elif self.__vc.is_paused():
+                await interaction.response.send_message("Играем дальше, значит. Ты задолбал", tts=True, delete_after=8)
+                self.__vc.resume()
+
+        @discord.ui.button(style=discord.ButtonStyle.blurple, label='Пропустить', emoji='⏭️')
+        async def button_skip(self, interaction: discord.Interaction, button: discord.ui.Button):
+            await interaction.response.send_message("Эту песню мы попускаем, потому что гивно", tts=True, delete_after=8)
+            self.__skip(self.__ctx)
 
     async def __play(self, context, url, video):
         self.__vc.play(discord.FFmpegPCMAudio(executable=settings['path_to_ffmpeg'], source = url, **FFMPEG_OPTIONS), after = lambda e: self.__skip(context = context))
@@ -97,18 +131,14 @@ async def __play(self, context, url, video):
                 .add_field(name = '⌛ Продолжительность', value = datetime.timedelta(seconds=duration))
                 .add_field(name = '🙃 Запросил', value = context.author.mention)
                 .set_thumbnail(url = video.get('thumbnail'))  )
-        firstmessage = await context.send(
-        embed = embed,
-        components = [
-            [
-                Button(style = ButtonStyle.red, label = 'Выход', emoji = '🚪'),
-                Button(style = ButtonStyle.red, label = 'Стоп', emoji = '🛑'),
-                Button(style = ButtonStyle.blue, label = 'Пауза / Продолжить', emoji = '⏯️'),
-                Button(style = ButtonStyle.blue, label = 'Пропустить', emoji = '⏭️'),
-            ]
-        ])
-
-        self.__playing_now_embed = firstmessage
+        await context.send(embed = embed, view=self.PlayerButtons(self.__vc,
+                                                                  context,
+                                                                  self.__leave,
+                                                                  self.__stop,
+                                                                  self.__pause,
+                                                                  self.__resume,
+                                                                  self.__skip))
+
         title = video.get('title')
         self.__queue.set_playing_now(title)
         while self.__vc.is_playing:
@@ -121,7 +151,6 @@ async def __play(self, context, url, video):
                     await responce.respond(content = '🚪 Бот вышел из голосового чата')
                     return
                 case 'Стоп':
-                    await self.__playing_now_embed.edit(embed=embed, components=[Button(style = ButtonStyle.red, label = 'Выход', emoji = '🚪')])
                     self.__stop(context)
                     await responce.respond(content = '🛑 Остановлено!')
                     return
@@ -138,7 +167,6 @@ async def __play(self, context, url, video):
                     await responce.respond(content = '⏭️ Скипаю...')
                     self.__skip(context)
 
-
     def __skip(self, context):
         asyncio.run_coroutine_threadsafe(self.__playing_now_embed.edit(components=[]), self.bot.loop)
         if self.__vc.is_playing():
@@ -171,7 +199,8 @@ def __pause(self, context):
     def __resume(self, context):
         if not self.__vc.is_playing():
             self.__vc.resume()
-        elif self.__vc.is_playing():asyncio.run_coroutine_threadsafe(context.send('🤪 Лол, я не на паузе, зачем ты ввёл эту команду?!'), self.bot.loop)
+        elif self.__vc.is_playing():
+            asyncio.run_coroutine_threadsafe(context.send('🤪 Лол, я не на паузе, зачем ты ввёл эту команду?!'), self.bot.loop)
 
     @commands.command()
     async def play(self, ctx, *, arg):
@@ -259,5 +288,5 @@ async def resume(self, ctx):
             await ctx.message.add_reaction("🤡")
             await ctx.send(f"{self.bot.get_emoji(settings['emojis']['wuuut'])} Для начала введи хотя бы `{settings.get('prefix')}play [название_песни]`")
 
-def setup(bot):
-    bot.add_cog(Music(bot))
+async def setup(bot, intents):
+    await bot.add_cog(Music(bot, intents))
diff --git a/cogs/text.py b/cogs/text.py
@@ -63,5 +63,5 @@ async def poll(self, ctx, question, *options: str):
         embed.set_footer(text= f'Poll ID: {react_message.id} \nКстати! Вопрос нужно указывать в кавычках!' )
         await react_message.edit(embed=embed)
 
-def setup(bot):
-    bot.add_cog(Text(bot))
+async def setup(bot):
+    await bot.add_cog(Text(bot))
diff --git a/config.py b/config.py
@@ -28,4 +28,4 @@
                    discord.Activity(type=discord.ActivityType.watching, name="поруху с собакой"),
                    discord.Game(name="игру"),
                    discord.Game(name="ебанину нахуй")]
-}
+}
diff --git a/main.py b/main.py
diff --git a/requirements.txt b/requirements.txt
diff --git a/runtime.txt b/runtime.txt
