修复一系列数据库路径问题

Front/back/Program_Run/database_operations.py

@@ -1,37 +1,69 @@
-# database_operations.py  数据库操作相关代码,主要用于展示数据库名字
+#selectedCode:D:\Code\ControlNetwork\UI\Front\back\Program_Run\database_operations.py#L1-L83
+# database_operations.py
 import sqlite3
 import os
+import sys
+import logging
 from PySide6.QtWidgets import QMessageBox
+from .utils import resource_path  # 从 utils.py 导入 resource_path
 
+# 配置日志
+def setup_logging():
+    # 获取日志文件路径
+    log_folder = os.path.join(os.getcwd(), 'logs')
+    if not os.path.exists(log_folder):
+        os.makedirs(log_folder)
+
+    log_file = os.path.join(log_folder, 'app.log')
+
+    # 配置日志
+    logging.basicConfig(
+        level=logging.DEBUG,  # 设置日志级别
+        format='%(asctime)s - %(levelname)s - %(message)s',  # 日志格式
+        handlers=[
+            logging.FileHandler(log_file),  # 输出到文件
+            logging.StreamHandler()  # 输出到控制台（可选）
+        ]
+    )
+
+# 获取 SQL 文件夹的路径
+def get_sql_folder():
+    """获取 SQL 文件夹的路径，支持开发和打包环境"""
+    return resource_path('SQL')  # 使用 resource_path 获取 SQL 文件夹路径
+
+def create_database_and_tables(main_window, ui):
+    # 设置日志
+    setup_logging()
 
-def create_database_and_tables(main_window,ui):
     db_name = main_window.ui.lineEdit.text().strip()
 
     if not db_name:
         QMessageBox.warning(main_window, "警告", "请输入数据库名称")
         return
-    # 转换为UTF-8编码
-    # db_name_utf8 = db_name.encode('utf-8')
-    # 获取当前脚本所在的目录
-    current_dir = os.getcwd()
-    # 构建 SQL 文件夹的相对路径（上一级的上一级中的 SQL 文件夹）
-    sql_folder = os.path.join(current_dir, '..', '..', 'SQL')
-    # 将相对路径转换为绝对路径
-    sql_folder = os.path.abspath(sql_folder)
-
-    # 确保 SQL 文件夹存在
-    if not os.path.exists(sql_folder):
-        os.makedirs(sql_folder)
-
-    # 构建数据库文件的完整路径
+
+    # 获取 SQL 文件夹路径
+    sql_folder = get_sql_folder()
+
+    # 构建数据库文件路径
     db_path = os.path.join(sql_folder, f"{db_name}.db")
-    # 读取现有的数据库结构
+
+    # 记录数据库路径
+    logging.info(f"数据库路径: {db_path}")
+
+    # 检查数据库文件是否已经存在
+    if os.path.exists(db_path):
+        QMessageBox.warning(main_window, "警告", "当前项目名重复，请重新设置项目名，或者删除旧项目再创建此项目")
+        return
+
+    # 获取现有数据库结构文件的路径
     existing_db_path = os.path.join(sql_folder, 'DataBase.db')
+
     if not os.path.exists(existing_db_path):
-        QMessageBox.critical(main_window.main_window, "错误", "找不到现有的数据库结构文件 DataBase.db")
+        QMessageBox.critical(main_window, "错误", "找不到现有的数据库结构文件 DataBase.db")
         return
+
     try:
-        # 连接到现有的数据库以获取表结构
+        # 连接到现有数据库以获取表结构
         with sqlite3.connect(existing_db_path) as existing_conn:
             existing_cursor = existing_conn.cursor()
             existing_cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
@@ -51,4 +83,5 @@ def create_database_and_tables(main_window,ui):
 
         QMessageBox.information(main_window, "成功", f"项目{db_name}已成功创建")
     except Exception as e:
-        QMessageBox.critical(main_window, "错误", f"创建项目时出错: {str(e)}")
+        logging.error(f"创建项目时出错: {str(e)}", exc_info=True)
+        QMessageBox.critical(main_window, "错误", f"创建项目时出错: {str(e)}")

Front/back/Program_Run/file_system_watching.py

@@ -3,13 +3,14 @@ from watchdog.events import FileSystemEventHandler
 from PySide6.QtCore import QCoreApplication
 from watchdog.observers import Observer
 import os
+from Front.back.Program_Run.utils import resource_path
 
 
 class ComboBoxUpdater(FileSystemEventHandler):
     def __init__(self, comboBox, sql_folder):
         super().__init__()
         self.comboBox = comboBox
-        self.sql_folder = sql_folder
+        self.sql_folder = resource_path(sql_folder)
         self.selected_db = None  # 新增属性来存储当前选中的数据库
 
     def on_modified(self, event):

Front/back/Program_Run/utils.py

@@ -0,0 +1,17 @@
+# utils.py
+import sys
+import os
+
+def resource_path(relative_path):
+    """获取资源的绝对路径，支持开发和打包环境"""
+    try:
+        # 如果是打包环境，base_path 是临时目录
+        base_path = sys._MEIPASS
+        # 获取安装目录
+        install_dir = os.path.dirname(sys.executable)
+        base_path = install_dir  # 使用安装目录作为基础路径
+    except AttributeError:
+        # 如果是开发环境，base_path 是项目根目录
+        base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
+
+    return os.path.join(base_path, relative_path)

Front/main.py

@@ -35,16 +35,8 @@ from Front.back.WD.WDExport import main_function_example
 from Front.modules import *
 from Front.modules import Settings
 from Front.widgets import *
-
-def resource_path(relative_path):
-    """ Get the absolute path to the resource, works for dev and for cx_Freeze """
-    try:
-        # cx_Freeze creates a temp folder and stores path in _MEIPASS
-        base_path = sys._MEIPASS
-    except Exception:
-        base_path = os.path.abspath(".")
-
-    return os.path.join(base_path, relative_path)
+# main.py
+from Front.back.Program_Run.utils import resource_path  # 从 utils.py 导入 resource_path
 
 # IMPORT / GUI AND MODULES AND WIDGETS
 # ///////////////////////////////////////////////////////////////
@@ -230,7 +222,7 @@ class ElTree(QWidget):
         # 获取数据库路径和表名
         dbname = data['listData'][0]
         tablename = data['listData'][2]
-        dbpath = resource_path(os.path.join('SQL', f"{dbname}.db"))
+        dbpath = resource_path(os.path.join('SQL', f"{dbname}.db"))  # 使用 resource_path 获取数据库路径
 
         # 确保 tablename 是 UTF-8 编码的字符串
         try:
@@ -970,8 +962,9 @@ class MainWindow(QMainWindow):
             pass
         else:
             # 数据库目录
-            inpath = os.path.abspath('../SQL')
-            file_path = inpath + '/' + str3 + '.db'
+            inpath = resource_path('SQL')
+            file_path = inpath + '\\' + str3 + '.db'
+            print(file_path)
             # 数据库路径，哪种方法，表名
             self.selectModel = UIFunctions.search_data_to_show(self, file_path, current_text, str1)
             # self.ui.resultTableView1.doubleClicked.connect(self.seleceModel_itemclicked)
@@ -1008,8 +1001,8 @@ class MainWindow(QMainWindow):
             pass
         else:
             # 数据库目录
-            inpath = os.path.abspath('../SQL')
-            file_path = inpath + '/' + str3 + '.db'
+            inpath = resource_path('SQL')
+            file_path = inpath + '\\' + str3 + '.db'
             # 数据库路径，哪种方法，表名
             self.selectModel = UIFunctions.search_data_to_show(self, file_path, current_text, str1)
             # self.ui.resultTableView1.doubleClicked.connect(self.seleceModel_itemclicked)
@@ -1077,7 +1070,7 @@ class MainWindow(QMainWindow):
         parent_item = select_item.parent()
         parent_name = parent_item.text(0)
         dbname = parent_item.parent().text(0)
-        dbpath = os.path.join(os.path.abspath('../SQL'), f"{dbname}.db")
+        dbpath = resource_path(os.path.join('SQL', f"{dbname}.db"))  # 使用 resource_path 获取数据库路径
         # 重新组织表的内容，方便重新计算
         UIFunctions.update_to_db(self, list1, parent_name, tablename_utf8, dbpath)
         # 1秒后自动跳转

Front/modules/ui_functions.py

@@ -14,6 +14,8 @@
 #
 # ///////////////////////////////////////////////////////////////
 from PySide6.QtWidgets import QMessageBox
+
+from Front.back.Program_Run.utils import resource_path
 # MAIN FILE
 # ///////////////////////////////////////////////////////////////
 
@@ -307,12 +309,14 @@ class UIFunctions(MainWindow):
         current_text = self.ui.comboBox_2.currentText()
         # db_path = r"D:/Code/ControlNetwork/UI/SQL/DataBase.db"
         # 获取当前脚本所在的目录
-        current_dir = os.getcwd()
-        # 构建 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")
+        # current_dir = os.getcwd()
+        # # 构建 SQL 文件夹的相对路径（上一级的上一级中的 SQL 文件夹）
+        # sql_folder = os.path.join(current_dir, '..', '..', 'SQL')
+        # # 将相对路径转换为绝对路径
+        # sql_folder = os.path.abspath(sql_folder)
+        # db_path = os.path.join(resource_path(sql_folder), f"{self.ui.comboBox.currentText()}.db")
+        db_path = os.path.join(resource_path('SQL'), f"{self.ui.comboBox.currentText()}.db")
+        print(db_path)
         file_path = file_path
         if current_text == "水准测段高差稳定计算":
             GC.main_function(file_path, db_path)
@@ -327,13 +331,14 @@ class UIFunctions(MainWindow):
     # 计算与展示文件的方法
     def compute_show_process_excel_file(self, file_path):
         current_text = self.ui.comboBox_2.currentText()
-        # 获取当前脚本所在的目录
-        current_dir = os.getcwd()
-        # 构建 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")
+        # # 获取当前脚本所在的目录
+        # current_dir = os.getcwd()
+        # # 构建 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")
+        db_path = os.path.join(resource_path('SQL'), f"{self.ui.comboBox.currentText()}.db")
         # 转换为utf-8
         excelname = os.path.basename(file_path)  # 文件名
         utf_en = excelname.encode('utf-8')  # 转换文件名为utf-8编码形式
@@ -350,13 +355,14 @@ class UIFunctions(MainWindow):
     # 文件导出的方法
     def export_database_to_excel(self, file_path):
         current_text = self.ui.comboBox_2.currentText()
-        # 获取当前脚本所在的目录
-        current_dir = os.getcwd()
-        # 构建 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")
+        # # 获取当前脚本所在的目录
+        # current_dir = os.getcwd()
+        # # 构建 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")
+        db_path = os.path.join(resource_path('SQL'), f"{self.ui.comboBox.currentText()}.db")
         # 转换为utf-8
         excelname = os.path.basename(file_path)  # 文件名
         utf_en = excelname.encode('utf-8')  # 转换文件名为utf-8编码形式
@@ -470,6 +476,7 @@ class UIFunctions(MainWindow):
     #编辑完后的数据更新
     #输入模型，方法,表名utf8，数据库
     def update_to_db(self,model1,methodName, tableName_utf8, dbPath):
+        print(dbPath)
         db1 = sqlite3.connect(dbPath)
         cursor1 = db1.cursor()
         # 通过行锁定对应的字段

Front/modules/ui_main.py

@@ -10,6 +10,8 @@
 
 import sys
 
+from ..back.Program_Run.utils import resource_path
+
 sys.path.append("../..")
 
 import sqlite3
@@ -1314,10 +1316,10 @@ class Ui_MainWindow(object):
 
         # 构建 SQL 文件夹的相对路径（上一级的上一级中的 SQL 文件夹）
         # sql_folder = os.path.join(current_dir, '..', '..', 'SQL')
-        sql_folder = os.path.join(os.getcwd(), 'SQL')
+        sql_folder = os.path.join('SQL')
 
         # 将相对路径转换为绝对路径
-        sql_folder = os.path.abspath(sql_folder)
+        sql_folder = resource_path(sql_folder)
 
         # 确保 SQL 文件夹存在
         if not os.path.exists(sql_folder):