ControlNetwork/Front/back/GC/GC.py

import openpyxl
import sqlite3
import os
import pandas as pd
from openpyxl import Workbook
from PySide6.QtWidgets import QApplication, QMessageBox


# 添加文件转换函数
def convert_xls_to_xlsx(file_path):
    # 使用 pandas 读取 .xls 文件
    df = pd.read_excel(file_path, engine='xlrd')
    # 创建一个新的 .xlsx 文件路径
    xlsx_file_path = file_path.replace('.xls', '.xlsx')
    # 将数据保存为 .xlsx 文件
    df.to_excel(xlsx_file_path, index=False)
    return xlsx_file_path


# 弹窗提示用户 .xls 文件将转换为 .xlsx 文件
def confirm_xls_to_xlsx_conversion(file_name):
    response = QMessageBox.question(None, "确认转换",
                                    f".xls文件将默认转换为.xlsx文件进行计算，系统会重新生成一个.xlsx文件副本用以计算，不会删除您本来的.xls文件，点击“OK”开始上传文件，点击“Cancel”取消上传文件",
                                    QMessageBox.Ok | QMessageBox.Cancel)
    return response == QMessageBox.Ok


# 读取指定单元格的数据
def read_cells_from_excel(cells, sheet):
    """
    读取指定单元格的数据并返回字典。
    """
    cell_values = {}
    for cell in cells:
        cell_values[cell] = sheet[cell].value
    return cell_values


# 获取最后一行有数字的单元格值
def get_last_numeric_cell_value(sheet, column):
    """
    获取指定列的最后一行有数字的单元格的值。
    """
    last_row = sheet.max_row
    for row in range(last_row, 0, -1):
        cell_value = sheet[f"{column}{row}"].value
        if isinstance(cell_value, (int, float)):
            return cell_value
    return None


# 计算指定列中数值型数据的总和
def calculate_column_sum(sheet, column):
    """
    计算指定列中数值型数据的总和。
    """
    total_sum = 0
    for row in range(1, sheet.max_row + 1):
        cell_value = sheet[f"{column}{row}"].value
        if isinstance(cell_value, (int, float)):
            total_sum += cell_value
    return total_sum


# 读取Excel文件并计算公式结果
def load_and_calculate_excel(file_path):
    # 加载Excel文件并计算单元格H1的公式结果。
    workbook = openpyxl.load_workbook(file_path, data_only=True)
    sheet = workbook.active
    h1_value = sheet['H1'].value
    return h1_value


# 弹窗提示用户是否更新数据
def ask_for_update(file_name):
    response = QMessageBox.question(None, "提示", f"检测到数据库中存在相同文件 '{file_name}'，是(Yes)否(No)更新数据？",
                                    QMessageBox.Yes | QMessageBox.No)
    return response == QMessageBox.Yes


# 数据库插入函数，插入数据到GC_Input_Param中
def insert_into_database(file_name, cell_values, ms_station, last_station_count, new_station_count, last_sum_hdiff,
                         new_sum_hdiff,
                         last_sum_rlen, new_sum_rlen, db_path, file_name_utf8):
    """
    将文件名和结果名称插入或更新到数据库的GC_Input_Param表中。
    """
    database = db_path
    conn = sqlite3.connect(database=database)
    cursor = conn.cursor()
    try:
        # 查询是否已存在相同的记录
        cursor.execute(
            "SELECT * FROM GC_Input_Param WHERE TableName = ?",
            (file_name_utf8,)
        )
        existing_record = cursor.fetchone()

        if existing_record:
            # 弹窗询问用户是否更新数据
            if not ask_for_update(file_name):
                print("用户选择不更新数据")
                return

            # 更新现有记录
            cursor.execute(
                """
                UPDATE GC_Input_Param 
                SET Last_ResultName = ?, ObservationLevel = ?, New_ResultName = ?, Ms_Station = ?, 
                    Last_StationCount = ?, New_StationCount = ?, Last_SumHDiff = ?, New_SumHDiff = ?, 
                    Last_SumRLen = ?, New_SumRLen = ?
                WHERE TableName = ?
                """,
                (cell_values['D1'], cell_values['C1'], cell_values['I1'], ms_station, last_station_count,
                 new_station_count, last_sum_hdiff, new_sum_hdiff, last_sum_rlen, new_sum_rlen, file_name_utf8)
            )
            QMessageBox.information(None, "提示",
                                    f"文件 '{file_name}' 已成功更新到本地数据库中，点击'OK'以关闭此对话框。")
        else:
            # 插入新记录
            cursor.execute(
                """
                INSERT INTO GC_Input_Param (TableName, Last_ResultName, ObservationLevel, New_ResultName, Ms_Station, 
                                            Last_StationCount, New_StationCount, Last_SumHDiff, New_SumHDiff, 
                                            Last_SumRLen, New_SumRLen) 
                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                """,
                (
                    file_name_utf8, cell_values['D1'], cell_values['C1'], cell_values['I1'], ms_station,
                    last_station_count,
                    new_station_count, last_sum_hdiff, new_sum_hdiff, last_sum_rlen, new_sum_rlen)
            )
            QMessageBox.information(None, "提示",
                                    f"文件 '{file_name}' 已成功添加到本地数据库中，点击'OK'以关闭此对话框。")

        conn.commit()
    except sqlite3.Error as e:
        print(f"插入或更新文件名时发生错误: {e}")
    except Exception as e:
        print(f"处理文件时发生错误: {e}")
    finally:
        conn.close()


# 遍历获取序号的数据
def get_data_from_excel(file_path):
    """
    从 Excel 文件中获取对应列的数据。
    """
    workbook = openpyxl.load_workbook(file_path)
    sheet = workbook.active
    data = []
    for row in range(3, sheet.max_row + 1):
        a_value = sheet[f"A{row}"].value
        b_value = sheet[f"B{row}"].value
        c_value = sheet[f"C{row}"].value
        d_value = sheet[f"D{row}"].value
        e_value = sheet[f"E{row}"].value
        f_value = sheet[f"F{row}"].value
        g_value = sheet[f"G{row}"].value
        h_value = sheet[f"H{row}"].value
        i_value = sheet[f"I{row}"].value
        j_value = sheet[f"J{row}"].value
        if a_value is not None or f_value is not None:
            data.append((a_value, b_value, c_value, d_value, e_value, f_value, g_value, h_value, i_value, j_value))
    return data


# 插入点数据到GC_Input_Point中
def insert_data_into_database(data, file_name, cell_values, db_path, file_name_utf8):
    """
    将数据插入到数据库表 GC_Input_Point 中。如果 TableName 相同，则清空表，
    然后插入新的数据。
    """
    database = db_path
    conn = sqlite3.connect(database=database)
    cursor = conn.cursor()

    try:
        # 检查是否已存在相同的 TableName
        cursor.execute(
            "SELECT * FROM GC_Input_Point WHERE TableName = ?",
            (file_name_utf8,)
        )
        existing_record = cursor.fetchone()

        if existing_record:
            # 如果存在相同的 TableName，清空表
            cursor.execute("DELETE FROM GC_Input_Point WHERE TableName = ?", (file_name_utf8,))
            conn.commit()

        # 插入新数据
        for a_value, b_value, c_value, d_value, e_value, f_value, g_value, h_value, i_value, j_value in data:
            cursor.execute(
                """
                INSERT INTO GC_Input_Point (Last_ID, Last_SPName, Last_EPName,Last_HDiff,Last_RLen, New_ID, New_SPName, New_EPName,New_HDiff,New_RLen, TableName, Last_ResultName, New_ResultName)
                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                """,
                (a_value, b_value, c_value, d_value, e_value, f_value, g_value, h_value, i_value, j_value,
                 file_name_utf8,
                 cell_values['D1'],
                 cell_values['I1'])
            )

        conn.commit()
    except sqlite3.Error as e:
        print(f"插入或更新点数据时发生错误: {e}")
    except Exception as e:
        print(f"处理文件时发生错误: {e}")
    finally:
        conn.close()


# 读取指定单元格的数据(编辑)
def read_cells_from_excel1(list1):
    cell_values = {}
    cell_values['D1'] = list1[0][1]
    cell_values['C1'] = list1[0][11]
    cell_values['I1'] = list1[0][6]
    return cell_values


# 获取最后一行有数字的单元格值(编辑)
def get_last_numeric_cell_value1(list1, index):
    ii = 0
    while ii < len(list1):
        if list1[ii][index] == '' or list1[ii][index] == ' ' or list1[ii][index] == None:
            ii = ii + 1
            break
        else:
            ii = ii + 1
    return ii


# 计算指定列中数值型数据的总和（编辑）
def calculate_column_sum1(list1, index):
    sum = 0
    for li in list1:
        if li[index] == '' or li[index] == ' ' or li[index] == None:
            break
        else:
            sum = sum + float(li[index])
    return sum


# 遍历获取序号的数据(修改数据)
def get_data_from_excel1(list1):
    data = []
    for row in range(len(list1)):
        a_value = list1[row][0]
        b_value = list1[row][2]
        c_value = list1[row][3]
        try:
            d_value = float(list1[row][4])
            e_value = float(list1[row][5])
        except:
            d_value = None
            e_value = None
        f_value = list1[row][0]
        g_value = list1[row][7]
        h_value = list1[row][8]
        try:
            i_value = float(list1[row][9])
            j_value = float(list1[row][10])
        except:
            i_value = None
            j_value = None
        if a_value is not None or f_value is not None:
            data.append((a_value, b_value, c_value, d_value, e_value, f_value, g_value, h_value, i_value, j_value))
    return data


# 主函数 读取excel文件
def main_function(file_path, db_path):
    # 检查文件扩展名
    if file_path.endswith('.xls'):
        # 弹窗提示用户 .xls 文件将转换为 .xlsx 文件
        if not confirm_xls_to_xlsx_conversion(os.path.basename(file_path)):
            print("用户取消了转换操作")
            return

        # 转换 .xls 文件为 .xlsx 文件
        file_path = convert_xls_to_xlsx(file_path)
    # 调用读取Excel文件的函数
    file_name = os.path.basename(file_path)
    file_name_utf8 = file_name.encode('utf-8')
    # 加载工作簿
    workbook = openpyxl.load_workbook(file_path)
    sheet = workbook.active
    # 计算单元格 H1 的公式结果
    ms_station = load_and_calculate_excel(file_path)
    # 读取指定单元格的数据
    cells_to_read = ['D1', 'C1', 'I1']
    cell_values = read_cells_from_excel(cells_to_read, sheet)
    # 获取最后一行有数字的单元格值
    last_station_count = get_last_numeric_cell_value(sheet, 'A')
    new_station_count = get_last_numeric_cell_value(sheet, 'F')
    # 计算 D 列和 I 列中数值型数据的总和  总高差
    last_sum_hdiff = calculate_column_sum(sheet, 'D')
    new_sum_hdiff = calculate_column_sum(sheet, 'I')
    # 总路线长
    last_sum_rlen = calculate_column_sum(sheet, 'E')
    new_sum_rlen = calculate_column_sum(sheet, 'J')
    # 插入点的数组
    data = get_data_from_excel(file_path)
    # 调用插入数据库的函数
    if file_name:
        insert_into_database(file_name, cell_values, ms_station, last_station_count, new_station_count, last_sum_hdiff,
                             new_sum_hdiff, last_sum_rlen, new_sum_rlen, db_path, file_name_utf8)
        insert_data_into_database(data, file_name, cell_values, db_path, file_name_utf8)
    else:
        print("没有数据可插入数据库")