1# coding=utf-8
2
3import sys
4import json
5import base64
6
7
8# 保证兼容python2以及python3
9IS_PY3 = sys.version_info.major == 3
10if IS_PY3:
11    from urllib.request import urlopen
12    from urllib.request import Request
13    from urllib.error import URLError
14    from urllib.parse import urlencode
15    from urllib.parse import quote_plus
16else:
17    import urllib2
18    from urllib import quote_plus
19    from urllib2 import urlopen
20    from urllib2 import Request
21    from urllib2 import URLError
22    from urllib import urlencode
23
24# 防止https证书校验不正确
25import ssl
26ssl._create_default_https_context = ssl._create_unverified_context
27
28API_KEY = 'AK'
29
30SECRET_KEY = 'SK'
31
32
33IMAGE_RECOGNIZE_URL = "https://aip.baidubce.com/rest/2.0/image-classify/v2/dish"
34
35
36"""  TOKEN start """
37TOKEN_URL = 'https://aip.baidubce.com/oauth/2.0/token'
38
39
40"""
41    获取token
42"""
43def fetch_token():
44    params = {'grant_type': 'client_credentials',
45              'client_id': API_KEY,
46              'client_secret': SECRET_KEY}
47    post_data = urlencode(params)
48    if (IS_PY3):
49        post_data = post_data.encode('utf-8')
50    req = Request(TOKEN_URL, post_data)
51    try:
52        f = urlopen(req, timeout=5)
53        result_str = f.read()
54    except URLError as err:
55        print(err)
56    if (IS_PY3):
57        result_str = result_str.decode()
58
59    result = json.loads(result_str)
60
61    if ('access_token' in result.keys() and 'scope' in result.keys()):
62        if not 'brain_all_scope' in result['scope'].split(' '):
63            print ('please ensure has check the  ability')
64            exit()
65        return result['access_token']
66    else:
67        print ('please overwrite the correct API_KEY and SECRET_KEY')
68        exit()
69
70"""
71    读取文件
72"""
73def read_file(image_path):
74    f = None
75    try:
76        f = open(image_path, 'rb')
77        return f.read()
78    except:
79        print('read image file fail')
80        return None
81    finally:
82        if f:
83            f.close()
84
85"""
86    调用远程服务
87"""
88def request(url, data):
89    req = Request(url, data.encode('utf-8'))
90    has_error = False
91    try:
92        f = urlopen(req)
93        result_str = f.read()
94        if (IS_PY3):
95            result_str = result_str.decode()
96        return result_str
97    except  URLError as err:
98        print(err)
99
100"""
101    调用菜品识别接口并打印结果
102"""
103def print_result(filename, url):
104    # 获取图片
105    file_content = read_file(filename)
106
107    response = request(url, urlencode(
108        {
109            'image': base64.b64encode(file_content),
110            'top_num': 1
111        }))
112    result_json = json.loads(response)
113
114    # 打印图片结果
115    for data in result_json["result"]:
116        print(u"  菜品名称: " + data["name"])
117        if data[u'has_calorie']:
118            print(u"  菜品热量: " + data["calorie"])
119
120if __name__ == '__main__':
121
122    # 获取access token
123    token = fetch_token()
124
125    # 拼接图像识别url
126    url = IMAGE_RECOGNIZE_URL + "?access_token=" + token
127
128    # 菜品图1
129    print("菜品1")
130    print_result("./food1.jpg", url)
131
132    # 菜品图3
133    print("菜品2")
134    print_result("./food2.jpg", url)
135
136    # 菜品图3
137    print("菜品3")
138    print_result("./food3.jpg", url)

1菜品1
2  菜品名称: 白灼菜心
3  菜品热量: 35
4菜品2
5  菜品名称: 香酥鲫鱼
6  菜品热量: 108
7菜品3
8  菜品名称: 猪排
9  菜品热量: 231