import configparser
import pandas as pd
from sqlalchemy import create_engine
from urllib.parse import quote_plus
# 读取配置文件
config = configparser.ConfigParser()
config.read('config.ini')

# 获取数据库连接配置
source_config = {
    'host': config.get('source_db', 'host'),
    'user': config.get('source_db', 'user'),
    'password': config.get('source_db', 'password'),
    'database': config.get('source_db', 'database'),
    'port': config.getint('source_db', 'port')
}

target_config = {
    'host': config.get('target_db', 'host'),
    'user': config.get('target_db', 'user'),
    'password': config.get('target_db', 'password'),
    'database': config.get('target_db', 'database'),
    'port': config.getint('target_db', 'port')
}

# 创建数据库引擎
source_engine = create_engine(
    f"mysql+pymysql://{source_config['user']}:{quote_plus(source_config['password'])}@{source_config['host']}:{source_config['port']}/{source_config['database']}"
)
target_engine = create_engine(
    f"mysql+pymysql://{target_config['user']}:{quote_plus(target_config['password'])}@{target_config['host']}:{target_config['port']}/{target_config['database']}"
)

# 定义COMPANY_ID到imp_unit的映射
company_mapping = {
    5: 340,  # 安徽顺全电力工程有限公司
    4: 102,  # 送电二分公司
    3: 327,  # 送电一分公司
    6: 101,  # 机具（物流）分公司
    7: 346,  # 运检分公司
    8: 338,  # 建筑分公司
    9: 309,  # 安徽宏源电力建设有限公司
    10: 347,  # 公司培训中心（宏源工业园）
    11: 337,  # 检修试验分公司
    12: 100,  # 变电分公司
    13: 348,  # 预制构件分公司
    14: 344,  # 机械化分公司
    15: 342,  # 公司机关
    16: 345,  # 外单位租赁业务
    17: 339,  # 安徽顺安电网建设有限公司
    18: 341  # 班组管理中心
}


def process_bm_project():
    """处理bm_project表数据"""
    # 读取源数据
    df = pd.read_sql("SELECT * FROM bm_project", source_engine)

    # 按照规则转换数据
    result = pd.DataFrame()
    result['pro_id'] = df['ID'] + 3000  # ID加3000
    result['pro_name'] = df['NAME']  # 直接复制
    result['pro_code'] = df['CODE']  # 直接复制
    result['external_id'] = df['PRO_ID']  # 直接复制
    result['create_time'] = df['CREATE_TIME']  # 直接复制
    result['imp_unit'] = df['COMPANY'].map(company_mapping)  # 替换映射

    # 写入目标表
    try:
        result.to_sql('bm_project', target_engine,
                      if_exists='append', index=False)
        print(f"成功转换并导入 {len(result)} 条记录到 bm_project")
        return True
    except Exception as e:
        print(f"处理 bm_project 时发生错误: {str(e)}")
        return False


if __name__ == "__main__":
    process_bm_project()