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

'''
@Author : dingjiawen
@Date : 2023/5/20 16:32
@Usage : 
@Desc :  随机向维度表中插入数据
'''

import pymysql
import pandas as pd
from pymysql.connections import Connection
import os
import random
import shutil
import math

root = r"E:\面试准备\interview-prepare\个人总结\04大数据\09项目\大型旋转机组健康管理系统软件\数仓\数仓建设\配置类数据库建立\20230327092414/"
root_file = os.path.join(root, "data_csv\\")

if not os.path.exists(root_file):
    os.makedirs(root_file)


def createWindTypeInfo():
    windTypeInfo = []

    for wt_no in range(25):
        windTypeInfo.append(("jingbian_siqi_" + str(wt_no), "jingbian_siqi", random.randint(0, 4)))
        windTypeInfo.append(("huanengsantanghu_" + str(wt_no), "huanengsantanghu", random.randint(0, 4)))
        windTypeInfo.append(("bandacheng_" + str(wt_no), "bandacheng", random.randint(0, 4)))
    return windTypeInfo
    pass


def createWindTypeDetail():
    windTypeDetail = []
    for type_id in range(5):
        windTypeDetail.append(
            (type_id, random.randint(1000, 2000), random.randint(20000, 50000), random.randint(5000, 10000),
             random.randint(1000, 5000), random.randint(10000, 50000), random.randint(70, 100))
        )
    return windTypeDetail
    pass


def createLocationInfo():
    locationInfo = []

    locationInfo.append(
        ("jingbian_siqi", "靖边四期", 100.65, 82.46, 20000, 20)
    )

    locationInfo.append(
        ("huanengsantanghu", "华能三塘湖", 110.65, 67.46, 20000, 30)
    )

    locationInfo.append(
        ("bandacheng", "扳达城", 60.65, 103.46, 16000, 10)
    )

    return locationInfo
    pass


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

def insert_type_detail(conn: Connection, if_clear=True):
    windTypeDetail = createWindTypeDetail()
    cs = conn.cursor()  # 获取光标
    if if_clear:
        cs.execute("truncate table wind_type_detail")
    cs.executemany("insert into wind_type_detail values(%s,%s,%s,%s,%s,%s,%s)", windTypeDetail)
    conn.commit()
    cs.close()
    conn.close()
    print('OK')
    pass

def insert_location(conn: Connection, if_clear=True):
    locationInfo = createLocationInfo()
    cs = conn.cursor()  # 获取光标
    if if_clear:
        cs.execute("truncate table wind_location_info")
    cs.executemany("insert into wind_location_info values(%s,%s,%s,%s,%s,%s)", locationInfo)
    conn.commit()
    cs.close()
    conn.close()
    print('OK')
    pass

def insert_type_info(conn: Connection, if_clear=True):

    windType_info = createWindTypeInfo()
    cs = conn.cursor()  # 获取光标
    if if_clear:
        cs.execute("truncate table wind_type_info")

    # 注意这里使用的是executemany而不是execute，下边有对executemany的详细说明
    cs.executemany("insert into wind_type_info values(%s,%s,%s)", windType_info)

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


def insert_data():
    conn = getConnection()
    insert_type_info(conn)
    insert_location(conn)
    insert_type_detail(conn)
    pass


if __name__ == '__main__':
    # nameChange()
    # loadData()
    # insert_data()
    conn = getConnection()
    insert_location(conn)