import urllib.request
import urllib.error
import time
import json

def uploadImage(postURL,imgURL):
    #构建 multipart/form-data 请求体
    border = '----------%s' % hex(int(time.time() * 1000))
    postData = []
    postData.append('--%s' % border)
    postData.append('Content-Disposition: form-data; name="%s"\r\n' % 'api_key')
    postData.append('XXXXXXXXXXX') #api_key
    postData.append('--%s' % border)
    postData.append('Content-Disposition: form-data; name="%s"\r\n' % 'api_secret')
    postData.append('XXXXXXXXXXX') #api_secret
    postData.append('--%s' % border)
    fr = open(imgURL, 'rb')
    postData.append('Content-Disposition: form-data; name="%s"; filename=" "' % 'image_file')
    postData.append('Content-Type: %s\r\n' % 'application/octet-stream')
    postData.append(fr.read())
    fr.close()
    postData.append('--%s--\r\n' % border)
    for i, d in enumerate(postData):
        if isinstance(d, str):
            postData[i] = d.encode('utf-8')
    http_body = b'\r\n'.join(postData)
    req = urllib.request.Request(url=postURL, data=http_body)
    req.add_header('Content-Type', 'multipart/form-data; boundary=%s' % border)
    try:
        res = urllib.request.urlopen(req, timeout=5)
        qrcont = res.read()
        resJson = json.loads(qrcont.decode('utf-8'))
        print(resJson['faces'][0]['face_token'])
    except urllib.error.HTTPError as e:
        print(e.read().decode('utf-8'))
        pass


uploadImage(postURL='https://api-cn.faceplusplus.com/facepp/v3/detect',imgURL='./face.jpg')

import cv2
import time
import mediapipe as mp
import RPi.GPIO as GPIO
import requests
import urllib.request
import urllib.error
import datetime
import json
import os

def setup_gpio(pin=20):
    GPIO.setmode(GPIO.BCM)
    GPIO.setup(pin, GPIO.IN)

def cleanup_gpio():
    GPIO.cleanup()

def get_weather(location, key):
    """获取天气信息"""
    url = f'https://devapi.qweather.com/v7/weather/now?location={location}&key={key}'
    response = requests.get(url)
    weather_json = response.json()['now']
    return weather_json['text'], weather_json['feelsLike'], weather_json['temp'], weather_json['windDir'], weather_json['windScale']

def check_body(pin=20):
    """检查人体传感器状态"""
    return GPIO.input(pin)

def face_compare(img_url, api_key, api_secret, face_token):
    """人脸比对，返回相似度和阈值"""
    border = f'----------{hex(int(time.time() * 1000))}'
    post_data = [
        f'--{border}',
        f'Content-Disposition: form-data; name="api_key"\r\n\r\n{api_key}',
        f'--{border}',
        f'Content-Disposition: form-data; name="api_secret"\r\n\r\n{api_secret}',
        f'--{border}',
    ]

    with open(img_url, 'rb') as fr:
        post_data.extend([
            'Content-Disposition: form-data; name="image_file1"; filename=" "',
            'Content-Type: application/octet-stream\r\n',
            fr.read(),
            f'--{border}',
        ])

    post_data.extend([
        f'Content-Disposition: form-data; name="face_token2"\r\n\r\n{face_token}',
        f'--{border}--\r\n',
    ])
    http_body = b'\r\n'.join(item if isinstance(item, bytes) else item.encode('utf-8') for item in post_data)

    req = urllib.request.Request(url='https://api-cn.faceplusplus.com/facepp/v3/compare', data=http_body)
    req.add_header('Content-Type', f'multipart/form-data; boundary={border}')
    try:
        res = urllib.request.urlopen(req, timeout=5)
        res_json = json.loads(res.read().decode('utf-8'))
        return res_json.get('confidence', 0), res_json.get('thresholds', {}).get('1e-5', 100)
    except urllib.error.HTTPError as e:
        print(e.read().decode('utf-8'))
        return 0, 100

def process_face_detection():
    """面部检测和图像捕获"""
    capture = cv2.VideoCapture(0)
    mp_face_mesh = mp.solutions.face_mesh.FaceMesh(max_num_faces=1)
    face_detection_time, videos_time = 0, 0

    while True:
        success, frame = capture.read()
        if not success:
            break
        img_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        results = mp_face_mesh.process(img_rgb)
        faces = []

        if results.multi_face_landmarks:
            for face_lms in results.multi_face_landmarks:
                ih, iw, _ = frame.shape
                face_coords = [(int(lm.x * iw), int(lm.y * ih)) for lm in face_lms.landmark]
                faces.append(face_coords)
        
        videos_time += 1
        if faces:
            videos_time = 0
            face_detection_time += 1
        
        if face_detection_time >= 20:
            cv2.imwrite("./Face.jpg", frame)
            return True
        
        elif videos_time >= 60:
            print('Timeout')
            break

    capture.release()
    return False

def main():
    setup_gpio()
    
    weather, feels_like, temperature, wind_dir, wind_scale = get_weather(location='XXXXXXXXXXX', key='XXXXXXXXXXX')
    push_text = f"早安~ 今天是{datetime.date.today()}，天气{weather}，温度{temperature}°，体感温度{feels_like}°，吹{wind_dir}{wind_scale}级，新的一天要加油~"

    while True:
        if check_body():
            if process_face_detection():
                confidence, thresholds = face_compare(img_url='./Face.jpg', api_key='XXXXXXXXXXX', api_secret='XXXXXXXXXXX', face_token='XXXXXXXXXXX')
                if confidence >= thresholds:
                    print('Success')
                    requests.get(f'https://xxxxx.com/?text=[早安推送]{push_text}')
                    os.remove('./Face.jpg')
                    cleanup_gpio()
                    exit(0)
                else:
                    print('Face Compare Failed')
                    os.remove('./Face.jpg')
                    continue
        time.sleep(1)

if __name__ == "__main__":
    main()