bot.py

import sys

sys.dont_write_bytecode = True

from smart_airdrop_claimer import base
from core.token import get_token
from core.info import get_info
from core.task import process_do_task, process_claim_ref
from core.upgrade import process_upgrade_character, process_upgrade_speed
from core.farm import process_farming

import time


class PitchTalk:
    def __init__(self):
        # Get file directory
        self.data_file = base.file_path(file_name="data.txt")
        self.config_file = base.file_path(file_name="config.json")

        # Initialize line
        self.line = base.create_line(length=50)

        # Initialize banner
        self.banner = base.create_banner(game_name="PitchTalk")

        # Get config
        self.auto_do_task = base.get_config(
            config_file=self.config_file, config_name="auto-do-task"
        )

        self.auto_claim_ref = base.get_config(
            config_file=self.config_file, config_name="auto-claim-ref"
        )

        self.auto_upgrade_character = base.get_config(
            config_file=self.config_file, config_name="auto-upgrade-character"
        )

        self.auto_upgrade_speed = base.get_config(
            config_file=self.config_file, config_name="auto-upgrade-speed"
        )

        self.auto_farm = base.get_config(
            config_file=self.config_file, config_name="auto-farm"
        )

    def main(self):
        while True:
            base.clear_terminal()
            print(self.banner)
            data = open(self.data_file, "r").read().splitlines()
            num_acc = len(data)
            base.log(self.line)
            base.log(f"{base.green}Number of accounts: {base.white}{num_acc}")

            for no, data in enumerate(data):
                base.log(self.line)
                base.log(f"{base.green}Account number: {base.white}{no+1}/{num_acc}")

                try:
                    token = get_token(data=data)

                    if token:

                        get_info(data=data, token=token)

                        # Do task
                        if self.auto_do_task:
                            base.log(f"{base.yellow}Auto Do Task: {base.green}ON")
                            process_do_task(data=data, token=token)
                        else:
                            base.log(f"{base.yellow}Auto Do Task: {base.red}OFF")

                        # Claim ref
                        if self.auto_claim_ref:
                            base.log(f"{base.yellow}Auto Claim Ref: {base.green}ON")
                            process_claim_ref(data=data, token=token)
                        else:
                            base.log(f"{base.yellow}Auto Claim Ref: {base.red}OFF")

                        # Upgrade character
                        if self.auto_upgrade_character:
                            base.log(
                                f"{base.yellow}Auto Upgrade Character: {base.green}ON"
                            )
                            process_upgrade_character(data=data, token=token)
                        else:
                            base.log(
                                f"{base.yellow}Auto Upgrade Character: {base.red}OFF"
                            )

                        # Upgrade speed
                        if self.auto_upgrade_speed:
                            base.log(f"{base.yellow}Auto Upgrade Speed: {base.green}ON")
                            process_upgrade_speed(data=data, token=token)
                        else:
                            base.log(f"{base.yellow}Auto Upgrade Speed: {base.red}OFF")

                        # Farm
                        if self.auto_farm:
                            base.log(f"{base.yellow}Auto Farm: {base.green}ON")
                            process_farming(data=data, token=token)
                        else:
                            base.log(f"{base.yellow}Auto Farm: {base.red}OFF")

                        get_info(data=data, token=token)

                    else:
                        base.log(f"{base.red}Token not found! Please get new query id")
                except Exception as e:
                    base.log(f"{base.red}Error: {base.white}{e}")

            print()
            wait_time = 60 * 60
            base.log(f"{base.yellow}Wait for {int(wait_time/60)} minutes!")
            time.sleep(wait_time)


if __name__ == "__main__":
    try:
        pitchtalk = PitchTalk()
        pitchtalk.main()
    except KeyboardInterrupt:
        sys.exit()