getQuestionAnswer.py

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

# @Author  : Skye
# @Time    : 2018/1/8 20:38
# @desc    : 答题闯关辅助，截屏 ，OCR 识别，百度搜索

import wda
from PIL import Image
from common import  ocr, methods
from threading import Thread
import os


# c = wda.Client()


isTrue = True

while isTrue:

    # 截图
    # c.screenshot('screenshot.png')

    # os.system("screencapture -R\"20,150,310,310\" ./screenshot.png")
    

    # 西瓜视频
    # os.system("screencapture -R\"20,150,310,100\" ./question_screenshot.png")
    # os.system("screencapture -R\"20,250,310,200\" ./choices_screenshot.png")

    os.system("screencapture -R\"20,190,310,100\" ./question_screenshot.png")
    os.system("screencapture -R\"20,320,310,180\" ./choices_screenshot.png")

    question_img = Image.open("./question_screenshot.png")
    choices_img = Image.open("./choices_screenshot.png")
    
    # 文字识别
    question, choices = ocr.ocr_img(question_img, choices_img)

    print("问题%s \n选择:%s"%(question, choices))
    # 用不同方法输出结果，取消某个方法在前面加上#

    # 打开浏览器方法搜索问题
    # methods.run_algorithm(0, question, choices) 
    # # 将问题与选项一起搜索方法，并获取搜索到的结果数目
    # methods.run_algorithm(1, question, choices)
    # # 用选项在问题页面中计数出现词频方法
    # methods.run_algorithm(2, question, choices)

    Thread(methods.run_algorithm(0, question, choices)).start()
    Thread(methods.run_algorithm(1, question, choices)).start()
    Thread(methods.run_algorithm(2, question, choices)).start()
    # m1.start()
    # m2.start()
    # m3.start()


    go = input('输入回车继续运行,输入 n 回车结束运行: ')
    if go == 'n':
        break

    print('------------------------')

    # isTrue = False