#-*- encoding:utf-8 -*-

'''
@Author : dingjiawen
@Date : 2023/5/17 21:43
@Usage : 
@Desc :  将TagMapping导入mysql
'''
import pymysql
import pandas as pd
from pymysql.connections import Connection

data_path = r"E:\面试准备\interview-prepare\个人总结\04大数据\09项目\大型旋转机组健康管理系统软件\数仓\数仓建设\配置类数据库建立\20230327092414\TagMapping_PL193CEPB_C.csv"
data_df = pd.read_csv(data_path, encoding='gbk')

data_df =data_df.fillna("")



def getConnection():
    # 打开数据库连接
    try:
        conn = pymysql.connect(host='Ding202', user='root', passwd='123456', port=3306, database='rt_phm')
        print('连接成功！')
    except:
        print('something wrong!')
    return conn

def add_test_users(conn:Connection):
    usersvalues = []
    for data in data_df.values:
        tag = str(data[1])
        if(len(tag)>0):
            data[1] = tag.split("].")[1]
        usersvalues.append((data[0],data[1],data[2],data[4],data[5],data[10],0,""))

    cs = conn.cursor()  # 获取光标
    # 注意这里使用的是executemany而不是execute，下边有对executemany的详细说明
    cs.executemany("insert into gas_code_process(decode_value,encode_value,tag_desc,unit,scale_factor,standard_unit,is_calulated,extend_info) values(%s,%s,%s,%s,%s,%s,%s,%s)", usersvalues)

    conn.commit()
    cs.close()
    conn.close()
    print('OK')

def insert_data():
    conn = getConnection()

    add_test_users(conn)
    pass


def load_data():
    insert_data()
    pass

if __name__ == '__main__':
    insert_data()