main.py

import os
import json
from sys import stderr, exit

import tls_client
import inquirer
import xlsxwriter
from art import text2art
from loguru import logger
from termcolor import colored
from inquirer.themes import load_theme_from_dict as loadth


# FILES SETTINGS
cwd = os.getcwd()
file_data1 = f'{cwd}/files/database1.json'
file_data2 = f'{cwd}/files/database2.json'
file_query1 = f'{cwd}/files/query1.json'
file_query2 = f'{cwd}/files/query2.json'
file_query3 = f'{cwd}/files/query3.json'
file_wallets = f'{cwd}/files/wallets.txt'
file_excel_table = f'{cwd}/LayerZero Stats.xlsx'

# LOGGING SETTING
logger.remove()
logger.add(stderr, format="<white>{time:HH:mm:ss}</white> | <level>{level: <8}</level> | <cyan>{line}</cyan> - <white>{message}</white>")

WALLETS = []
QUERY1 = 2464151
QUERY2 =  2492847


def is_exists(path) -> bool:
    return os.path.isfile(path)


def filter_wallets1(wallet: dict) -> bool:
    if (wallet['user_address'].lower() in WALLETS):
        return True
    return False

def filter_wallets2(wallet: dict) -> bool:
    try:
        if (wallet['address'].lower() in WALLETS):
            return True
    except:
        return False
    return False


def load_wallets() -> None:
    global WALLETS
    with open(file_wallets, 'r') as file:
        WALLETS = [row.strip().lower() for row in file]


def edit_dates1(wallets: list) -> None:
    for wallet in wallets:
        for i in wallet:
            if (i in (['initial_block_date', 'last_block_date'])):
                wallet[i] = wallet[i][:10]
            if (i == 'amount_usd' and wallet[i] != None):
                wallet[i] = round(wallet[i],2)
                
def edit_dates2(wallets: list) -> None:
    for wallet in wallets:
        for i in wallet:
            if (i == 'eth_total' and wallet[i] != None):
                wallet[i] = f'{round(wallet[i],4)} ({round(wallet[i]*1800,2)})'
            if (i == 'usd_total' and wallet[i] != None):
                wallet[i] = round(wallet[i],2)
                
def get_filtered_wallets(data_file) -> list:
    with open(data_file, 'r') as file:
        data = json.load(file)

    all_wallet_info = data['data']['get_execution']['execution_succeeded']['data']
    
    if (data_file == file_data1):
        filtered_wallets = list(filter(filter_wallets1, all_wallet_info))
        edit_dates1(filtered_wallets)
    else:
        filtered_wallets = list(filter(filter_wallets2, all_wallet_info))
        edit_dates2(filtered_wallets)
    return filtered_wallets


def save_to_excel(wallets1: list, wallets2: list) -> None:
    columns = list(wallets1[0].keys())

    columns.insert(5,"eth_total")
    columns.insert(6,"stables_total")
    pretty_columns = [' '.join([j.title() for j in i.split('_')]) for i in columns]

    for wallet in wallets1:
        for i, wallet2 in enumerate(wallets2):
            if wallet["user_address"] == wallet2["address"]:
                break
        else:
            wallet["eth_total"] = 0
            wallet["stables_total"] = 0
            continue
        wallet["eth_total"] = wallets2[i]["eth_total"]
        wallet["stables_total"] = wallets2[i]["usd_total"]
    
    workbook = xlsxwriter.Workbook(file_excel_table)
    worksheet = workbook.add_worksheet("Stats")

    header_format = workbook.add_format({
        'bold': True,
        'align': 'center',
        'valign': 'vcenter',
        'text_wrap': True,
        'border': 1
    })
    for col_num, column in enumerate(pretty_columns):
        worksheet.write(0, col_num, column, header_format)

    for row_num, wallet in enumerate(wallets1, 1):
        for col_num, col in enumerate(columns):
            worksheet.write(row_num, col_num, wallet[col])

    worksheet.write(len(wallets1) + 3, 0, 'Donate:')
    worksheet.write(len(wallets1) + 3, 1, '0x2e69Da32b0F7e75549F920CD2aCB0532Cc2aF0E7')

    row_format = workbook.add_format({'align': 'center'})
    sizes = [9, 45, 8, 10, 12, 12, 12, 12, 11, 15, 11, 12, 11, 9, 11, 11]
    for col_num, size in enumerate(sizes):
        worksheet.set_column(col_num, col_num, size, row_format)

    first_row_format = workbook.add_format({
        'text_wrap': True,
        'valign': 'vcenter',
        'align': 'center',
        'border': 1
    })
    worksheet.set_row(0, 50, first_row_format)

    workbook.close()


def get_execution_id(session, query_id):
    with open(file_query1, 'r') as file:
        payload = json.load(file)

    payload['variables']['query_id'] = query_id

    while True:
        try:
            response = session.post('https://core-hsr.dune.com/v1/graphql', json=payload)
            if (response.status_code == 200):
                break
            else:
                logger.error(f'Ошибка обновления базы данных: {response.text} | Cтатус запроса: {response.status_code}')
        except Exception as error:
            logger.error(f'Ошибка обновления базы данных: {error}')

    execution_id = response.json()['data']['get_result_v3']['result_id']
    return execution_id



def setup_session():
    session = tls_client.Session(
        client_identifier="chrome112",
        random_tls_extension_order=True
    )

    headers = {
        'origin': 'https://dune.com',
        'referer': 'https://dune.com/',
        'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
    }

    session.headers = headers
    return session


def update_database() -> None:
    session = setup_session()

    logger.info('Начинаю скачивание баз данных. Процесс может занять несколько минут...')

    with open(file_query2, 'r') as file:
        payload = json.load(file)

    execution_id = get_execution_id(session, QUERY1)
    logger.info(f'ID запроса №{QUERY1}: {execution_id}')
    payload['variables']['execution_id'] = execution_id

    while True:
        try:
            response = session.post('https://app-api.dune.com/v1/graphql', json=payload)
            if (response.status_code == 200):
                logger.success(f'База данных №{QUERY1} успешно скачана!')
                break
            else:
                logger.error(f'Ошибка обновления базы данных: {response.text} | Cтатус запроса: {response.status_code}')
        except Exception as error:
            logger.error(f'Ошибка обновления базы данных: {error}')
    
    with open(file_data1, 'w') as file:
        json.dump(response.json(), file)

    #----------------------------------------------

    logger.info(f'Скачиваю вторую базу данных')

    with open(file_query3, 'r') as file:
        payload = json.load(file)

    execution_id = get_execution_id(session, QUERY2)
    logger.info(f'ID запроса №{QUERY2}: {execution_id}')
    payload['variables']['execution_id'] = execution_id

    while True:
        try:
            response = session.post('https://app-api.dune.com/v1/graphql', json=payload)
            if (response.status_code == 200):
                logger.success(f'База данных №{QUERY2} успешно скачана!')
                break
            else:
                logger.error(f'Ошибка обновления базы данных: {response.text} | Cтатус запроса: {response.status_code}')
        except Exception as error:
            logger.error(f'Ошибка обновления базы данных: {error}')
    
    with open(file_data2, 'w') as file:
        json.dump(response.json(), file)

    logger.success(f'Готово!\n')


def make_table() -> None:
    exists1 = is_exists(file_data1)
    exists2 = is_exists(file_data2)
    if (not exists1 or not exists2):
        logger.info('Файлы баз данных отстутствуют!')
        update_database()

    load_wallets()
    logger.info(f'Загружено {len(WALLETS)} кошельков')
    filtered_wallets1 = get_filtered_wallets(file_data1)
    filtered_wallets2 = get_filtered_wallets(file_data2)
    if (len(filtered_wallets1) == 0):
        logger.error('Не найден ни один кошелек в базе!')
        return
    save_to_excel(filtered_wallets1, filtered_wallets2)
    logger.success('Готово!\n')
    WALLETS.clear()


def get_action() -> str:
    theme = {
        "Question": {
            "brackets_color": "bright_yellow"
        },
        "List": {
            "selection_color": "bright_blue"
        }
    }

    question = [
        inquirer.List(
            "action",
            message=colored("Выберите действие", 'light_yellow'),
            choices=["Обновить базу данных", "Составить Excel таблицу", "Выход"],
        )
    ]
    action = inquirer.prompt(question, theme=loadth(theme))['action']
    return action


def main():
    art = text2art(text="LAYERZERO   STATS", font="standart")
    print(colored(art,'light_blue'))
    print(colored('Автор: t.me/cryptogovnozavod\n','light_cyan'))

    while True:
        action = get_action()

        match action:
            case 'Обновить базу данных':
                update_database()
            case 'Составить Excel таблицу':
                make_table()
            case 'Выход':
                exit()
            case _:
                pass


if (__name__ == '__main__'):
    main()