为头脑王者微信小程序做一个外挂

0.背景

每逢周六,如果手头项目不紧,有空闲的话,大家会酝酿一个weekend的小项目, 这个周六下午,大家聊起了"头脑王者"这个微信小程序,大家讨论的不亦乐乎,忽然有人提议为这款游戏做一款问题查询辅助类的程序,算不上外挂,姑且叫助手之类的吧.
离下班还有一个小时,于是乎大家开干......

Screen Shot 2018-01-28 at 00.05.38.png

1.思路

既然开始了,马上就有了思路:

1. 截取手机的屏幕 ,获取题目的图片
2. 裁剪图片,获取题目部分的图片
3. 图片通过OCR识别成文字,这个就是题目关键字
4. 把题目关键字通过搜索引擎查询,把查询结果显示出来

截取手机的屏幕 ,获取题目的图片

裁剪后的图片,只有题目

OCR返回文字 就是题目关键字

Screen Shot 2018-01-28 at 01.00.27.png

通过搜索引擎查询出结果

2.运行

因为是周末项目,只有一个小时的时间,做的比较匆忙,算是抛砖引玉吧.
本来像抓取过来搜索结果,再做正则表达式处理,没想到遇到了百度https加密的小坑,就直接采用webbrowser.open的方式打开.
实测手机出现题目,执行程序大概2秒钟就可以返回查询结果,这个要远远高于你手工的操作,程序自动处理手机截屏,题目关键字的OCR识别,自动打开浏览器并查询关键字,全部自动化了,希望能在"头脑王者"过关斩将中助你一臂之力!

3.总结

作为一个weekend的练手项目,主要是提高我们程序人员的兴趣,扩大大家的知识面,这个提升大家的开发水平有很大帮助.
另外,我们程序猿们不仅有寻找BUG的烦恼,也有享受编程的快乐和生活的乐趣, Hello, World!

两枚程序猿祝你周末愉快!

4.代码

'''
这款头脑王者微信小程序,最近是特别的火,借助微信朋友圈的这把火烧的得更旺了
我们周末的一个weekend小项目,做了一个 头脑王者小助手
'''


import os
from PIL import Image
from aip import AipOcr  # 引入baidu文字识别OCR SDK
from urllib.parse import quote

# 定义常量
APP_ID = '你的APP_ID'
API_KEY = '你的API_KEY'
SECRET_KEY = '你的SECRET_KEY'


# 读取图片
def get_file_content(filePath):
    with open(filePath, 'rb') as fp:
        return fp.read()


'''
适用于安卓手机,通过adb命令截屏,并把图片传给电脑
'''


def screencap():
    mcom = "adb shell /system/bin/screencap -p /sdcard/mscreen.png"
    os.system(mcom)
    mcom = "adb pull /sdcard/mscreen.png ./mscreen.png"
    os.system(mcom)


'''
裁剪图片,只把题目部分的截图保存

'''


def cropimg():
    im = Image.open("mscreen.png")
    img_size = im.size
    print("图片宽度和高度分别是{}".format(img_size))
    x = 50
    y = 400
    w = 650
    h = 200
    region = im.crop((x, y, x+w, y+h))
    region.save("screen.png")
    return "screen.png"


def ocr(imgUrl):
    # 初始化ApiOcr对象
    aipOcr = AipOcr(APP_ID, API_KEY, SECRET_KEY)
    # 调用通用文字识别接口
    result = aipOcr.basicGeneral(get_file_content(imgUrl))

    print(result['words_result_num'])
    s = ''
    for i in range(result['words_result_num']):
        s += result['words_result'][i]['words']
    print(s)
    return s


def baidu(word):
    url = "http://www.baidu.com/s?wd=%s" %  quote(word)
    print(url)
    import webbrowser
    webbrowser.open(url)



if __name__ == "__main__":

    # 通过adb命令截屏,并把图片传给电脑
    #screencap()

    # 裁剪图片,只把题目部分的截图保存
    url = cropimg()

    # 调用baidu通用文字识别接口,返回识别的文字
    s = ocr(url)
    print(s)
    # 打开默认浏览器,百度一下获取的题目内容
    baidu(s)

注意:本代码的百度API需要换成你自己的API参数,这个不费事也就分分钟的事情申请一个就OK.