main.py

import asyncio
import sys

from config import TITLE
from termcolor import cprint
from questionary import Choice, select
from modules.interfaces import SoftwareException
from utils.modules_runner import Runner
from utils.route_generator import RouteGenerator
from utils.tools import check_progress_file


def are_you_sure(module=None, gen_route:bool = False):
    if gen_route or check_progress_file():
        answer = select(
            '\n ⚠️⚠️⚠️ THAT ACTION WILL DELETE ALL PREVIOUS PROGRESS FOR CLASSIC-ROUTES, continue? ⚠️⚠️⚠️ \n',
            choices=[
                Choice("❌ NO", 'main'),
                Choice("✅ YES", 'module'),
            ],
            qmark='☢️',
            pointer='👉'
        ).ask()
        print()
        if answer == 'main':
            main()
        else:
            if module:
                module()


def main():
    cprint(TITLE, color='light_cyan')
    cprint(f'\n❤️ My channel for latest updates: https://t.me/askaer\n', 'light_green', attrs=["blink"])
    try:
        while True:
            answer = select(
                'What do you want to do?',
                choices=[
                    Choice("🚀 Start running classic routes for each wallet", 'classic_routes_run'),
                    Choice("📄 Generate classic-route for each wallet", 'classic_routes_gen'),
                    Choice("✅ Check the connection of each proxy", 'check_proxy'),
                    Choice('❌ Exit', "exit")
                ],
                qmark='🛠️',
                pointer='👉'
            ).ask()

            runner = Runner()

            if answer == 'check_proxy':
                print()
                asyncio.run(runner.check_proxies_status())
                print()
            elif answer == 'classic_routes_run':
                print()
                asyncio.run(runner.run_accounts())
                print()
            elif answer == 'classic_routes_gen':
                generator = RouteGenerator()
                are_you_sure(generator.classic_routes_json_save, gen_route=True)
            elif answer == 'exit':
                sys.exit()
            else:
                print()
                answer()
                print()
    except KeyboardInterrupt:
        cprint(f'\nQuick software shutdown by <ctrl + C>', color='light_yellow')
        sys.exit()

    except SoftwareException as error:
        cprint(f'\n{error}', color='light_red')
        sys.exit()


if __name__ == "__main__":
    main()