main.py

import keyboard
import pyautogui
import threading
import time

import utility
import atomic_game_actions as actions
from ShopItems import ShopItems


# Main thread state

shop_item_to_buy = ShopItems.FRESHWATER_FRIENDS
running = False
main_thread = None


def change_shop_item(shop_item):
    global shop_item_to_buy
    shop_item_to_buy = shop_item
    print(f"Shop item to buy set to: {shop_item_to_buy}")


def change_running_state(next_state):
    global running, main_thread

    running = next_state

    if running and main_thread == None:
        print("Starting main logic")
        main_thread = threading.Thread(target=main)
        main_thread.start()
    elif running and main_thread:
        print("Main logic will continue indefinitely again")
    elif not running and main_thread == None:
        print("No actively running main logic thread found to close")
    elif not running and main_thread:
        print("Main logic will stop gracefully after it finishes its current iteration")


def main():
    global main_thread
    while running:
        try:
            actions.close_any_popup_windows()
            actions.open_shop()
            actions.open_shop_fish_tab()
            actions.buy_max_fish(shop_item_to_buy)
            print("Closing shop")
            actions.close_any_popup_windows()
            actions.open_all_packs()
            actions.put_fish_into_tank()
            actions.open_current_tank()
            actions.click_color_tank_filter()
            actions.filter_golden_and_rainbow_to_other_tank()
            actions.click_color_tank_filter()
            actions.filter_golden_and_rainbow_to_other_tank()
            actions.sell_all_unlocked_fish()
            pyautogui.moveTo(utility.get_window_centre_point())
            time.sleep(1)
        except:
            print("... Main logic failed")
        utility.print_help()
    main_thread = None
    print("Main thread stopped gracefully")


# Registering keyboard events
keyboard.on_press_key("F1", lambda e: change_shop_item(ShopItems.FRESHWATER_FRIENDS))
keyboard.on_press_key("F2", lambda e: change_shop_item(ShopItems.RIVERS_AND_PONDS))
keyboard.on_press_key("F3", lambda e: change_shop_item(ShopItems.REEF_FELLAS))
keyboard.on_press_key("F4", lambda e: change_shop_item(ShopItems.MARINE_DWELLERS))
keyboard.on_press_key("F5", lambda e: change_shop_item(ShopItems.GIANTS))
keyboard.on_press_key("F6", lambda e: change_shop_item(ShopItems.SPRING_PALS))
keyboard.on_press_key("F11", lambda e: change_running_state(not running))

utility.print_help()

# Keep the program running and listening
keyboard.wait()