Bonus-Transfer-Machines/machines/新购配件.py

import configparser
import pandas as pd
from sqlalchemy import create_engine
from urllib.parse import quote_plus
# 读取配置文件
config = configparser.ConfigParser()
config.read(r'D:\code\Bonus-Transfer-Machines\machines\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']}"
)

def process_tm_task():
    """处理新购任务数据"""
    try:
        # 执行转换SQL
        sql = """
        SELECT
        bps.ID as task_id,
        12 as task_type,
        IF( SUM(ttmt.MACHINES_NUM) -  IFNULL(SUM(ttmt.INPUT_NUM),0) =0,1,0) as task_status,
        tt.`CODE` as code,
        bps.BUYER as create_by,
        tt.CREATE_TIME as create_time
        FROM
        ba_pa_shop bps 
        LEFT JOIN tm_task tt on bps.ID = tt.ID
        LEFT JOIN tm_task_status tts on tt.`STATUS` = tts.`CODE`
        LEFT JOIN tm_task_ma_type ttmt on bps.ID = ttmt.TASK_ID
        WHERE tt.CREATE_TIME BETWEEN '2025-01-01' and NOW() 
        GROUP BY bps.ID
        """
        df = pd.read_sql(sql, source_engine)

        # 写入目标表
        df.to_sql('tm_task', target_engine,
                  if_exists='append', index=False)

        print(f"成功导入 {len(df)} 条新购配件任务表数据")
        return True

    except Exception as e:
        print(f"处理新购配件任务表数据时出错: {str(e)}")
        return False

def process_purchase_part_info():
        try:
            # 执行转换SQL
            sql = """
            SELECT
            bps.ID as task_id,
            bps.BUY_TIME as purchase_time,
            bps.ACCEPT_TIME as arrival_time,
            ttmt.MANUFACTURER_ID as supplier_id,
            bps.BUYER as create_by,
            tt.CREATE_TIME as create_time
            FROM
            ba_pa_shop bps 
            LEFT JOIN tm_task tt on bps.ID = tt.ID
            LEFT JOIN tm_task_status tts on tt.`STATUS` = tts.`CODE`
            LEFT JOIN tm_task_ma_type ttmt on bps.ID = ttmt.TASK_ID
            WHERE tt.CREATE_TIME BETWEEN '2025-01-01' and NOW() 
            GROUP BY bps.ID
            """
            df = pd.read_sql(sql, source_engine)

            # 写入目标表
            df.to_sql('purchase_part_info', target_engine,
                      if_exists='append', index=False)

            print(f"成功导入 {len(df)} 条新购配件主表数据")
            return True

        except Exception as e:
            print(f"处理新购配件主表数据时出错: {str(e)}")
            return False

def process_purchase_part_details():
    try:
        # 执行转换SQL
        sql = """
        SELECT
        bps.ID as task_id,
        ttmt.MA_TYPE_ID as part_id,
        ttmt.PRICE as purchase_price,
        ttmt.MACHINES_NUM as purchase_num,
        ttmt.MACHINES_NUM as check_num,
        ttmt.INPUT_NUM as input_num,
        ttmt.MANUFACTURER_ID as supplier_id,
        IF(ttmt.MACHINES_NUM - IFNULL(ttmt.INPUT_NUM,0) = 0,1,0) as status,
        bps.BUYER as create_by,
        tt.CREATE_TIME as as create_time
        FROM
        ba_pa_shop bps 
        LEFT JOIN tm_task tt on bps.ID = tt.ID
        LEFT JOIN tm_task_ma_type ttmt on bps.ID = ttmt.TASK_ID
        WHERE tt.CREATE_TIME BETWEEN '2025-01-01' and NOW() 
        GROUP BY bps.ID,ttmt.MA_TYPE_ID
        """
        df = pd.read_sql(sql, source_engine)

        # 写入目标表
        df.to_sql('purchase_part_details', target_engine,
                  if_exists='append', index=False)

        print(f"成功导入 {len(df)} 条新购配件详情表数据")
        return True

    except Exception as e:
        print(f"处理新购配件详情表数据时出错: {str(e)}")
        return False

if __name__ == "__main__":
    # 执行3个转换流程
    success1 = process_tm_task()
    success2 = process_purchase_part_info()
    success3 = process_purchase_part_details()

    if success1 and success2 and success3:
        print("所有数据转换完成！")
    else:
        print("数据转换过程中出现错误，请检查日志")