main.py

from config import *
from src.translate_qr_code import get_link_qr_code
from src.check_url import check_link


@dp.message_handler(commands=['start'])
@dp.message_handler(content_types=['photo'])
@dp.message_handler(Text(equals='Начать 📝'))
async def process_start_command(message: types.Message):
    Start, URL, QRcode, Backup = 'Начать 📝', 'Отправить URL 👀', 'Загрузить QR_code 🖥', 'Назад ↩️'
    keyboard = types.ReplyKeyboardMarkup(resize_keyboard=True)
    keyboard.add(Start)
    keyboard.add(URL)
    keyboard.add(QRcode)
    keyboard.add(Backup)
    await message.answer('🔄 <b> БОТ Information Defender </b> - ЗАПУСКАЕТСЯ, ПОЖАЛУЙСТА, ПОДОЖДИТЕ... 🔄',
                         parse_mode='HTML')
    with open('img.png', 'rb') as file:
        file.seek(0)
        await message.answer_photo(photo=file)
        file.close()
        await message.answer(
            """
|----------------------------------
| <b>Приветствую тебя пользователь!</b>
|----------------------------------
| <b>Посмотрите на список команд:</b>
|----------------------------------
|
|<b>---> /start - Запуск бота </b> 🖥
|
|<b>---> /qr_code - Проверка QR </b> 💾
|
|<b>---> /url - Проверка url </b> 💾
|
|<b>---> /help - Помощь </b> 🔧
|
|----------------------------------
            """, parse_mode="HTML", reply_markup=keyboard)


@dp.message_handler(commands=["qr_code"])
async def cmd_qrcode(message: types.Message, state: FSMContext):
    await state.set_state(TestStates.all()[0])
    await bot.send_message(message.from_user.id, "💾Отправь <b>QR-код</b>, который хочешь проверить: 💾",
                           parse_mode='HTML')


@dp.message_handler(commands=["url"])
async def cmd_url(message: types.Message, state: FSMContext):
    await state.set_state(TestStates.all()[1])
    await bot.send_message(message.from_user.id, "💾 Напиши <b>URL</b>, который хочешь проверить: 💾", parse_mode='HTML')


@dp.message_handler(commands=["help"])
async def cmd_help(message: types.Message):
    Start, URL, QRcode, Backup = 'Начать 📝', 'Отправить URL 👀', 'Загрузить QR_code 🖥', 'Назад ↩️'
    keyboard = types.ReplyKeyboardMarkup(resize_keyboard=True)
    keyboard.add(Start)
    keyboard.add(URL)
    keyboard.add(QRcode)
    keyboard.add(Backup)
    with open('img_1.png', 'rb') as file:
        file.seek(0)
        await message.answer_photo(photo=file)
        file.close()
        await message.answer(
            """
<b>ВЫ ОБРАТИЛИСЬ К КОМАНДЕ /help</b>

1️⃣ /start - запустить бота для пользования
            
2️⃣ /qrcode - при указании QRcode'а в данной команде, вы получите ссылку, а затем ссылка будет проверенна на различные факторы безопасности ссылки и затем выводиться список.
            
3️⃣ /url - при указании ссылки в данной команде, выполнится проверка на различные факторы безопасности и тут же вам выводиться список

4️⃣ /help - инструкция, какие команды доступны и что они выполняют

            """, parse_mode='HTML', reply_markup=keyboard)


@dp.message_handler(Text(equals='Отправить URL 👀'))
async def processing_url(message: types.Message, state: FSMContext):
    await state.set_state(TestStates.all()[1])
    await bot.send_message(message.from_user.id, """
💾 Напиши <b>URL</b>, который хочешь проверить: 💾
""", parse_mode='HTML')


@dp.message_handler(Text(equals='Загрузить QR_code 🖥'))
async def processing_qr_code(message: types.Message, state: FSMContext):
    await state.set_state(TestStates.all()[0])
    await bot.send_message(message.from_user.id, """
💾Отправь <b>QR-код</b>, который хочешь проверить: 💾
""", parse_mode='HTML')


@dp.message_handler(Text(equals='Назад ↩️'), state=TestStates.QR_STATE[0])
async def processing_qr_code(message: types.Message, state: FSMContext):
    await state.reset_state()
    Start, URL, QRcode, Backup = 'Начать 📝', 'Отправить URL 👀', 'Загрузить QR_code 🖥', 'Назад ↩️'
    keyboard = types.ReplyKeyboardMarkup(resize_keyboard=True)
    keyboard.add(Start)
    keyboard.add(URL)
    keyboard.add(QRcode)
    keyboard.add(Backup)
    await message.answer('🔄 <b>ИДЁТ ПЕРЕНАПРАВЛЕНИЕ В ГЛАВНОЕ ОКНО, ПОЖАЛУЙСТА, ПОДОЖДИТЕ...</b> 🔄', parse_mode='HTML')
    with open('img.png', 'rb') as file:
        file.seek(0)
        await message.answer_photo(photo=file)
        file.close()
        await message.answer(
            """
|----------------------------------
| <b>Приветствую тебя пользователь!</b>
|----------------------------------
| <b>Посмотрите на список команд:</b>
|----------------------------------
|
|<b>---> /start - Запуск бота </b> 🖥
|
|<b>---> /qr_code - Проверка QR </b> 💾
|
|<b>---> /url - Проверка url </b> 💾
|
|<b>---> /help - Помощь </b> 🔧
|
|----------------------------------
            """, parse_mode="HTML", reply_markup=keyboard)


@dp.message_handler(Text(equals='Назад ↩️'), state=TestStates.URL_STATE[0])
async def processing_url(message: types.Message, state: FSMContext):
    await state.reset_state()
    Start, URL, QRcode, Backup = 'Начать 📝', 'Отправить URL 👀', 'Загрузить QR_code 🖥', 'Назад ↩️'
    keyboard = types.ReplyKeyboardMarkup(resize_keyboard=True)
    keyboard.add(Start)
    keyboard.add(URL)
    keyboard.add(QRcode)
    keyboard.add(Backup)
    await message.answer('🔄 <b>ИДЁТ ПЕРЕНАПРАВЛЕНИЕ В ГЛАВНОЕ ОКНО, ПОЖАЛУЙСТА, ПОДОЖДИТЕ...</b> 🔄', parse_mode='HTML')
    with open('img.png', 'rb') as file:
        file.seek(0)
        await message.answer_photo(photo=file)
        file.close()
        await message.answer(
            """
|----------------------------------
| <b>Приветствую тебя пользователь!</b>
|----------------------------------
| <b>Посмотрите на список команд:</b>
|----------------------------------
|
|<b>---> /start - Запуск бота </b> 🖥
|
|<b>---> /qr_code - Проверка QR </b> 💾
|
|<b>---> /url - Проверка url </b> 💾
|
|<b>---> /help - Помощь </b> 🔧
|
|----------------------------------
            """, parse_mode="HTML", reply_markup=keyboard)


@dp.message_handler(state=TestStates.URL_STATE[0])
async def solution_url(message: types.Message, state: FSMContext):
    url = message.text
    await message.reply \
        ("""                    
🔅 Идёт проверка ссылки... 🔅                               
        """, reply=False)
    time.sleep(1)
    try:
        parsed_url = urlparse(url)
        domain = parsed_url.scheme + '://' + parsed_url.netloc + '/'
        result = check_link(domain)
        galochka, krestik = '✅', '❌'
        with open('img_2.png', 'rb') as file:
            file.seek(0)
            await message.answer_photo(photo=file)
            file.close()
            time.sleep(1)
        card = \
            f'|----------------------------------\n' \
            f'|\n' \
            f'|<b>----> URL:</b> {domain}\n' \
            f'|<b>----> Перенаправления:</b>     {krestik if result["redirect"] == True else galochka}\n' \
            f'|<b>----> Поддержка https:</b>          {galochka}\n' \
            f'|<b>----> SSL сертификат:</b>              {galochka if result["ssl"] == True else krestik}\n' \
            f'|<b>----> Известный домен:</b>           {krestik if result["suspicious"] == True else galochka}\n' \
            f'|<b>----> Подозрительный JS код:</b> {krestik if result["suspicious_js"] == False else galochka}\n' \
            f'|<b>----> Доменные уровни:</b>           {krestik if result["Long level"] == True else galochka}\n' \
            f'|<b>----> Читаемый домен:</b>           {krestik if result["Unreadability"] == True else galochka}\n' \
            f'|\n' \
            f'|----------------------------------'
        await message.reply(card, reply=False)
        await state.reset_state()
    except:
        try:
            parsed_url = urlparse(url)
            domain = 'http://' + parsed_url.netloc + '/'
            result = check_link(domain)
            galochka, krestik = '✅', '❌'
            with open('img_2.png', 'rb') as file:
                file.seek(0)
                await message.answer_photo(photo=file)
                file.close()
                time.sleep(1)
                await message.reply(
                    """
<b>⚠️Найден запрет протокола https⚠️</b>
<b>🖥Идёт замена протокола https...🖥</b>
                    """, parse_mode='HTML', reply=False)
                time.sleep(1)
            card = \
                f'|----------------------------------\n' \
                f'|\n' \
                f'|<b>----> URL:</b> {domain}\n' \
                f'|<b>----> Перенаправления:</b>     {krestik if result["redirect"] == True else galochka}\n' \
                f'|<b>----> Поддержка https:</b>          {galochka}\n' \
                f'|<b>----> SSL сертификат:</b>              {galochka if result["ssl"] == True else krestik}\n' \
                f'|<b>----> Известный домен:</b>           {krestik if result["suspicious"] == True else galochka}\n' \
                f'|<b>----> Подозрительный JS код:</b> {krestik if result["suspicious_js"] == False else galochka}\n' \
                f'|<b>----> Доменные уровни:</b>           {krestik if result["Long level"] == True else galochka}\n' \
                f'|<b>----> Читаемый домен:</b>           {krestik if result["Unreadability"] == True else galochka}\n' \
                f'|\n' \
                f'|----------------------------------'
            await message.reply(card, parse_mode="HTML", reply=False)
            await state.reset_state()
        except:
            await message.reply(
                """
|----------------------------------
|                ⚠️⚠️⚠️
|----------------------------------                
|           <b>ОШИБКА URL</b>
|----------------------------------
|   <b>Причиной может быть:</b>
|----------------------------------
|
|---> 1️⃣ - Не существующий домен
|
|---> 2️⃣ - Не указан протокол
|
|---> 3️⃣ - Ссылка локальная
|
|----------------------------------
                """, parse_mode='HTML', reply=False)
            await state.reset_state()


@dp.message_handler(state=TestStates.QR_STATE[0], content_types=['photo'])
async def solution_QRcode(message: types.Message, state: FSMContext):
    await message.reply("📎 Изображение скачивается... 📎 ", reply=False)
    time.sleep(1)
    await message.photo[-1].download('src/img.png')
    try:
        url = get_link_qr_code()
        await message.reply("""                    
        🔅 Идёт проверка ссылки... 🔅                               
            """, reply=False)
        time.sleep(1)
        try:
            with open('img_3.png', 'rb') as file:
                file.seek(0)
                await message.answer_photo(photo=file)
                file.close()
                time.sleep(1)
            parsed_url = urlparse(url)
            domain = parsed_url.scheme + '://' + parsed_url.netloc + '/'
            result = check_link(domain)
            galochka, krestik = '✅', '❌'
            card = \
                f'|----------------------------------\n' \
                f'|\n' \
                f'|<b>----> URL:</b> {domain}\n' \
                f'|<b>----> Перенаправления:</b>     {krestik if result["redirect"] == True else galochka}\n' \
                f'|<b>----> Поддержка https:</b>          {galochka}\n' \
                f'|<b>----> SSL сертификат:</b>              {galochka if result["ssl"] == True else krestik}\n' \
                f'|<b>----> Известный домен:</b>           {krestik if result["suspicious"] == True else galochka}\n' \
                f'|<b>----> Подозрительный JS код:</b> {krestik if result["suspicious_js"] == False else galochka}\n' \
                f'|<b>----> Доменные уровни:</b>           {krestik if result["Long level"] == True else galochka}\n' \
                f'|<b>----> Читаемый домен:</b>           {krestik if result["Unreadability"] == True else galochka}\n' \
                f'|\n' \
                f'|----------------------------------'
            await message.reply(card, reply=False)
            await state.reset_state()
        except:
            try:
                time.sleep(1)
                parsed_url = urlparse(url)
                domain = 'http://' + parsed_url.netloc + parsed_url.path
                site = 'https://' + parsed_url.netloc
                result = check_link(domain)
                galochka, krestik = '✅', '❌'
                await message.reply(
                    """
<b>⚠️Найден запрет протокола https⚠️</b>
<b>🖥Идёт замена протокола https...🖥</b>
                    """, parse_mode='HTML', reply=False)
                time.sleep(1)
                card = \
                    f'|----------------------------------\n' \
                    f'|\n' \
                    f'|<b>----> URL:</b> {site}\n' \
                    f'|<b>----> Перенаправления:</b>     {krestik if result["redirect"] == True else galochka}\n' \
                    f'|<b>----> Поддержка https:</b>          {galochka}\n' \
                    f'|<b>----> SSL сертификат:</b>              {galochka if result["ssl"] == True else krestik}\n' \
                    f'|<b>----> Известный домен:</b>           {krestik if result["suspicious"] == True else galochka}\n' \
                    f'|<b>----> Подозрительный JS код:</b> {krestik if result["suspicious_js"] == True else galochka}\n' \
                    f'|<b>----> Доменные уровни:</b>           {krestik if result["Long level"] == True else galochka}\n' \
                    f'|<b>----> Читаемый домен:</b>           {krestik if result["Unreadability"] == True else galochka}\n' \
                    f'|\n' \
                    f'|----------------------------------'
                await message.reply(card, parse_mode='HTML', reply=False)
                await state.reset_state()
            except:
                await message.reply("<b>⚠️ Ссылка в QR-коде некорректная... ⚠️</b>", parse_mode='HTML', reply=False)
                await state.reset_state()
    except:
        await message.reply(
            """
|----------------------------------
|                ⚠️⚠️⚠️
|----------------------------------                
|       <b>ОШИБКА QR-КОДА</b>
|----------------------------------
|   <b>Причиной может быть:</b>
|----------------------------------
|
|---> 1️⃣ - Плохое качество картинки
|
|---> 2️⃣ - QR-код недействителен
|
|---> 3️⃣ - Недопустимый формат
|
|----------------------------------
            """, parse_mode='HTML', reply=False)
        await state.reset_state()


if __name__ == '__main__':
    executor.start_polling(dp)