Report_Generate_Server/tools/Get_Json.py

import requests
import json
from tools.API import *

def get_data(url : str, data_type : str = "data", params : dict = None ) -> dict:
    headers = {
        "content_type": "application/x-www-form-urlencoded"
    }
    try:
        response = requests.get(url, headers=headers, params=params)
        if response.status_code == 200:
            data = json.loads(response.text)
            print(f"获取到数据：{data}")
            return data[data_type]
        else:
            print(f"请求数据失败，状态码：{response.status_code}")
            return None
    except Exception as e:
        print(f"请求数据失败，异常：{e}")
        return None

def get_project_info(projectId : str) -> dict:
    projecturl = DTURL + GETPROJECTINFO.format(projectId=projectId)
    return get_data(projecturl, "data")
    
def get_jizu_info(turbineId : str) -> dict:
    jizuurl = DTURL + GETJIZUINFO.format(turbineId=turbineId)
    return get_data(jizuurl, "data")

def get_jizu_shigong_info(turbineId : str) -> dict:
    jizuurl = DTURL + GETSHIGONGINFO
    params = {
        "rows": {
            "turbineId": turbineId
        }
    }
    result = get_data(jizuurl, "rows", params)
    return result[0] if result else None

def get_weather(weatherid : str) -> dict:
    weatherurl = DTURL + GETWEATHERINFO.format(weatherCode=weatherid)
    return get_data(weatherurl, "data")

def get_part_list(turbineId : str) -> dict:
    parturl = DTURL + GETPARTLIST
    params = {
        "turbineId" : turbineId
    }
    result = get_data(parturl, params = params)
    print(f"获取到部件{result}")
    return result, [item["partId"] for item in result]



def get_part_picture(turbineId : str) -> tuple[list[dict], list[dict]]:
    """获取对应机组所有图片
    
    Args:
        turbineId (str): 机组ID
    Return:
        list: 包含所有对应机组的部件信息的列表
        
        list: 包含所有图片信息的列表(按部件分list)
        [
            {
                "imageId": "图片id",
                "imageName": "图片名",
                "imagePath": "图片路径",
                "partId": "部件id",
                "partName": "部件名",
                "imageResolution": "字符串",
                "focalDistance": "字符串/None",
                "shootingTime": "字符串/None",
                "cameraManufacturer": "字符串/None",
                "cameraModel": "字符串/None",
                "weather": "字符串/None",
                "weatherLabel": "字符串/None",
                "humidness": "数值/None",
                "temperature": "字符串/None",
                "windLevel": "数值/None",
                "shootingMethod": "字符串/None",
                "shootingMethodLabel": "字符串/None",
                "shootingDistance": "数值/None",
                "collectorName": "字符串/None",
                "imageType": "None",
                "imageTypeLabel": "None",
                "audioList": "None",
                "gps": "字符串/None"
            },
            ...
        ]
    """
    picturerul = DTURL + GETPICTURELIST
    part_data, part_list = get_part_list(turbineId)
    result = []
    for part_id in part_list:
        params = {
            "partId" : part_id,
        }
        for images in get_data(picturerul, params = params):
            result.append(images)
    print(f"图片数据获取成功：{result}")
    return part_data, result
    
def get_yepian_xiangqing(yepian_id : str) -> dict:
    url = GETYEPIANINFO.format(partId=yepian_id)
    return get_data(DTURL + url, "data")
    
def find_defect_record(defect_pictures : list[dict]):
    """查找缺陷图片的缺陷记录

    Args:
        defect_pictures (list[dict]): 缺陷图列表
    """