from urllib import request

 import base64

 import requests

 import re

 import json

 import urllib

 import os

 # 获取token

 def get_token(client_id,client_secret):

     host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id='+ client_id + '&client_secret='+ client_secret + ''

     headers = {'Content-Type': 'application/json; charset=UTF-8'}

     res = requests.post(host, headers=headers)

     access_token = re.findall('"access_token":"(.*?)"', res.text)[0]

     return access_token

 # 打开文件夹,进行对比

 def geturlPath(path):

     dirs = os.listdir(path)

     lst = []

     for dir1 in dirs:

         pa = path + dir1

         lst.append(pa)

     return lst

 # 打开本地图片,并转化为base64

 def open_pic2base(image):

     f = open(image,'rb')

     img = base64.b64encode(f.read()).decode('utf-8')

     return img

 def search(image3,client_id,client_secret,groups):

     request_url = "https://aip.baidubce.com/rest/2.0/face/v3/search"

     for image in image3:

         a = open_pic2base(image)

         params =json.dumps({"image":a,"image_type":"BASE64","group_id_list":groups})

         access_token = get_token(client_id, client_secret)

         request_url = request_url + "?access_token=" + access_token

         params = params.encode("utf-8")

         request = urllib.request.Request(url=request_url, data=params)

         request.add_header('Content-Type', 'application/json')

         response = urllib.request.urlopen(request)

         content = response.read()

         user = 'not find'

         if content:

             result = json.loads(content.decode('utf-8'))

             print(result)

             if result['error_code']==0:

                 lst = []

                 sets = {}

                 user=result['result']['user_list'][0]

                 group_id = user["group_id"]

                 user_id=user['user_id']

                 score = user['score']

                 sets["所在组为"] = group_id

                 sets["相似用户为"] = user_id

                 sets["相似得分为"] = score

                 lst.append(sets)

                 print(lst)

                 if score > 85:

                     add_face2ku = face_add(image,group_id,user_id)

                 else:

                   pass

 def face_add(image,group_id,user_id):

     request_url = 'https://aip.baidubce.com/rest/2.0/face/v3/faceset/user/add'

     params = json.dumps({"image":open_pic2base(image),"image_type":"BASE64","group_id":group_id,"user_id":user_id})

     access_token = get_token(client_id, client_secret)

     request_url = request_url + "?access_token=" + access_token

     params = params.encode("utf-8")

     request = urllib.request.Request(url=request_url, data=params)

     request.add_header('Content-Type', 'application/json')

     response = urllib.request.urlopen(request)

     return response

 if __name__ == '__main__':

     client_id = '*********************'

     client_secret = '*********************'

     path = r'G:/images5/'

     image3 = geturlPath(path)

     # 将图片与图片库进行对比,传入指定的组

     search(image3, client_id, client_secret, 'actor')