Merge branch 'master' of http://oa.sccehui.com:6101/cmc-coding2/ControlNetwork

Back/Program_Run/file_system_watching.py

@@ -4,11 +4,13 @@ from PySide6.QtCore import QCoreApplication
 from watchdog.observers import Observer
 import os
 
+
 class ComboBoxUpdater(FileSystemEventHandler):
     def __init__(self, comboBox, sql_folder):
         super().__init__()
         self.comboBox = comboBox
         self.sql_folder = sql_folder
+        self.selected_db = None  # 新增属性来存储当前选中的数据库
 
     def on_modified(self, event):
         if event.is_directory:
@@ -29,6 +31,8 @@ class ComboBoxUpdater(FileSystemEventHandler):
             self.update_combo_box()
 
     def update_combo_box(self):
+        # 保存当前选中的数据库
+        self.selected_db = self.comboBox.currentText()
         # 清空现有的 comboBox 内容
         self.comboBox.clear()
 
@@ -42,4 +46,8 @@ class ComboBoxUpdater(FileSystemEventHandler):
 
         # 如果没有找到任何数据库文件，显示提示信息
         if not db_files:
-            self.comboBox.addItem(QCoreApplication.translate("MainWindow", "未找到数据库文件", None))
+            self.comboBox.addItem(QCoreApplication.translate("MainWindow", "未找到数据库文件", None))
+
+        # 重新设置之前选中的数据库为选中状态
+        if self.selected_db in [self.comboBox.itemText(i) for i in range(self.comboBox.count())]:
+            self.comboBox.setCurrentText(self.selected_db)

Front/main.py

@@ -21,6 +21,7 @@ import sys
 import os
 import platform
 
+import Back.Program_Run.database_operations
 # IMPORT / GUI AND MODULES AND WIDGETS
 # ///////////////////////////////////////////////////////////////
 from modules import *
@@ -66,6 +67,7 @@ class MainWindow(QMainWindow):
         # ///////////////////////////////////////////////////////////////
         self.ui = Ui_MainWindow()
         self.ui.setupUi(self)
+        self.ui.createFile.clicked.connect(self.on_create_file_clicked)
         # self.comboBox_2 = QComboBox(self)
         # ...此处为省略代码...
         global widgets
@@ -245,6 +247,11 @@ class MainWindow(QMainWindow):
             # 触发按钮点击事件
             btn.click()
 
+
+    # 新增槽函数来调用 create_database_and_tables
+    def on_create_file_clicked(self):
+        Back.Program_Run.database_operations.create_database_and_tables(self.ui)
+
     # RESIZE EVENTS
     # ///////////////////////////////////////////////////////////////
     def resizeEvent(self, event):
@@ -270,4 +277,4 @@ if __name__ == "__main__":
     app.setWindowIcon(QIcon("icon.ico"))
     window = MainWindow()
     # window.resize(1440, 960)  # 高宽
-    sys.exit(app.exec_())
+    sys.exit(app.exec())

Front/modules/ui_functions.py

@@ -296,7 +296,14 @@ class UIFunctions(MainWindow):
     # 上传文件的方法
     def execute_script_based_on_selection(self, file_path):
         current_text = self.ui.comboBox_2.currentText()
-        db_path = r"D:/Code/ControlNetwork/UI/SQL/DataBase.db"
+        # db_path = r"D:/Code/ControlNetwork/UI/SQL/DataBase.db"
+        # 获取当前脚本所在的目录
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        # 构建 SQL 文件夹的相对路径（上一级的上一级中的 SQL 文件夹）
+        sql_folder = os.path.join(current_dir, '..', '..', 'SQL')
+        # 将相对路径转换为绝对路径
+        sql_folder = os.path.abspath(sql_folder)
+        db_path = os.path.join(sql_folder, f"{self.ui.comboBox.currentText()}.db")
         file_path = file_path
         if current_text == "水准测段高差稳定计算":
             GC.main_function(file_path, db_path)
@@ -311,8 +318,15 @@ class UIFunctions(MainWindow):
     # 计算与展示文件的方法
     def compute_show_process_excel_file(self, file_path):
         current_text = self.ui.comboBox_2.currentText()
-        db_path = r"D:/Code/ControlNetwork/UI/SQL/DataBase.db"
-        #转换为utf-8
+        # db_path = r"D:/Code/ControlNetwork/UI/SQL/DataBase.db"
+        # 获取当前脚本所在的目录
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        # 构建 SQL 文件夹的相对路径（上一级的上一级中的 SQL 文件夹）
+        sql_folder = os.path.join(current_dir, '..', '..', 'SQL')
+        # 将相对路径转换为绝对路径
+        sql_folder = os.path.abspath(sql_folder)
+        db_path = os.path.join(sql_folder, f"{self.ui.comboBox.currentText()}.db")
+        # 转换为utf-8
         excelname = os.path.basename(file_path)
         utf_en = excelname.encode('utf-8')
         if current_text == "水准测段高差稳定计算":