main.py

# -*- coding:utf-8 -*-


"""

    问答助手~

"""
import textwrap
import time
from argparse import ArgumentParser
from pyhooked import Hook, KeyboardEvent, MouseEvent
import win32gui

from config import data_directory, hanwan_appcode,app_name,vm_name,search_engine
from core.windows import analyze_current_screen_text
from core.hanwanocr import get_text_from_image

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

def parse_args():
    parser = ArgumentParser(description="Assistant")
    parser.add_argument(
        "-t", "--timeout",
        type=int,
        default=5,
        help="default http request timeout"
    )
    return parser.parse_args()


def main():
    print('我来识别这个题目是啥!!!')
    args = parse_args()
    timeout = args.timeout

    start = time.time()
    text_binary = analyze_current_screen_text(
    	label=vm_name,
        directory=data_directory
    )
    keyword = get_text_from_image(
        image_data=text_binary,
        appcode=hanwan_appcode
    )
    if not keyword:
        print("没识别出来，随机选吧!!!\n")
        print("题目出现的时候按F2，我就自动帮你去搜啦~\n")
        return

    keyword = keyword.split(r"．")[-1]
    keywords = keyword.split(" ")
    keyword = "".join([e.strip("\r\n") for e in keywords if e])
    keyword.replace('\n',' ')
    if len(keyword) < 2:
        print("没识别出来，随机选吧!!!\n")
        print("题目出现的时候按F2，我就自动帮你去搜啦~\n")
        return
    print("我用关键词:\" ", keyword,"\"去百度答案啦!")
    
    elem = browser.find_element_by_id("kw")
    elem.clear()
    elem.send_keys(keyword)
    elem.send_keys(Keys.RETURN)

    print("结果在浏览器里啦~\n")
    print("题目出现的时候按F2，我就自动帮你去搜啦~\n")

def handle_events(args):
    if isinstance(args, KeyboardEvent):
        if args.current_key == 'F2' and args.event_type == 'key down':
            main()
        elif args.current_key == 'Q' and args.event_type == 'key down':
            hk.stop()
            print('退出啦~')


if __name__ == "__main__":
    browser = webdriver.Chrome(r'.\tools\chromedriver.exe')
    browser.get(search_engine)
    hld = win32gui.FindWindow(None, vm_name)
    if hld > 0:
        print('用模拟器打开对应应用~~\n题目出现的时候按F2，我就自动帮你去搜啦~\n')
        hk = Hook()  
        hk.handler = handle_events  
        hk.hook()  
    else:
        print('咦，你没打开'+vm_name+'吧!')