# -*- coding: utf-8 -*-
'''
PyKeyBoardFairy
Python编写的简单版键盘精灵

Python Flask Django 开心学习交流q群：217840699
Author  : J.sky
Mail    : bosichong@qq.com

'''

__version__ = "0.1.0"
from pynput.keyboard import Key, KeyCode, Controller, Listener
import threading
import time
'''
按键配置说明：
"key_type": "interval",

按键种类：
interval：魔法辅助技能键，间隔一定时间按一次
combination：组合技能，一组按键按照一定顺序和间隔时间的模拟按下(每个技能键只按一次)
always：一直按着不放开的键，中间可以有少量的时间暂停
当'key_type'=always 是 会多处一个时间参数"t1",具体看先边的解释

"key_switch": Key.ctrl,#开关控制键，负责控制模拟这个技能键的开关，按下ctrl才会启动按年模拟，再次按下ctrl模拟暂停
"key": 'b',#需要模拟按下的技能键
"is_start": 0,#开关，确定当前技能键在程序开启时，默认是关闭的，一般为0及可。
"t": 0.5,#当前按键模拟按下离开的间隔时间，以秒为单位
"t1": 5, #当'key_type'=always此属性有效，表示为按键一直按着不松开5秒。

'''
##################################################################
'''
# keyList 为程序的配置文件，是一个包含字典的列表，每一个字典为一个按键配置。
'''
keyList = [
    {
        "key_type": "interval",
        "key_switch": 'z',
        "key": 'b',
        "is_start": 0,
        "t": 0.5,
    },
    {
        "key_type": "always",
        "key_switch": 'z',
        "key": 'c',
        "is_start": 0,
        "t": 0.5,
        "t1": 5
    },
    {
        "key_type": "combination",
        "key_switch": 'shift_r',
        "key": 'd',
        "is_start": 0,
        "t": 0.1,
    },
    {
        "key_type": "combination",
        "key_switch": 'shift_r',
        "key": 'f',
        "is_start": 0,
        "t": 1,
    },
    {
        "key_type": "combination",
        "key_switch": 'shift',
        "key": 'e',
        "is_start": 0,
        "t": 2,
    },
]

PYBF_KEY = [
    'a',
    'b',
    'c',
    'd',
    'e',
    'f',
    'g',
    'h',
    'i',
    'j',
    'k',
    'l',
    'm',
    'n',
    'o',
    'p',
    'q',
    'r',
    's',
    't',
    'u',
    'v',
    'w',
    'x',
    'y',
    'z',
    'A',
    'B',
    'C',
    'D',
    'E',
    'F',
    'G',
    'H',
    'I',
    'J',
    'K',
    'L',
    'M',
    'N',
    'O',
    'P',
    'Q',
    'R',
    's',
    't',
    'u',
    'v',
    'w',
    'x',
    'y',
    'Z',
]
PYBF_OTHER_KEYS = {
    "ctrl":Key.ctrl_l,
    "shift":Key.shift_l,
    "alt":Key.alt_l,
    "ctrl_r":Key.ctrl_r,
    "shift_r":Key.shift_r,
    "alt_r":Key.alt_r,
    "caps_lock":Key.caps_lock,
    "tab":Key.tab,
    "esc":Key.esc,
    "space":Key.space,
    "enter":Key.enter,
    "backspace":Key.backspace,
    "up":Key.up,
    "left":Key.left,
    "right":Key.right,
    "down":Key.down,
}

keyboard = Controller()

def changeList(keylist):
    '''将普通的字符串转换成可识别的按键代码'''
    for k in keylist:
        if k["key_switch"] in PYBF_KEY:
            k["key_switch"] = KeyCode.from_char(k["key_switch"])
        else:
            try:
                k["key_switch"] = PYBF_OTHER_KEYS[k["key_switch"]]
            except KeyError:
                raise("KeyError:请检查'key_switch'是否使用了不存在的键盘按键，或是设置出错！")
                
        # print(k["key_switch"])

def on_press(key):
    ''''
    当开关键被按下，暂停或启动模拟按键
    '''
    for k in keyList:
        if key == k["key_switch"]:
            if k["is_start"] == 0:
                k["is_start"] = 1
            else:
                k["is_start"] = 0
        # print(k["is_start"])


def on_release(key):
    pass


def pressKey(key):
    '''技能键，有间隔时间'''
    while True:
        if key["is_start"]:
            keyboard.press(key["key"])
            keyboard.release(key["key"])
        time.sleep(key["t"])
        # print(keys["is_start"])


def pressKeyAlways(key):
    '''技能键，一直按着的及技能键，中间有少量时间间隔。'''
    while True:
        if key["is_start"]:
            keyboard.press(key["key"])
            time.sleep(key["t1"])
            keyboard.release(key["key"])
        time.sleep(key["t"])


        # print(keys["is_start"])
def pressKeyCombination(key):
    '''
    组合技能，一组按键按照一定顺序和间隔时间的模拟按下
    '''
    while True:
        if key["is_start"]:
            time.sleep(key["t"])
            keyboard.press(key["key"])
            keyboard.release(key["key"])
            key["is_start"] = 0


def createKeyThread(keylist):
    '''
    根据配置文件，创建所有的按键线程。
    '''
    for key in keylist:
        if key["key_type"] == "interval":
            # print(key["key_type"])
            t = threading.Thread(target=pressKey, args=(key, ))
            t.start()
        elif key["key_type"] == "always":
            # print(key["key_type"])
            t = threading.Thread(target=pressKeyAlways, args=(key, ))
            t.start()
        elif key["key_type"] == "combination":
            # print(key["key_type"])
            t = threading.Thread(target=pressKeyCombination, args=(key, ))
            t.start()
        else:
            raise "没有这种类型的按键，请检查并重新设置按键的类型！"


def main():
    changeList(keyList)
    createKeyThread(keyList)

    with Listener(on_press=on_press, on_release=on_release) as listener:
        listener.join()


if __name__ == '__main__':
    main()