llmacro/playback.py at main · CardboardDog/llmacro · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from pynput import *
from threading import Event
import pickle
inputs = []
settings = None
mouse_controller = None
keyboard_controller = None
sleep_event = None
stopped = False

# callback used to detect stop key
def callback_press_key(key,injected):
    global stopped
    global settings
    if injected:
        return
    if key == settings["stop"]:
        stopped = True
        sleep_event.set()

# takes an input and performs the action
def handle_input(input_event):
    global mouse_controller
    global keyboard_controller
    if input_event[1] == "key_down":
        keyboard_controller.press(input_event[2])
    elif input_event[1] == "key_up":
        keyboard_controller.release(input_event[2])
    elif input_event[1] == "move":
        mouse_controller.position = (input_event[2],input_event[3])
    elif input_event[1] == "button":
        mouse_controller.position = (input_event[2],input_event[3])
        if input_event[5]:
            mouse_controller.press(input_event[4])
        else:
            mouse_controller.release(input_event[4])
    elif input_event[1] == "scroll":
        mouse_controller.position = (input_event[2],input_event[3])
        mouse_controller.scroll(input_event[4],input_event[5])

# start playback
def begin(macro_settings):
    # load settings
    global inputs
    global settings
    global mouse_controller
    global keyboard_controller
    global sleep_event
    global stopped
    settings = macro_settings
    read_inputs = open(settings["file"],"rb")
    inputs = pickle.load(read_inputs)
    read_inputs.close()
    mouse_controller = mouse.Controller()
    keyboard_controller = keyboard.Controller()
    sleep_event = Event()
    stopped = False

    # setup callbacks
    keyboard_listener = keyboard.Listener(on_press = callback_press_key)
    keyboard_listener.start()

    # begin playing
    while not stopped:
        for input_event in inputs:
            sleep_event.wait(input_event[0])
            if stopped:
               return
            handle_input(input_event)
        if not settings["loop"]:
            return