Update bot: return Xur checker and refactor code

Author: Interligo

bot/bot_functions.py

@@ -5,13 +5,13 @@
 from fuzzywuzzy import process
 
 import image_parser as parser
-import bot_words_storage as st
+import bot_storage as st
 
 
 def word_lists_counter() -> int:
     """Function to get number of word lists from file 'bot_words_storage.py'."""
     lists_counter = 0
-    storage_path = os.path.join(os.path.dirname(__file__), 'bot_words_storage.py')
+    storage_path = os.path.join(os.path.dirname(__file__), 'bot_storage.py')
 
     if os.path.exists(storage_path):
         with open(storage_path) as file:
@@ -42,7 +42,7 @@ def discord_message_analysis(message: discord.MessageType) -> str:
         for iteration in range(word_lists_number):
             if result is None:
                 found_word, rating = process.extractOne(word, referring_to_word_lists[iteration])
-                result = callback_from_word_lists[iteration] if rating > 85 else None
+                result = callback_from_word_lists[iteration] if rating > 95 else None
 
     return result
 
@@ -58,17 +58,22 @@ def bad_word_finder(message: discord.MessageType) -> str:
     for word in message_content_list:
         found_word, rating = process.extractOne(word, st.bad_words_list)
         if result is None:
-            result = found_word if rating > 75 else None
+            result = found_word if rating > 95 else None
 
     return result
 
 
+def select_answer() -> str:
+    """Function to select random answer."""
+    index = random.randint(0, len(st.ANSWERS) - 1)
+    return st.ANSWERS[index]
+
+
 def image_selection() -> str:
     """Function to randomize image choice."""
     start_of_link = 'http://babenki.info/'
     image_links_list = parser.get_images()
-    image_count = len(image_links_list)
-    choice = random.randint(0, image_count)
+    choice = random.randint(0, len(image_links_list)-1)
 
     link = start_of_link + image_links_list[choice]
 

bot/bot_main.py

@@ -1,15 +1,14 @@
 import os
 import asyncio
 import random
+
 import discord
-from discord.ext import commands
+from discord import Intents
+from discord.ext import commands, tasks
 
 from load_environment import load_environment
-from bot_functions import discord_message_analysis
-from bot_functions import bad_word_finder
-from bot_functions import image_selection
-from xur_destiny2 import is_xur_here
-from xur_destiny2 import get_xur_location
+from bot_functions import discord_message_analysis, bad_word_finder, select_answer, image_selection
+from xur_destiny2 import XurChecker
 
 
 load_environment()
@@ -18,13 +17,17 @@
 BOT_TOKEN = os.getenv('BOT_TOKEN')
 PREFIX = '!'
 
-bot = commands.Bot(command_prefix=PREFIX)
+bot = commands.Bot(command_prefix=PREFIX, intents=Intents.all())
+bot.intents.members = True
 bot.remove_command('help')
 
+xur_checker = XurChecker()
+
 
 @bot.event
 async def on_ready():
     print(f'{bot.user.name} в сети.')
+    is_xur_arrived.start()
 
 
 @bot.event
@@ -33,7 +36,7 @@ async def on_member_join(member):
                       f'Можешь ознакомиться со списком доступных команд !help и обязательно прочти наши !rules. '
                       f'Мой баннхаммер не знает пощады!')
 
-    guest_role = discord.utils.get(member.guild.roles, name='Гость')
+    guest_role = discord.utils.get(member.guild.roles, id=780843833817563166)
     await member.add_roles(guest_role)
 
 
@@ -65,10 +68,12 @@ async def on_message(message):
             await message.channel.send(f'До скорого, {message_author.mention}!')
         elif result == 'bad':
             bad_word = bad_word_finder(message)
-            await message.channel.send(f'Так ты сам, {message_author.mention}, {bad_word} получается.')
+            answer = select_answer()
+            await message.channel.send(f'Выявлен нарушитель священного пункта № 1 правил Общества '
+                                       f'в связи с использованием "{bad_word}"!')
+            await message.channel.send(f'{message_author.mention} {answer}')
         elif result == 'help':
-            await message.channel.send(f'{message_author.mention}, хочешь узнать список команд? '
-                                       f'Напиши !help в чат.')
+            await message.channel.send(f'{message_author.mention}, хочешь узнать список команд? Напиши !help в чат.')
 
     await bot.process_commands(message)
 
@@ -77,27 +82,53 @@ async def on_message(message):
 async def help(ctx):
     await ctx.channel.purge(limit=1)
 
-    embed = discord.Embed(colour=discord.Color.gold(), title='Навигация по командам:')
+    embed = discord.Embed(
+        colour=discord.Color.gold(),
+        title='Навигация по командам:'
+    )
+
     embed.set_author(name=bot.user.name, icon_url=bot.user.avatar_url)
-    embed.add_field(name=f'{PREFIX}help', value='Выводит список команд.')
-    embed.add_field(name=f'{PREFIX}rules', value='Выводит список правил сервера.')
-    embed.add_field(name=f'{PREFIX}boobs', value='Безмерно радует всех Сиськолюбов.')
-    embed.add_field(name=f'{PREFIX}duel @target_name', value='Позволяет вызвать на дуэль и испытать удачу. '
-                                                             'Проигравший отправляется в мут на 60 секунд.')
+
+    embed.add_field(name=f'{PREFIX}help', value='Выводит список команд.', inline=False)
+    embed.add_field(name=f'{PREFIX}rules', value='Выводит список правил сервера.', inline=False)
+    embed.add_field(name=f'{PREFIX}boobs', value='Безмерно радует всех Сиськолюбов.', inline=False)
+    embed.add_field(name=f'{PREFIX}xur', value='Сообщает местонахождение Зура.', inline=False)
+    embed.add_field(name=f'{PREFIX}duel @target_name',
+                    value='Позволяет вызвать на дуэль и испытать удачу. Проигравший отправляется в мут на 60 секунд.',
+                    inline=False)
 
     await ctx.send(embed=embed)
 
 
-@bot.command()
+@bot.command(aliases=['правила', 'Правила'])
 async def rules(ctx):
     await ctx.channel.purge(limit=1)
 
-    embed = discord.Embed(colour=discord.Color.gold(), title='Правила сервера:')
-    embed.set_author(name=bot.user.name, icon_url=bot.user.avatar_url)
-    embed.add_field(name='0.', value='Всегда уважай Великий Разум!', inline=True)
-    embed.add_field(name='1.', value='Мы здесь отдыхаем, поэтому будь дружелюбнее.', inline=True)
-    embed.add_field(name='2.', value='Запрещено быть душным мудилой и говорить "можно скажу?" в голосовом чате.',
-                    inline=True)
+    embed = discord.Embed(
+        colour=discord.Color.gold(),
+        title='Правила сервера:'
+    )
+
+    embed.set_author(
+        name=bot.user.name,
+        icon_url=bot.user.avatar_url
+    )
+
+    embed.add_field(
+        name='Пункт № 0.',
+        value='Всегда уважай Великий Разум!',
+        inline=False
+    )
+    embed.add_field(
+        name='Пункт № 1.',
+        value='Мы здесь отдыхаем, поэтому будь дружелюбнее.',
+        inline=False
+    )
+    embed.add_field(
+        name='Пункт № 2.',
+        value='Запрещено быть душным мудилой и говорить "можно скажу?" в голосовом чате.',
+        inline=False
+    )
 
     await ctx.send(embed=embed)
 
@@ -108,41 +139,41 @@ async def clear(ctx, amount_to_delete=2):
     await ctx.channel.purge(limit=amount_to_delete)
 
 
-@bot.command(aliases=['бан'])
+@bot.command(aliases=['бан', 'Бан'])
 @commands.has_permissions(administrator=True)
 async def mute(ctx, member: discord.Member):
     await ctx.channel.purge(limit=1)
 
-    mute_role = discord.utils.get(ctx.message.guild.roles, name='Заключенный')
+    mute_role = discord.utils.get(ctx.message.guild.roles, id=780439303603355648)
     await member.add_roles(mute_role)
 
     await ctx.send(f'Великий Разум пожелал, чтобы {member.mention} немного помолчал.')
     await member.send(f'Фреймы призваны помогать людям, поэтому я здесь. Тебя забанил {ctx.author.name}. '
-                      f'Причина мне неизвестна, однако Великий Разум обратил на тебя своё внимание, это такая честь!')
+                      f'Причина мне неизвестна, однако Великий Разум ничего не делает просто так!')
 
 
-@bot.command(aliases=['разбан'])
+@bot.command(aliases=['разбан', 'Разбан'])
 @commands.has_permissions(administrator=True)
 async def unmute(ctx, member: discord.Member):
     await ctx.channel.purge(limit=1)
 
-    mute_role = discord.utils.get(ctx.message.guild.roles, name='Заключенный')
+    mute_role = discord.utils.get(ctx.message.guild.roles, id=780439303603355648)
     await member.remove_roles(mute_role)
 
     await ctx.send(f'Великий Разум сжалился над {member.mention} и он снова может говорить.')
     await member.send(f'Ты снова можешь говорить. '
                       f'Пожалуйста, для твоего блага, постарайся больше не раздражать Великий Разум.')
 
 
-@bot.command()
+@bot.command(aliases=['дуэль', 'Дуэль'])
 async def duel(ctx, member: discord.Member):
     await ctx.channel.purge(limit=1)
 
     duelist1 = ctx.author.name
     duelist2 = member.name
     duelists = [duelist1, duelist2]
 
-    mute_role = discord.utils.get(ctx.message.guild.roles, name='Заключенный')
+    mute_role = discord.utils.get(ctx.message.guild.roles, id=780439303603355648)
 
     await ctx.send(f'{duelist1} бросает перчатку вызова {duelist2}. Приготовиться к дуэли! '
                    f'Великий Разум назначил меня вашим секундантом.')
@@ -172,36 +203,48 @@ async def duel(ctx, member: discord.Member):
 async def boobs(ctx):
     await ctx.channel.purge(limit=1)
 
-    image = image_selection()
+    image_url = image_selection()
+
+    embed = discord.Embed(
+        colour=discord.Color.gold(),
+        description='Решил порадовать друзей.'
+    )
+    embed.set_author(name=ctx.author.name, icon_url=ctx.author.avatar_url)
+
+    # Проверка URL изображения. Если длина менее 20 символов, значит изображение не спарсилось и нужно вывести ошибку.
+    if len(image_url) < 25:
+        embed.add_field(
+            name='Но у него ничего не вышло!',
+            value='Видимо, все девочки заняты, попробуй позже <:harold:970308021546795038>',
+            inline=False
+        )
+    else:
+        embed.set_image(url=image_url)
 
-    await ctx.send(image)
-    await ctx.send(f'{ctx.author.name} решил порадовать друзей.')
+    await ctx.send(embed=embed)
 
 
 @bot.command(aliases=['зур', 'Зур'])
 async def xur(ctx):
     await ctx.channel.purge(limit=1)
 
-    xur_location = get_xur_location()
+    xur_location = xur_checker.get_xur_location()
 
     await ctx.send(f'{ctx.author.mention} {xur_location}')
 
 
+@tasks.loop(hours=1)
 async def is_xur_arrived():
     await bot.wait_until_ready()
     channel = bot.get_channel(int(CHANNEL_ID))
-    message_sent = False
-
-    while True:
-        if is_xur_here():
-            if not message_sent:
-                xur_location = get_xur_location()
-                await channel.send(f'{xur_location}')
-                message_sent = True
-        if not is_xur_here():
-            message_sent = False
-
-        await asyncio.sleep(60 * 60)
+
+    if xur_checker.is_xur_here():
+        if not xur_checker.message_is_sent:
+            xur_location = xur_checker.get_xur_location()
+            await channel.send(f'{xur_location}')
+            xur_checker.message_is_sent = True
+    if not xur_checker.is_xur_here():
+        xur_checker.message_is_sent = False
 
 
 if __name__ == '__main__':

bot/bot_storage.py

@@ -0,0 +1,16 @@
+hello_words_list = ('привет', 'здорова', 'салам')
+
+by_words_list = ('пока', 'до свидания', 'сваливаю')
+
+help_words_list = ('помощь', 'хелп', 'команды')
+
+bad_words_list = ('пидор', 'хуй', 'сука', 'блядь', 'бля', 'ебать', 'ебанул', 'пизда', 'заебись', 'жопа', 'говно',
+                  'нахуй')
+
+ANSWERS = (
+    'В таких случаях Великий Разум позволил мне использовать могучий артефакт из Уно - <:reverse:970267180891512842>!',
+    'НЕ ВЫРАЖАЙСЯ! <:captan:970266647208263690>',
+    'Ублюдок, мать твою, а ну иди сюда, гавно собачье, решил ко мне лезть? '
+    'Ты, засранец вонючий, мать твою, а ну иди сюда, попробуй меня трахнуть! '
+    'Я тебя сам трахну, ублюдок, онанист чертов, будь ты проклят!'
+)

bot/bot_words_storage.py

@@ -5,7 +5,7 @@
 URL = 'http://babenki.info/siski/7487-krasivaya-golaya-grud-78-foto.html'
 
 
-def get_images():
+def get_images() -> []:
     """Function parsing shared link to find photos."""
     session = requests.Session()
     response = session.get(URL)
@@ -31,7 +31,4 @@ def get_images():
             except TypeError:
                 pass
 
-    else:
-        return 'Сайт не отвечает на запросы.'
-
     return image_list