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

# dirPath = input('请输入源文件地址:')
dirPath = 'C:\\tmp\\'
path = dirPath + 'data_shengdb_202405zzxz.xlsx'
monthname = '2024年05月'

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)

    if i == '220202':  # 判断变量是否为 python
        # 条件成立时设置标志为真
        cname = '昌邑区'

    if i == '220203':  # 判断变量是否为 python
        # 条件成立时设置标志为真
        cname = '龙潭区'

    if i == '220204':  # 判断变量是否为 python
        # 条件成立时设置标志为真
        cname = '船营区'

    if i == '220211':  # 判断变量是否为 python
        # 条件成立时设置标志为真
        cname = '丰满区'

    if i == '220292':  # 判断变量是否为 python
        # 条件成立时设置标志为真
        cname = '高新区'

    if i == '220291':  # 判断变量是否为 python
        # 条件成立时设置标志为真
        cname = '经开区'

    print(cname)

    writer = pd.ExcelWriter(dirPath + cname + '-'+monthname+'-低保导入模版（新增及迁移家庭）.xlsx', engine='xlsxwriter')
    tempdata = orgName[orgName['分区'] == int(i)]
    # 删除分区这列
    data = tempdata.drop(columns='分区')
    # 首列插入序号
    data.insert(0, '序号', range(1, 1 + len(data)))
    data[['村/社区区划编码', '家庭编号']] = data[['村/社区区划编码', '家庭编号']].astype(str)
    data.to_excel(writer, index=False)
    writer.close()