UniversalRecorder.py

import re

from pynput import mouse, keyboard
from Event import ScreenWidth as SW, ScreenHeight as SH
import Recorder.globals as globalv

record_signals = globalv.RecordSignal()

buttondic = {mouse.Button.left: 'left',
             mouse.Button.right: 'right',
             mouse.Button.middle: 'middle'
             }

# some keyname recorded by pynput is not supported in pyautogui
renamedic = {'cmd': 'win', 'shift_r': 'shiftright', 'alt_r': 'altright', 'ctrl_r': 'ctrlright',
             'caps_lock': 'capslock', 'num_lock': 'numlock',
             'page_up': 'pageup', 'page_down': 'pagedown', 'print_screen': 'printscreen'}


def get_delay(action_type):
    delay = globalv.current_ts() - globalv.latest_time

    # 录制鼠标轨迹的精度，数值越小越精准，但同时可能产生大量的冗余
    mouse_move_interval_ms = globalv.mouse_interval_ms or 999999

    if action_type == 'mouse move' and delay < mouse_move_interval_ms:
        return -1

    if globalv.latest_time < 0:
        delay = 0
    globalv.latest_time = globalv.current_ts()
    return delay


def get_mouse_event(x, y, action_type):
    tx = x / SW
    ty = y / SH
    tpos = (tx, ty)
    delay = get_delay(action_type)
    if delay < 0:
        return None
    else:
        return globalv.ScriptEvent({
            'delay': delay,
            'event_type': 'EM',
            'action_type': action_type,
            'action': tpos
        })


def on_move(x, y):
    event = get_mouse_event(x, y, 'mouse move')
    if event:
        record_signals.event_signal.emit(event)
        record_signals.cursor_pos_change.emit((x, y))


def on_click(x, y, button, pressed):
    action_type = 'mouse {0} {1}'.format(buttondic[button],
                                     'down' if pressed else 'up')
    event = get_mouse_event(x, y, action_type)
    if event:
        record_signals.event_signal.emit(event)


def on_scroll(x, y, dx, dy):
    action_type = 'mouse wheel {0}'.format('down' if dy < 0 else 'up')
    event = get_mouse_event(x, y, action_type)
    if event:
        record_signals.event_signal.emit(event)


def get_keyboard_event(key, action_type):
    #TODO: maybe can helo https://blog.csdn.net/haiyangdaozhang/article/details/109158793
    delay = get_delay(action_type)
    if delay < 0:
        return None
    else:
        try:
            keycode = key.value.vk
            keyname = renamedic.get(key.name, key.name)
        except AttributeError:
            keycode = key.vk
            keyname = key.char
        if keyname is None:
            return None
        if re.match('^([0-9])$', keyname) and keycode is None:
            keyname = 'num{}'.format(keyname)
        event = globalv.ScriptEvent({
            'delay': delay,
            'event_type': 'EK',
            'action_type': action_type,
            'action': (keycode, keyname, 0)
        })
        return event


def on_press(key):
    event = get_keyboard_event(key, 'key down')
    if event:
        record_signals.event_signal.emit(event)


def on_release(key):
    event = get_keyboard_event(key, 'key up')
    if event:
        record_signals.event_signal.emit(event)


def setuphook(commandline=False):
    if not commandline:
        mouselistener = mouse.Listener(
            on_move=on_move,
            on_scroll=on_scroll,
            on_click=on_click
        )
        mouselistener.start()
    keyboardlistener = keyboard.Listener(
        on_press=on_press,
        on_release=on_release
    )
    keyboardlistener.start()