cyberbot-server/cyber/keyboard_listener.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Contact :   liuyuqi.gov@msn.cn
@Time    :   2022/10/05 06:19:13
@License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
@Desc    :   按键监听
'''

import keyboard
import time
from screen_shot import ScreenCapture
import io


class KeyboardListener:
    def __init__(self, tcpServer):
        self.tcpServer = tcpServer
        self.t = 0
        self.c = 0
        self.key_state_map = {}
        self.screen_capture = None

    def listen_keyboard(self, callback):
        self.callback = callback
        keyboard.hook(self.onKeyEvent)
        keyboard.wait()

    def onImgCapture(self, pic):
        imgByteArr = io.BytesIO()
        pic.save(imgByteArr, format='JPEG')
        bytes_data = imgByteArr.getvalue()
        self.tcpServer.send_img(bytes_data)

    def isCtrlHolding(self):
        return ('ctrl' in self.key_state_map and self.key_state_map['ctrl'] == 'down')\
            or ('left ctrl' in self.key_state_map and self.key_state_map['left ctrl'] == 'down')\
            or ('right ctrl' in self.key_state_map and self.key_state_map['right ctrl'] == 'down')

    def isAltHolding(self):
        return ('alt' in self.key_state_map and self.key_state_map['alt'] == 'down')\
            or ('left alt' in self.key_state_map and self.key_state_map['left alt'] == 'down')\
            or ('right alt' in self.key_state_map and self.key_state_map['right alt'] == 'down')

    def isKeyHolding(self, key):
        return (key in self.key_state_map and self.key_state_map[key] == 'down')

    def onKeyEvent(self, key):
        # update key_state_map
        self.key_state_map[key.name.lower()] = key.event_type

        # is screenshoot?

        if self.isKeyHolding("caps lock")\
                and key.event_type == "down"\
                and key.name.lower() == "a":
            self.screen_capture = ScreenCapture()
            self.screen_capture.are_capture(self.onImgCapture)

        print(self.key_state_map)
        # is triple c?
        if key.event_type == "down" \
                and key.name.lower() == "c" \
                and self.isCtrlHolding():

            if self.t == 0:
                self.t = time.time()
                self.c += 1
                print("wait for nex c", self.c)
                return

            if (time.time()-self.t < 0.5):
                self.t = time.time()
                self.c += 1
                print("wait for nex c:", self.c)

            else:
                self.c = 0
                self.t = 0
                print("wait for nex c", self.c)

            if self.c >= 2:
                self.c = 0
                print("need trans")
                if self.callback:
                    self.callback()