from datetime import datetime, timedeltaimport loggingimport osimport pandas as pdimport reimport numpy as npfrom openpyxl import load_workbookfrom openpyxl.styles import Font, Alignmentfrom openpyxl.styles import Border, Side# -------------------------- 配置区（仅修改这里即可，无需动核心代码）--------------------------LOG_LEVEL = logging.DEBUG  # 日志级别（不用改）LOG_FILE = 'logs/app.log'  # 日志文件路径（不用改，自动生成）TEMPLATE_PATH = './data/购销合同模板.xlsx'  # 集团合同模板路径，替换成自己的路径FACTORY_INFO_PATH = './data/杂品工厂抬头信息.xlsx'  # 工厂信息表路径，替换成自己的路径INPUT_DIRECTORY = './'  # 开票资料所在目录，替换成自己的路径PARTY_A_INFO = {  # 甲方信息，替换成自己公司的信息    "TD": {"name": "威海卓达进出口有限公司", "addr": "威海市新威路17-2（威高广场）SOHO A座15层", "phone": "0631-5280617", "short_name": "卓达"},    "MI": {"name": "威海名禾渔具有限公司", "addr": "威海市环翠区新威路17-2（威高广场SOHO A座）", "phone": "0631-5280617", "short_name": "铭禾"}}# ----------------------------------------------------------------------------------------# 创建日志目录（如果不存在）log_dir = os.path.dirname(LOG_FILE)if log_dir and not os.path.exists(log_dir):    os.makedirs(log_dir)# 配置日志logging.basicConfig(    filename=LOG_FILE,  # 日志文件路径    level=LOG_LEVEL,    # 日志级别    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',  # 日志格式    datefmt='%Y-%m-%d %H:%M:%S'  # 日期格式)def number_to_chinese_amount(number):    """    将阿拉伯数字转换为中文金额大写（无需修改，直接用）    :param number: 数字（整数或浮点数）    :return: 中文金额大写字符串    """    units = ['', '万', '亿']    digits = ['零', '壹', '贰', '叁', '肆', '伍', '陆', '柒', '捌', '玖']    scales = ['', '拾', '佰', '仟']    if not isinstance(number, (int, float)):        return "无效输入"    # 处理负数    if number < 0:        return "负数不支持转换"    # 分离整数和小数部分    number = round(number, 2)  # 保留两位小数    integer_part, decimal_part = f"{number:.2f}".split('.')    integer_part = int(integer_part)    decimal_part = int(decimal_part)    # 转换整数部分    def convert_integer(n):        if n == 0:            return digits[0]        result = ''        unit_index = 0        while n > 0:            part = n % 10000            if part != 0:                part_str = ''                for i, digit in enumerate(str(part)[::-1]):                    if digit != '0':                        part_str = digits[int(digit)] + scales[i] + part_str                    else:                        if part_str and part_str[0] != digits[0]:                            part_str = digits[0] + part_str                result = part_str + units[unit_index] + result            else:                if result and result[0] != digits[0]:                    result = digits[0] + result            n = n // 10000            unit_index += 1        return result    # 转换小数部分（角和分）    def convert_decimal(n):        if n == 0:            return '整'        jiao = n // 10  # 角        fen = n % 10    # 分        result = ''        if jiao != 0:            result += digits[jiao] + '角'        if fen != 0:            result += digits[fen] + '分'        return result    # 组合结果    chinese_integer = convert_integer(integer_part)    chinese_decimal = convert_decimal(decimal_part)    return chinese_integer + '元' + chinese_decimaldef is_factory_in_column_name(factory_name, column_names):    """判断工厂简称是否在列名中（无需修改）"""    return any(factory_name in str(col) for col in column_names)def get_date():    """获取所需日期（无需修改）"""    today = datetime.now()    today_str = today.strftime('%Y-%m-%d')    # 获取前20天、前5个月日期（适配合同交付、签订时间，可根据需求修改天数）    five_months_ago = today - timedelta(days=5 * 30)  # 近似计算，每月按30天    five_months_ago_str = five_months_ago.strftime('%Y-%m-%d')    twenty_days_ago = today - timedelta(days=20)    twenty_days_ago_str = twenty_days_ago.strftime('%Y-%m-%d')    return twenty_days_ago_str, five_months_ago_str, today_strdef deal_with_input(file_path):    """解析开票资料，提取所需信息（无需修改）"""    result_data = {}    result_data["delivery_time"], result_data["sign_time"], result_data["today_time"] = get_date()    # 读取开票资料（兼容.xls和.xlsx格式）    if file_path.endswith('.xls'):        data = pd.read_excel(file_path, engine='xlrd', header=None)    else:        data = pd.read_excel(file_path, header=None)    # 定位第一行非空行，作为列名    first_non_empty_row = data.dropna(how='all').index[0]    data.columns = data.iloc[first_non_empty_row]    data = data.iloc[first_non_empty_row + 1:].reset_index(drop=True)    # 清洗数据，去除空行空列    data_no_na = data.dropna(how='all').reset_index(drop=True)    df = data_no_na    # 定位并提取产品明细（关键词可根据需求修改）    start_index = df[df.apply(lambda row: row.astype(str).str.contains('品名|产品名称').any(), axis=1)].index[0]    end_index = df[df.apply(lambda row: row.astype(str).str.contains('备注|备注说明').any(), axis=1)].index[0]    prod_infos = data_no_na.iloc[start_index:end_index]    prod_infos.columns = prod_infos.iloc[0]    prod_infos = prod_infos[1:].reset_index(drop=True)    # 删除可能残留的“品名”行，清洗明细数据    mask = prod_infos.apply(lambda row: row.astype(str).str.contains('品名|产品名称').any(), axis=1)    prod_infos = prod_infos[~mask].dropna(axis=1, how='all').reset_index(drop=True)    # 汇总金额并转换为中文大写    total_amount = prod_infos['金额'].sum()    chinese_amount = number_to_chinese_amount(total_amount)    # 添加金额汇总行    new_row = [f"合计人民币金额（大写）：{chinese_amount}"] + [np.nan] * (len(prod_infos.columns) - 2) + [total_amount]    prod_infos.loc[len(prod_infos)] = new_row    result_data["prod_infos"] = prod_infos    # 提取合同号（正则规则可根据需求修改）    pattern = r'(TD\d{5}|MI\d{5})'    matches = df.applymap(lambda x: re.findall(pattern, str(x))).stack().tolist()    unique_matches = list(set([item for sublist in matches for item in sublist]))    # 匹配甲方信息    if len(unique_matches) == 1:        result_data["contract_num"] = unique_matches[0]        prefix = "TD" if "TD" in unique_matches[0] else "MI"        result_data["partyA_info"] = PARTY_A_INFO[prefix]        result_data["partyA_name"] = PARTY_A_INFO[prefix]["short_name"]    # 匹配乙方信息（工厂信息表）    df_factory = pd.read_excel(FACTORY_INFO_PATH, header=0)    result_flag = df_factory['工厂简称'].apply(is_factory_in_column_name, column_names=df.columns)    partyB_infos = df_factory[result_flag].iloc[0].to_dict()    result_data["partyB_info"] = partyB_infos    return result_datadef write_excel(data):    """将提取的信息写入合同模板（无需修改）"""    # 设置表格边框（美化格式）    thin_border = Border(        left=Side(style='thin'), right=Side(style='thin'),        top=Side(style='thin'), bottom=Side(style='thin')    )    # 加载集团合同模板    wb = load_workbook(TEMPLATE_PATH)    ws = wb.active    # 写入合同编号    ws['E1'] = f"编号：{data['contract_num']}"    # 写入甲乙方信息    ws['A3'] = f"供方（甲方）：{data['partyA_info']['name']}                需方（乙方）：{data['partyB_info']['工厂全称']}"    ws['A4'] = f"地址：{data['partyA_info']['addr']}      地址：{data['partyB_info']['地址']}"    ws['A5'] = f"电话：{data['partyA_info']['phone']}                                  电话：{data['partyB_info']['电话']}"    # 写入产品明细    df = data["prod_infos"]    start_row = 10  # 产品明细起始行，可根据模板修改    start_col = 2   # 产品明细起始列，可根据模板修改    # 写入列名并添加边框    for col_idx, col_name in enumerate(df.columns, start=start_col):        cell = ws.cell(row=start_row, column=col_idx, value=col_name)        cell.border = thin_border    # 写入明细数据并添加边框    for row_idx, row in enumerate(df.values, start=start_row + 1):        for col_idx, value in enumerate(row, start=start_col):            cell = ws.cell(row=row_idx, column=col_idx, value=value)            cell.border = thin_border    # 写入合同条款（可根据集团模板修改）    next_row = start_row + len(df) + 2    ws[f"A{next_row}"] = f"二、质量要求：以客户确认的样品质量为准，如因产品质量问题引起客户索赔，供方需付连带责任。"    ws[f"A{next_row+1}"] = f"三、交货地点及方式：{data['delivery_time']}前工厂交货。"    ws[f"A{next_row+2}"] = f"四、结算方式及期限：甲方凭增值税发票及出口货物专用缴款书装箱后30天付清。"    ws[f"A{next_row+3}"] = f"五、包装要求：出口用双瓦楞纸箱，纸箱尺寸按客户要求。"    ws[f"A{next_row+4}"] = f"六、违约责任：违约损失及连带责任由违约方承担。"    ws[f"A{next_row+5}"] = f"七、争议的解决方式：本合同在履行过程中如发生争议，由当事人双方协商解决。协商不成，提交威海市仲裁委员会仲裁。"    ws[f"A{next_row+6}"] = f"八、本合同甲、乙双方盖章签字后生效，甲乙双方各持壹份。"    ws[f"B{next_row+15}"] = f"甲方："    ws[f"F{next_row+15}"] = f"乙方："    ws[f"F{next_row+17}"] = data['sign_time']    # 创建结果文件夹（自动生成）    os.makedirs("./result", exist_ok=True)    # 保存合同文件（命名格式可修改）    output_path = f"./result/{data['partyB_info']['工厂简称']}-{data['contract_num']}-购销合同-{data['today_time']}.xlsx"    wb.save(output_path)    print(f"合同生成成功：{output_path}")if __name__ == "__main__":    # 批量处理指定目录下的所有Excel开票资料    excel_files = [os.path.join(INPUT_DIRECTORY, f) for f in os.listdir(INPUT_DIRECTORY) if f.endswith(('.xlsx', '.xls'))]    for file in excel_files:        try:            logging.info(f"开始处理文件：{file}")            result_data = deal_with_input(file)            write_excel(result_data)            logging.info(f"文件{file}处理成功")        except Exception as e:            logging.error(f"文件{file}处理失败，错误信息：{str(e)}")            print(f"文件{file}处理失败，可查看logs文件夹下的日志了解详情")