ControlNetwork/Back/GC/GC.py

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


# 读取指定单元格的数据
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()


# 主函数 读取excel文件
def main_function(file_path, db_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("没有数据可插入数据库")