修复了GC中上传相同文件无法覆盖计算的问题

Front/back/GC/GCcompute.py

@@ -217,6 +217,7 @@ def calculate_period_differences(old_hdiffs, result_height_differences):
 def insert_records_to_output_table(db_path, file_name, dis_ass, new_hdiff, correction_numbers, period_differences):
     with sqlite3.connect(db_path) as conn:
         cursor = conn.cursor()
+
         # 查询 GC_Input_Point 表中的数据
         query = """
             SELECT New_ID, New_ResultName, New_SPName, New_EPName, New_RLen, TableName
@@ -225,50 +226,49 @@ def insert_records_to_output_table(db_path, file_name, dis_ass, new_hdiff, corre
         """
         cursor.execute(query, (file_name,))
         records = cursor.fetchall()
-        # 检查是否存在相同的 TableName
-        check_query = """
-            SELECT COUNT(*)
-            FROM GC_Output_Point
-            WHERE TableName = ?
-        """
-        cursor.execute(check_query, (file_name,))
-        count = cursor.fetchone()[0]
+
         # 获取最长数组的长度
         max_length = max(len(dis_ass), len(new_hdiff), len(correction_numbers), len(period_differences))
+
         # 填充缺失值
         dis_ass = dis_ass + [None] * (max_length - len(dis_ass))
         new_hdiff = new_hdiff + [None] * (max_length - len(new_hdiff))
         correction_numbers = correction_numbers + [None] * (max_length - len(correction_numbers))
         period_differences = period_differences + [None] * (max_length - len(period_differences))
-        if count > 0:
-            # 更新操作
-            update_query = """
-                UPDATE GC_Output_Point
-                SET Dis_Ass = ?, New_HDiff = ?, Correct_Factor = ?, Period_Diff = ?
+
+        # 遍历记录并更新或插入数据
+        for record, deformation_result, height_difference, correction, period_diff in zip(records, dis_ass, new_hdiff, correction_numbers, period_differences):
+            # 检查是否存在相同的 New_ID 和 TableName
+            check_query = """
+                SELECT COUNT(*)
+                FROM GC_Output_Point
                 WHERE New_ID = ? AND TableName = ?
             """
-            updated_records = []
-            for record, deformation_result, height_difference, correction, period_diff in zip(records, dis_ass,
-                                                                                              new_hdiff,
-                                                                                              correction_numbers,
-                                                                                              period_differences):
-                updated_record = [deformation_result, height_difference, correction, period_diff, record[0], file_name]
-                updated_records.append(updated_record)
-            cursor.executemany(update_query, updated_records)
-        else:
-            # 插入操作
-            insert_query = """
-                INSERT INTO GC_Output_Point (New_ID, New_ResultName, New_SPName, New_EPName, New_RLen, TableName, Dis_Ass, New_HDiff, Correct_Factor, Period_Diff)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-            """
-            updated_records = []
-            for record, deformation_result, height_difference, correction, period_diff in zip(records, dis_ass,
-                                                                                              new_hdiff,
-                                                                                              correction_numbers,
-                                                                                              period_differences):
-                updated_record = list(record) + [deformation_result, height_difference, correction, period_diff]
-                updated_records.append(updated_record)
-            cursor.executemany(insert_query, updated_records)
+            cursor.execute(check_query, (record[0], file_name))
+            count = cursor.fetchone()[0]
+
+            if count > 0:
+                # 更新操作，以 GC_Input_Point 中的数据为准
+                update_query = """
+                    UPDATE GC_Output_Point
+                    SET New_ResultName = ?, New_SPName = ?, New_EPName = ?, New_RLen = ?, 
+                        Dis_Ass = ?, New_HDiff = ?, Correct_Factor = ?, Period_Diff = ?
+                    WHERE New_ID = ? AND TableName = ?
+                """
+                updated_record = [record[1], record[2], record[3], record[4],
+                                 deformation_result, height_difference, correction, period_diff,
+                                 record[0], file_name]
+                cursor.execute(update_query, updated_record)
+            else:
+                # 插入操作
+                insert_query = """
+                    INSERT INTO GC_Output_Point (New_ID, New_ResultName, New_SPName, New_EPName, New_RLen, TableName, Dis_Ass, New_HDiff, Correct_Factor, Period_Diff)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                """
+                updated_record = [record[0], record[1], record[2], record[3], record[4], file_name,
+                                 deformation_result, height_difference, correction, period_diff]
+                cursor.execute(insert_query, updated_record)
+
         conn.commit()