0.背景
每逢周六,如果手头项目不紧,有空闲的话,大家会酝酿一个weekend的小项目, 这个周六下午,大家聊起了"头脑王者"这个微信小程序,大家讨论的不亦乐乎,忽然有人提议为这款游戏做一款问题查询辅助类的程序,算不上外挂,姑且叫助手之类的吧.
离下班还有一个小时,于是乎大家开干......
Screen Shot 2018-01-28 at 00.05.38.png
1.思路
既然开始了,马上就有了思路:
- 截取手机的屏幕 ,获取题目的图片
- 裁剪图片,获取题目部分的图片
- 图片通过OCR识别成文字,这个就是题目关键字
- 把题目关键字通过搜索引擎查询,把查询结果显示出来
截取手机的屏幕 ,获取题目的图片
裁剪后的图片,只有题目
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.
