# 步骤一：读取表格
import pandas as pd

# dirPath = input('请输入源文件地址:')
# db_shengls_202401zz.xlsx
dirPath = 'C:\\tmp\\'
ym = '202411'
monthname = '2024年11月'
path = dirPath + 'db_shengls_' + ym + 'zz.xlsx'

orgName = pd.read_excel(path)
org_list = list(orgName['区'].drop_duplicates())
print(org_list)

# 步骤二：完成表格的拆分  表1 xx县-x月-低保导入模版（在保家庭及人员）.xls
for i2 in org_list:
    print(i2)
    cname = '其他'
    i = str(i2)

     # 使用字典简化条件判断
    cname_map = {
        '昌邑区': '昌邑区',
        '龙潭区': '龙潭区',
        '船营区': '船营区',
        '丰满区': '丰满区',
        '高新区': '高新区',
        '经开区': '经开区'
    }



    cname = cname_map.get(i, '其他')

    print(cname)

    writer = pd.ExcelWriter(dirPath + cname + '-' + monthname + '-临时救助导入模版.xlsx',
                            engine='xlsxwriter')
    tempdata = orgName[orgName['区'] ==  i]
    # 删除分区这列
    data = tempdata.drop(columns='区')
    #data=tempdata
    # 首列插入序号
    # data.insert(0, '序号', range(1, 1 + len(data)))
    data[['村/社区区划编码']] = data[['村/社区区划编码']].astype(str)
    data.to_excel(writer, index=False)
    writer.close()