main.py

import asyncio
import itertools
from copy import deepcopy
from multiprocessing.dummy import Pool
from random import randint
from sys import exit
from sys import platform

import config
from core import start_reger_wrapper
from twitter_core import start_subs, say_gm
from utils import format_range, logger, validate_token

if platform == "windows":
    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())

if __name__ == '__main__':
    print('Donate (any EVM) - 0xDEADf12DE9A24b47Da0a43E1bA70B8972F5296F2\n')

    with open('accounts.txt', 'r', encoding='utf-8-sig') as file:
        accounts_list: list[str] = [validate_token(input_string=row.strip()) for row in file]

    accounts_list: list = [item for item in accounts_list if item is not None]

    with open('proxies.txt', 'r', encoding='utf-8-sig') as file:
        proxies_list: list[str] = [row.strip() for row in file]

    with open('private_keys.txt', 'r', encoding='utf-8-sig') as file:
        private_keys_list: list[str] = [f'0x{row.strip()}' if not row.strip().startswith('0x') else row.strip() for row
                                        in file]

    cycled_proxies_list = itertools.cycle(proxies_list) if proxies_list else None

    logger.info(f'Загружено {len(accounts_list)} аккаунтов / {len(proxies_list)} '
                f'прокси / {len(private_keys_list)} приват-кеев')

    user_action: int = int(input('\n1. Запуск накрутки MEMELand\n'
                                 '2. Подписка между аккаунтами Twitter\n'
                                 '3. Say GM\n'
                                 'Введите ваше действие: '))

    threads: int = 1 if config.CHANGE_PROXY_URL else int(input('Threads: '))

    match user_action:
        case 1:
            print()

            formatted_accounts_list: list = [
                {
                    'account_token': current_account,
                    'account_proxy': next(cycled_proxies_list) if cycled_proxies_list else None,
                    'account_private_key': private_keys_list.pop(0) if private_keys_list else None
                } for current_account in accounts_list
            ]

            with Pool(processes=threads) as executor:
                tasks_result: list = executor.map(start_reger_wrapper, formatted_accounts_list)

            success_count: int = sum(tasks_result) if tasks_result.count(True) else 0
            fail_count: int = len(tasks_result) - success_count

            logger.info(f'Статистика работы: {success_count} SUCCESS | {fail_count} FAILED')

        case 2:
            subs_range: str = input('Введите диапазон необходимого количества подписок (ex: 3-5, 4-10, 5, 10): ')

            first_int_subs_range, second_int_subs_range = format_range(value=subs_range)

            if second_int_subs_range >= len(accounts_list):
                second_int_subs_range: int = len(accounts_list) - 1

            if first_int_subs_range > second_int_subs_range:
                logger.error('Неверно введен диапазон количества подписок')
                input('\nPress Enter To Exit..')
                exit()

            print()

            formatted_accounts_list: list = [
                {
                    'target_account_token': current_account,
                    'accounts_list': deepcopy(accounts_list),
                    'proxies_list': cycled_proxies_list,
                    'subs_count': randint(first_int_subs_range, second_int_subs_range)
                } for current_account in accounts_list
            ]

            with Pool(processes=threads) as executor:
                executor.map(start_subs, formatted_accounts_list)

        case 3:
            print()

            formatted_accounts_list: list = [
                {
                    'account_token': current_account,
                    'account_proxy': next(cycled_proxies_list) if cycled_proxies_list else None,
                } for current_account in accounts_list
            ]

            with Pool(processes=threads) as executor:
                tasks_result: list = executor.map(say_gm, formatted_accounts_list)

            success_count: int = sum(tasks_result) if tasks_result.count(True) else 0
            fail_count: int = len(tasks_result) - success_count

            logger.info(f'Статистика работы: {success_count} SUCCESS | {fail_count} FAILED')

    logger.success('Работа успешно завершена')
    input('\nPress Enter To Exit..')