gui.py

# -*- coding:utf-8 -*-
from Tkinter import *

import re
from GodTian_Pinyin import GodTian_Pinyin


def shift(event):
    canv.focus_set()


def update():
    global display
    canv.delete(id)
    input.delete('0.0', END)
    display = [str(j + 1) + i for j, i in
               enumerate(result[5 * (page - 1):5 * page if len(result) > 5 * page else len(result)])]
    input.insert(END, "\t".join(display))
    return canv.create_text(20, 50, text=input.get("0.0", END), anchor=W, width=300)


def shurufa(event):
    command = event.keysym
    m1 = re.match(r'[a-z\']', event.char)   # True / False
    m2 = re.match(r'\d',event.char)
    global py, result, a, id, page, display
    print(len(py))
    if m1:  # 输入的字母
        py.append(m1.group())
        var.set("".join(py))
        if "".join(py) not in godtian.cache:
            hz, two_part = godtian.handle_current_input("".join(py), 15, 15)
            godtian.cache["".join(py)] = hz
        else:
            hz = godtian.cache["".join(py)]
        # hz, two_part = godtian.handle_current_input("".join(py), 15, 5)
        result = ["".join(i.path) for j, i in enumerate(hz)]
        id = update()

    elif command == 'Return':
        if len(py) > 0:
            if "".join(py) not in godtian.cache:
                hz, two_part = godtian.handle_current_input("".join(py), 15, 15)
                godtian.cache["".join(py)] = hz
            else:
                hz = godtian.cache["".join(py)]
            # hz, two_part = godtian.handle_current_input("".join(py), 15, 15)
            result = ["".join(i.path) for j, i in enumerate(hz)]
            id = update()

    elif command == "Down":
        if len(result)> page*5:
            page+=1
            id = update()

    elif command == "Up":
        if page > 1:
            page -= 1
            id = update()

    elif command == 'BackSpace':
        if len(py) > 0:
            py.pop(-1)
            var.set("".join(py))
            if "".join(py) not in godtian.cache:
                hz, two_part = godtian.handle_current_input("".join(py), 15, 15)
                godtian.cache["".join(py)] = hz
            else:
                hz = godtian.cache["".join(py)]
            result = ["".join(i.path) for j, i in enumerate(hz)]
            id = update()
    elif command in ['F5']:
        godtian.save_memo_and_cache()

    elif command == 'space':
        if len(py) > 0:
            num = 1
            if len(display) >= num:
                var.set("")
                text.insert(END, display[num - 1][1:])
                display = []
                py = []
                result = []
                canv.delete(id)
    elif m2:
        num = int(event.char)
        if len(display) >= num:
            var.set("")
            text.insert(END, display[num-1][1:])
            display = []
            py = []
            result = []
            canv.delete(id)
    else:
        pass

if __name__ == '__main__':
    id=-1
    page=1
    py=[]
    result = []
    display=[]
    godtian = GodTian_Pinyin()
    root = Tk()
    root.title("天神输入法")
    var = StringVar()

    text=Text(root,yscrollcommand=1)
    text.pack(side=TOP, expand=YES, fill=X)
    text.bind("<Shift_L>",shift)

    label=Label(root, textvariable=var)
    label.pack(side=BOTTOM, fill=X)

    canv = Canvas(root, width=400, height=100)
    canv.pack(side=BOTTOM, expand=YES, fill=X)

    input = Text(canv)
    canv.focus_set()
    canv.bind("<Key>", shurufa)
    root.mainloop()