Report_Generate_Server/Jf_report.py

# 文档处理工具
from tools.document_tools import (
    create_document, add_documents,add_table_and_replace,
    add_table_to_document,add_dynamic_table,
    process_server_images_table,add_header,add_landscape_section,
    merge_documents,add_table,add_defect_info_table,add_table_title
)

# 内容处理工具
from tools.content_tools import (
    add_picture,split_table_by_row_content,
    search_and_replace
)

from tools.get_pictures import (
    process_picture_data,get_records_with_pic
)

from tools.Get_Json import (
    get_project_info,get_jizu_info,
    get_jizu_shigong_info,get_defect_detail,
    get_part_picture,get_yepian_xiangqing,
    check_pic_url,get_full_picture_url,
    get_defect_record_list
)

from tools.dataproccess import (
    caculate_work_days,get_year_month,
    merge_info,get_defect_str,get_defect_json,
    safe_get,get_resource_path,get_defedct_info,
    tree_dict_to_table_data,merge_dicts,
    defect_list_addtitle
)

from tools.json_to_docx import json_to_docx, list_to_json_with_merges


from core.tables import fill_tables

from tools.defines import *
import os, re, datetime

async def generate_jf_report(base_info, baogao_info):
    #获取模板编号、模板名称
    num_to_chinese = {1 : '一', 2 : '二', 3 : '三', 4 : '四', 5 : '五', 6 : '六', 7 : '七', 8 : '八', 9 : '九', 10 : '十', 11 : '十一', 12 : '十二'}
    cover_encode = "encode"
    cover_project = "project"
    baogao_name1 = "baogaoname1"
    baogao_name2 = "baogaoname2"
    company_name_yi = "company_name_yi"
    cover_date = "time"
    TITLE_OF_REPORT = "companyencode"
    jiegou_xuhao = 'num'
    
    print(f"获取到参数：基本信息：{base_info}\n\n报告信息：{baogao_info}")

    try:
        base_info = merge_info(base_info, DEFAULT_BASE_INFO)
        turbine_id = base_info['turbine_id']
        
        jizu_data = get_jizu_info(turbine_id)
        project_data = get_project_info(jizu_data['projectId'])
        shigong_data = get_jizu_shigong_info(turbine_id)

        fengchang_name = project_data['farmName']
        Yi_company = project_data['inspectionUnit']
        yi_fuzeren = project_data['inspectionContact']
        yi_phone = project_data['inspectionPhone']
        fengchang_location = project_data['farmAddress']
        Jia_company = project_data['client']
        jia_fuzeren = project_data['clientContact']
        jia_phone = project_data['clientPhone']
        jizu_num = project_data['scale']
        jizu_xinghao = project_data['turbineModel']
        jizu_manufacture = project_data.get('turbineManufacturer', "")
        project_name = project_data['projectName']
        jizu_bianhao = jizu_data["turbineName"]
        start_date = project_data['startDate']
        end_date = project_data['endDate']
        cover_url = project_data['coverUrl']
        gongqi = caculate_work_days(start_date, end_date)
    except Exception as e:
        print(f"数据库的项目-机组基本信息获取失败：{e}")
        return
        
    try:
        baogao_date = datetime.datetime.now().strftime("%Y年%m月%d日 %H:%M")   #现在的时间
        date_year_month = get_year_month(baogao_date)
        
        #前端信息
        baogao_info = merge_info(baogao_info, DEFAULT_BAOGAO_INFO)

        key_words= re.compile('|'.join(map(re.escape, baogao_info['key_words'].split(',')))) #关键字

        shengcheng_dir = baogao_info['shengcheng_dir']     #路径
        if shengcheng_dir == "":
            print("未配置生成路径，请检查配置")
            return
        if_waibu = baogao_info["if_waibu"] 
        if_neibu = baogao_info["if_neibu"]
        if_fanglei = baogao_info["if_fanglei"]
            
        quexian_type = baogao_info['quexian_enum']
        
        dianxing_type = baogao_info['dianxing_enum']

        other_type = baogao_info["other_enum"]

        jiancha_renyuan = baogao_info['jiancha_renyuan'] #检查人员，目前是从命令行参数获取,需要完善
        check_date = baogao_info['check_date']
        if check_date == None:
            check_date = "未获取"
        baogao_bianzhi = baogao_info["userName"]
        baogao_shenghe = baogao_info["baogaoCheck"]
        Jiancha_date = baogao_info["check_date"]
        data_processor = baogao_info["data_processor"]
        coverurl = baogao_info["coverurl"]
        

        #数据库拉取信息
        # Jiancha_date = shigong_data["startTime"].replace("T", " ")   #检查日期
        # image_count = shigong_data['imageCount']  #从施工方案获取的图片数量，待定!!!
        # temperature = shigong_data['temperature']   #温度
        # wind_speed = shigong_data['windSpeed'] #风速
        # weather = get_weather(shigong_data["weatherCode"])   #天气 不从此接口获取，待定！！！
        
        #拉取部件、图片数据
        part_data, picture_data1, picture_data2, Yepians = get_part_picture(turbine_id)
        picture_data = merge_dicts(picture_data1, picture_data2)
        print(Yepians)
        Y1_info = get_yepian_xiangqing(Yepians[0]["partId"])
        Y_Code = [yepian["partCode"] for yepian in Yepians]
        partType = Y1_info["partType"]
        partManufacturer = Y1_info["partManufacturer"]
        print(f"找到叶片号{Y_Code},厂商{partManufacturer}")
        
        image_source_to_find = []
        baogao_label = []
        renyuan_peizhi = []
        gongzuo_neirong = []
        shigong_fangan = []
        shebei_peizhi = []
        beizhu = []
        jiancha = []
        neirong = []
        neirong2 = []
        use_tool_table = []
        table_index = 1
        #获取对应枚举字段
        if if_waibu:
            baogao_label.append("外观")
            image_source_to_find.append(baogao_info['waibu_enum'])
            use_tool_table.extend(
                [ [str(idx)] + item 
                for idx, item in enumerate(USE_TOOL_ENUM.OUT.LIST, start=table_index) ]
            )
            table_index += len(USE_TOOL_ENUM.OUT.LIST)
            if baogao_info["shigong_fangan"] == None:
                print("未传入施工方案,使用已有枚举")
                renyuan_peizhi.append(SHIGONG_FANGAN_ENUM.WAIBU.RENYUAN_PEIZHI)
                gongzuo_neirong.append(SHIGONG_FANGAN_ENUM.WAIBU.GONGZUO_NEIRONG)
                shebei_peizhi.append(SHIGONG_FANGAN_ENUM.WAIBU.SHEBEI_PEIZHI)
                shigong_fangan.append(SHIGONG_FANGAN_ENUM.WAIBU.SHIGONG_FANGAN)
            else:
                pass  #待添加如果从平台传入枚举，但目前没必要，如果这种枚举标准信息库有更好的优化则可以实现
            jiancha.append("无人机外观检查")
            neirong.append(f"、".join(Y_Code) + f"共{len(Y_Code)}支叶片的前缘、后缘、迎风面、背风面。")
            neirong2.append("前缘、后缘、迎风面、背风面。")
        if if_neibu:
            baogao_label.append("内部")
            image_source_to_find.append(baogao_info['neibu_enum'])
            use_tool_table.extend(
                [ [str(idx)] + item 
                for idx, item in enumerate(USE_TOOL_ENUM.IN.LIST, start=table_index) ]
            )
            table_index += len(USE_TOOL_ENUM.IN.LIST)
            if baogao_info["shigong_fangan"] == None:
                renyuan_peizhi.append(SHIGONG_FANGAN_ENUM.NEIBU.RENYUAN_PEIZHI)
                gongzuo_neirong.append(SHIGONG_FANGAN_ENUM.NEIBU.GONGZUO_NEIRONG)
                shebei_peizhi.append(SHIGONG_FANGAN_ENUM.NEIBU.SHEBEI_PEIZHI)
                shigong_fangan.append(SHIGONG_FANGAN_ENUM.NEIBU.SHIGONG_FANGAN)
            else:
                pass
            jiancha.append("叶片内部检查")
            neirong.append(f"、".join(Y_Code) +  f"共{len(Y_Code)}支叶片的内部导雷卡、腹板、透光、人孔盖版、叶根盖板...")
            neirong2.append("内部导雷卡、腹板、透光、人孔盖版、叶根盖板...")
        if if_fanglei:
            baogao_label.append("防雷")
            image_source_to_find.append(baogao_info['fanglei_enum'])
            use_tool_table.extend(
                [ [str(idx)] + item 
                for idx, item in enumerate(USE_TOOL_ENUM.LIGHT.LIST, start=table_index) ]
            )
            table_index += len(USE_TOOL_ENUM.LIGHT.LIST)
            if baogao_info["shigong_fangan"] == None:
                renyuan_peizhi.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.RENYUAN_PEIZHI)
                gongzuo_neirong.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.GONGZUO_NEIRONG)
                shebei_peizhi.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.SHEBEI_PEIZHI)
                shigong_fangan.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.SHIGONG_FANGAN)
            else:
                pass
            jiancha.append("叶片导通测试")
            neirong.append(f"轮毂至塔基导通、内部导线线阻、外部导线线阻...")
            neirong2.append("轮毂至塔基导通、内部导线线阻、外部导线线阻...")
            
        #获取缺陷图列表和典型图列表
        filtered_picture_data, total_picture_num = process_picture_data(picture_data, image_source_to_find)
        #获取所有缺陷记录
        defect_records = get_defect_record_list()
        #获取缺陷记录中对应图片的记录
        defect_records_with_pic, error_pic_with_no_record = get_records_with_pic(defect_records, filtered_picture_data, quexian_type)
        if len(error_pic_with_no_record) > 0:
            print(f"!!!有部分缺陷图片没有对应的缺陷记录，将不会生成这些图片的缺陷表：{error_pic_with_no_record} ")
        print(f"对应缺陷图的缺陷记录列表:{defect_records_with_pic}")
    except Exception as e:
        print(f"报告基本信息获取失败：{e}")
        return

    #检查参数合法性
    if not if_fanglei or not if_neibu or not if_waibu:
        print("请至少选择一种检查项目")
        return
    if not os.path.exists(shengcheng_dir):
        print(f"生成路径{shengcheng_dir}不存在")
        return
    
    baogao_name = "叶片" + "、".join(baogao_label) + "检查报告"
    output_dir = os.path.normpath(f"{shengcheng_dir}/{project_name}项目{baogao_name}{jizu_bianhao}{baogao_date.split(' ')[0]}版.docx")
    
    version = 1
    while os.path.exists(output_dir):
        if version != 1:
            output_dir = output_dir.replace(f"版{version - 1}",f"版{version}")
        else:
            output_dir = output_dir.replace("版",f"版{version}")
        version += 1
    add_list = [
        get_resource_path(MUBAN_DIR + '/jf_fengmian.png'),
        get_resource_path(MUBAN_DIR + '/jf_fengmian1.png'),
        get_resource_path(MUBAN_DIR + '/jinfeng_fengmian_name.docx'),
        get_resource_path(MUBAN_DIR + '/jf_fengmian2.png'),
        get_resource_path(MUBAN_DIR + '/jinfeng_fengmian_renyuan.docx'),
        get_resource_path(MUBAN_DIR + '/jf_fengmian2.png'),
        get_resource_path(MUBAN_DIR + '/jinfeng_fengmian_riqi.docx'),
        get_resource_path(MUBAN_DIR + '/jf_fengmian1.png'),
        get_resource_path(MUBAN_DIR + '/jinfeng_fengmian_luokuan.docx'),
        get_resource_path(MUBAN_DIR + '/jf_fengmian3.png'),
    ]
    print(await create_document(output_dir, section_args={
        "page_height" : JF_HEIGHT,
        "page_width" : JF_WIDTH,
        "left_margin" : JF_L_MARGIN,
        "right_margin" : JF_R_MARGIN,
        "top_margin" : JF_T_MARGIN,
        "bottom_margin" : JF_B_MARGIN,
    }))
    #加封面
    merge_documents(output_dir, add_list)
    #页眉
    add_header(output_dir, TEMPLATE_HEADER.JINFENG_HEADER.ENUM)
    #项目机组信息表格
    print(add_documents(output_dir, get_resource_path(MUBAN_DIR + '/jf_table_title.docx')))
    total_table_num = 0
    print(add_table(output_dir, get_resource_path(MUBAN_DIR + '/jinfeng_table.docx'), REPORT_ENUM=TEMPLATE_HEADER.JINFENG_HEADER.ENUM))
    total_picture_num += 1
    list_to_replace = {
            'jia_company_name' : Jia_company,
            'fengchang_name' : fengchang_name,
            'jizu_hao' : jizu_bianhao,
            'bianzhi_renyuan': baogao_bianzhi,
            'bianzhi_riqi': baogao_date,
            'fengchang_name' : fengchang_name,
            'jizu_hao' : jizu_bianhao,
            'fengchang_location' : fengchang_location,
            'kehu_company' : Yi_company,
            'xiangmuguige' : jizu_num,
            'kehu_fuzeren' : yi_fuzeren,
            'yezhu_phone' : yi_phone,
            'shigong_company' : JINFENG_COMPANY,
            'shigong_date' : "未填写",#shigong_date,
            'shigong_fuzeren' : "未填写",
            'shigong_phone' : "未填写",
            'zhengji_changjia' : jizu_manufacture,
            'yepian_changjia' : partManufacturer,
            'jizu_type' : jizu_xinghao,
            'yepian_type' : partType,
            'jiancha_fangshi' : '、'.join(jiancha),
            'jiancha_renyuan' : jiancha_renyuan
        }
    for find_text, replace_text in list_to_replace.items():
        print(search_and_replace(output_dir, find_text, replace_text))
    print(f"静态内容生成完毕，开始生成动态内容")

    #缺陷信息表格1
    defect_info, defect_part_type_list = get_defedct_info(defect_records_with_pic)

    total_picture_num = add_defect_info_table(output_dir, defect_info, get_resource_path(MUBAN_DIR + '/quexian_liebiao.docx'), total_table_num,TEMPLATE_HEADER.JINFENG_HEADER.ENUM)

    #缺陷信息表格2


    table_list = tree_dict_to_table_data(defect_part_type_list)
    table_list = defect_list_addtitle(table_list, jizu_bianhao)
    print(table_list)
    add_table_title(output_dir, DEFECT_TABLE_TITLE)
    table_json = list_to_json_with_merges(table_list, style_config=STYLE_CONFIG, merge_columns=2)
    json_to_docx(table_json, output_dir).save(output_dir)
    
    #使用器具记录表
    add_table_title(output_dir, USE_TOOL_TABLE_TITLE)
    print(use_tool_table)
    json_to_docx(list_to_json_with_merges(use_tool_table, style_config=STYLE_CONFIG, detect_merges=False),output_dir).save(output_dir)