main.py

import json
import random
from urllib.parse import parse_qs, unquote
from colorama import Fore, Style, init
import time
from tomarket import Tomarket, print_timestamp
import sys

def load_credentials():
    try:
        with open('query_id.txt', 'r') as f:
            queries = [line.strip() for line in f.readlines()]
        return queries
    except FileNotFoundError:
        print("File query_id.txt tidak ditemukan.")
        return [  ]
    except Exception as e:
        print("Terjadi kesalahan saat memuat token:", str(e))
        return [  ]

def parse_query(query: str):
    parsed_query = parse_qs(query)
    parsed_query = {k: v[0] for k, v in parsed_query.items()}
    user_data = json.loads(unquote(parsed_query['user']))
    parsed_query['user'] = user_data
    return parsed_query

def get(id):
        tokens = json.loads(open("tokens.json").read())
        if str(id) not in tokens.keys():
            return None
        return tokens[str(id)]

def save(id, token):
        tokens = json.loads(open("tokens.json").read())
        tokens[str(id)] = token
        open("tokens.json", "w").write(json.dumps(tokens, indent=4))

def generate_token():
    tom = Tomarket()
    queries = load_credentials()
    sum = len(queries)
    for index, query in enumerate(queries):
            parse = parse_query(query)
            user = parse.get('user')
            print_timestamp(f"{Fore.CYAN + Style.BRIGHT}[ Account {index+1}/{sum} {user.get('username','')} ]{Style.RESET_ALL}")
            token = get(user['id'])
            if token == None:
                print_timestamp("Generate token...")
                time.sleep(2)
                token = tom.user_login(query)
                save(user.get('id'), token)
                print_timestamp("Generate Token Done!")
                

def main():
    init(autoreset=True)
    tom = Tomarket()
    auto_task = input("auto clear task y/n  : ").strip().lower()
    auto_game = input("auto play game  y/n  : ").strip().lower()
    auto_combo = input("auto claim combo puzzle y/n : ").strip().lower()
    random_number = input("set random score in game 300-500  y/n  : ").strip().lower()
    free_raffle = input("enable free raffle  y/n  : ").strip().lower()
    # selector_game = input("playing random score game (400-600) y/n ? ").strip().lower()
    used_stars = input("use star for : 1. upgrade rank | 2.auto spin | n.(skip all) (1/2/n): ").strip().lower()
    while True:
        queries = load_credentials()
        sum = len(queries)
        delay = int(3 * random.randint(3700, 3750))
        # generate_token()
        start_time = time.time()
                
        for index, query in enumerate(queries):
            mid_time = time.time()
            total = delay - (mid_time-start_time)
            parse = parse_query(query)
            user = parse.get('user')
            token = get(user['id'])
            if token == None:
                token = tom.user_login(query)
                save(user.get('id'), token)
                time.sleep(2)
            print_timestamp(f"{Fore.CYAN + Style.BRIGHT}[ Account {index+1}/{sum} {user.get('username','')} ]{Style.RESET_ALL}")
            tom.rank_data(token=token, selector=used_stars)
            time.sleep(2)
            tom.claim_daily(token=token)
            time.sleep(2)
            tom.start_farm(token=token)
            time.sleep(2)
            if free_raffle == "y":
                tom.free_spin(token=token, query=query)
            time.sleep(2)
        
        if auto_task == 'y':
            for index, query in enumerate(queries):
                mid_time = time.time()
                total = delay - (mid_time-start_time)
                if total <= 0:
                    break
                parse = parse_query(query)
                user = parse.get('user')
                token = get(user['id'])
                if token == None:
                    token = tom.user_login(query)
                print_timestamp(f"{Fore.CYAN + Style.BRIGHT}[ Account {index+1}/{sum} {user.get('username','')} ]{Style.RESET_ALL}")
                tom.list_tasks(token=token,query=query)
                if auto_combo == 'y':
                    tom.puzzle_task(token, query)
                # tom.rank_data(token=token)
                time.sleep(2)   
                
        if auto_game == 'y':
            for index, query in enumerate(queries):
                mid_time = time.time()
                total = delay - (mid_time-start_time)
                if total <= 0:
                    break
                parse = parse_query(query)
                user = parse.get('user')
                token = get(user['id'])
                if token == None:
                    token = tom.user_login(query)
                print_timestamp(f"{Fore.CYAN + Style.BRIGHT}[ Account {index+1}/{sum} {user.get('username','')} ]{Style.RESET_ALL}")
                tom.user_balance(token=token, random_number=random_number)
                time.sleep(2)

        

        end_time = time.time()
        total = delay - (end_time-start_time)
        hours, remainder = divmod(total, 3600)
        minutes, seconds = divmod(remainder, 60)
        print(f"{Fore.YELLOW + Style.BRIGHT}[ Restarting In {int(hours)} Hours {int(minutes)} Minutes {int(seconds)} Seconds ]{Style.RESET_ALL}")
        if total > 0:
            time.sleep(total)


def start():
    print(r"""
╔════════════ ⋆ TOMARKET BOT ⋆ ═════════════╗
║              Bot Automation                   ║
║         Developed by @iamak_roy               ║
║     Join here : https://t.me/scripthub00      ║
╚═══════════════════════════════════════════╝
                  dark life 
                          
          select this one :
        1. claim daily
        2. generate token
          
          """)
    selector = input("Select the one  : ").strip().lower()

    if selector == '1':
        main()
    elif selector == '2':
        generate_token()
    else:
        exit()

if __name__ == '__main__':
    try:
        start()
    except Exception as e:
        print_timestamp(f"{Fore.RED + Style.BRIGHT}[ {e} ]{Style.RESET_ALL}")
    except KeyboardInterrupt:
        sys.exit(0)