wabbajack autoclicker.py

import pyautogui
import keyboard
import time
import psutil

def get_available_networks():
    """
    Returns a list of available network interfaces that are up and running.
    """
    # Get the addresses and stats of all network interfaces
    addresses = psutil.net_if_addrs()
    stats = psutil.net_if_stats()

    available_networks = []
    for intface, addr_list in addresses.items():
        # Skip link-local addresses
        if any(getattr(addr, 'address').startswith("169.254") for addr in addr_list):
            continue
        # Include the interface only if it is up
        elif intface in stats and getattr(stats[intface], "isup"):
            available_networks.append(intface)

    return available_networks

def choose_network_interface(available_networks):
    """
    Displays available network interfaces for the user to choose from
    and returns the chosen network interface.
    """
    print('Select network interface to monitor:')
    for i, intface in enumerate(available_networks):
        print(f"{i} - {intface}")
    chosen_intface = int(input())
    return available_networks[chosen_intface]

def measure_network_download_rate(inf):
    """
    Measures current network download speed
    """
    net_stat = psutil.net_io_counters(pernic=True, nowrap=True)[inf]
    net_in_1 = net_stat.bytes_recv
    time.sleep(1)
    net_stat = psutil.net_io_counters(pernic=True, nowrap=True)[inf]
    net_in_2 = net_stat.bytes_recv
    net_in = round((net_in_2 - net_in_1) / 1024 / 1024, 3)
    print(f"Current net download: {net_in} MB/s")
    return net_in

def click_on_image(image_path):
    """
    Searches for an image on screen and clicks on it if found.
    """
    buttonloc = pyautogui.locateOnScreen(image_path)
    if buttonloc is None:
        print(f"Couldn't find the download button!")
    else:
        center = pyautogui.center(buttonloc)
        oldpos = pyautogui.position()
        print(f"Found the download button, clicking on [{center[0]}, {center[1]}].")
        pyautogui.click(x=center[0], y=center[1])
        pyautogui.moveTo(oldpos)

def main():
    available_networks = get_available_networks()
    inf = choose_network_interface(available_networks)
    print(f"{inf} selected.")

    input('Move this console Window so that it doesn\'t obstruct the download button. Then press Enter.')
    
    while True:
        # Break loop when CTRL+SHIFT+E is pressed
        if keyboard.is_pressed('ctrl') and keyboard.is_pressed('shift') and keyboard.is_pressed('e'):
            break
            
        # Measure current network download speed
        net_in = measure_network_download_rate(inf)

        # If download speed is less than 1 MB/s, search for a slow download button image and click on it
        if net_in < 1:        
            sleep_timer = 8
            print('Searching for an image ...')
            click_on_image('slow download button.png')
        # If download speed is greater than 1 MB/s, set a shorter sleep timer to start over
        else:
            sleep_timer = 0.5

        # Sleep and start over
        print('Sleeping now. You can end this script by holding CTRL+SHIFT+E.')
        time.sleep(sleep_timer)

if __name__ == "__main__":
    main()