diff --git a/.gitignore b/.gitignore index 8558508..d98c921 100644 --- a/.gitignore +++ b/.gitignore @@ -5,3 +5,6 @@ build/ ~$据库本地运行.docx 数据库组件本地安装/ 数据库组件本地安装.zip +ComacDBInstall.spec +src/__pycache__/Test.cpython-311.pyc +dist/ComacDBInstall.exe diff --git a/build.txt b/build.txt index b12806c..951e610 100644 --- a/build.txt +++ b/build.txt @@ -1,2 +1 @@ -pyinstaller --onefile --add-data "datas/mariadb.zip:datas" --add-data "datas/init.sql:datas" --add-data "datas/jdk.zip:datas" --add-data "datas/start.bat:datas" src/install.py -pyinstaller --onefile --add-data "datas/electromagnetic.jar:datas" src/start.py \ No newline at end of file +pyinstaller --onefile --add-data "datas/electromagnetic.jar:datas" --add-data "datas/mariadb.zip:datas" --add-data "datas/init.sql:datas" --add-data "datas/jdk.zip:datas" --add-data "datas/start.bat:datas" --add-data "datas/nssm.exe:datas" src/ComacDBInstall.py \ No newline at end of file diff --git a/datas/nssm.exe b/datas/nssm.exe new file mode 100644 index 0000000..6ccfe3c Binary files /dev/null and b/datas/nssm.exe differ diff --git a/dist/install.exe b/dist/install.exe deleted file mode 100644 index e2de8c6..0000000 Binary files a/dist/install.exe and /dev/null differ diff --git a/dist/start.exe b/dist/start.exe deleted file mode 100644 index 07f93db..0000000 Binary files a/dist/start.exe and /dev/null differ diff --git a/src/ComacDBInstall.py b/src/ComacDBInstall.py new file mode 100644 index 0000000..ce7a2ce --- /dev/null +++ b/src/ComacDBInstall.py @@ -0,0 +1,272 @@ +import os +import shutil +import subprocess +import sys +import time +import zipfile +from datetime import datetime +from pathlib import Path + +from loguru import logger + + +def ensure_dir(directory): + if not os.path.exists(directory): + os.makedirs(directory) + logger.info(f"目录 {directory} 已创建") + else: + logger.info(f"目录 {directory} 已存在") + +def get_resource_path(relative_path): + """ 获取资源绝对路径，适用于开发环境和PyInstaller打包后 """ + if hasattr(sys, '_MEIPASS'): + # 打包后的资源路径：sys._MEIPASS + 相对路径 + base_path = sys._MEIPASS + else: + # 开发环境的基础路径 + base_path = os.path.dirname(os.path.abspath(".")) + return os.path.join(base_path, relative_path) + +def check_service_exist(service_name): + result = subprocess.run(['sc', 'query', service_name], + capture_output=True, text=True) + if '1060' in result.stdout: # 服务不存在 + return False + return True + +def start_service(service_name): + result = subprocess.run(['sc', 'query', service_name], + capture_output=True, text=True) + if 'RUNNING' in result.stdout: + logger.info(f"服务 {service_name} 已经在运行") + return True + else: + logger.info(f"服务 {service_name} 当前未运行") + logger.info("尝试启动服务...") + + # 尝试启动服务 + start_result = subprocess.run(['sc', 'start', service_name], + capture_output=True, text=True) + + if start_result.returncode == 0: + logger.info(f"服务 {service_name} 启动成功") + return True + else: + logger.info(f"启动服务 {service_name} 失败: {start_result.stderr}") + return False + +def unzip_file(zip_file_path): + new_path = get_resource_path(os.path.join('datas', 'mariadb.zip')) + with zipfile.ZipFile(str(new_path), 'r') as zip_ref: + zip_ref.extractall(zip_file_path) # 将文件解压到指定路径 + +def start_service_if_not_running(service_name): + result = subprocess.run(['sc', 'query', service_name], + capture_output=True, text=True) + if 'RUNNING' in result.stdout: + logger.info(f"服务 {service_name} 已经在运行") + else: + logger.info(f"服务 {service_name} 当前未运行") + logger.info("尝试启动服务...") + + # 尝试启动服务 + start_result = subprocess.run(['sc', 'start', service_name], + capture_output=True, text=True) + + if start_result.returncode == 0: + logger.info(f"服务 {service_name} 启动成功") + else: + logger.info(f"启动服务 {service_name} 失败: {start_result.stderr}") + raise MyCustomError(rf"服务 {service_name} 启动失败") + + +class MyCustomError(Exception): + """自定义异常""" + + def __init__(self, message="发生了一个错误"): + # 调用基类的构造函数 + super().__init__(message) + +class InstallMariaDb: + + def __init__(self): + self.service_name = 'ComacMariaDB' + self.root_path = r'D:\database' + self.mariadb_data_path = r'D:\database\mariadb\data' + self.mariadb_port = 3417 + self.passowrd = '1qaz@WSX' + self.init_schema = 'em_data_prod' + self.init_sql = r'D:/database/mariadb/data/init.sql' + pass + + def start_install_mariadb(self): + existStatus = check_service_exist(self.service_name) + if existStatus: + start_service_if_not_running(self.service_name) + return + logger.info("开始安装MariaDB") + ensure_dir(self.root_path) + logger.info("开始解压文件") + unzip_file(self.root_path) + logger.info(f"开始注册服务 {self.service_name}") + self.__register_service() + logger.info("启动数据库") + self.__start_mariadb() + logger.info("数据库用户初始化") + self.__init_db_user() + logger.info("数据库初始化") + self.__init_db() + logger.info("设置Java运行环境") + self.__set_java_env() + logger.info("生成脚本文件") + self.__set_bat() + logger.info("安装完成，10秒后自动退出") + time.sleep(10) + pass + + def __start_mariadb(self): + command = 'net start ' + self.service_name + with os.popen(command) as stream: + res = stream.read() + logger.info(res) + + def __register_service(self): + args1 = self.root_path + '/mariadb/bin/mysql_install_db.exe' + args2 = '--datadir=' + self.mariadb_data_path + args3 = '--service=' + self.service_name + args4 = '--password=' + self.passowrd + args5 = '--port=' + str(self.mariadb_port) + res = subprocess.run([args1, args2, args3, args4, args5], text=True, capture_output=True) + logger.info(res.stdout) + return res + + def __init_db_user(self): + command1 = fr'{self.root_path}\mariadb\bin\mysql -u root -p{self.passowrd} -P {self.mariadb_port} -e "create database if not exists {self.init_schema}; use {self.init_schema};"' + with os.popen(command1) as stream: + res = stream.read() + logger.info(res) + + def __init_db(self): + new_path = get_resource_path(os.path.join('datas', 'init.sql')) + shutil.copy(str(new_path), self.mariadb_data_path) + command2 = fr'{self.root_path}\mariadb\bin\mysql --no-defaults -u root -p{self.passowrd} -P {self.mariadb_port} {self.init_schema} < {self.init_sql}' + with os.popen(command2) as stream: + res2 = stream.read() + logger.info(res2) + os.remove(self.init_sql) + pass + + def __set_java_env(self): + new_path = get_resource_path(os.path.join('datas', 'jdk.zip')) + with zipfile.ZipFile(str(new_path), 'r') as zip_ref: + zip_ref.extractall(self.root_path) # 将文件解压到指定路径 + pass + # + # def __set_bat(self): + # new_path = get_resource_path(os.path.join('datas', 'start.bat')) + # shutil.copy(str(new_path), self.root_path) + + pass + +class InstallComacDb: + + def __init__(self): + + self.project_dir = get_resource_path(os.path.dirname(Path(__file__).parent.absolute())) + self.app_log_dir = 'D:/database/logs' + self.comac_db_running_port = 12396 + self.jar_path = get_resource_path(os.path.join("datas", "electromagnetic.jar")) + self.new_java_path = 'D:/database/jdk/bin/java.exe' + self.url = f'http://127.0.0.1:{self.comac_db_running_port}/index' + self.service_name = "comacDatabase" + self.service_description = "数据库组件服务" + # 路径配置 + self.datas_dir = get_resource_path(os.path.join('datas')) + # NSSM可执行文件路径 + self.nssm_exe = get_resource_path(os.path.join('datas', 'nssm.exe')) + # Java和JAR文件路径 + self.java_path = "D:/database/jdk/bin/java.exe" + ensure_dir(self.app_log_dir) + pass + + def start_comac_db(self): + self.__delete_old_files(self.app_log_dir) + existStatus = check_service_exist(self.service_name) + if existStatus: + start_service_if_not_running(self.service_name) + return + + self.__register_and_start_service() + pass + + def __delete_old_files(self, directory, days=2): + # 计算时间阈值（当前时间 - days天） + threshold_time = time.time() - days * 24 * 60 * 60 + deleted_count = 0 + try: + # 遍历目录中的文件 + for filename in os.listdir(directory): + filepath = os.path.join(directory, filename) + + # 确保是文件而不是目录 + if os.path.isfile(filepath): + try: + # 获取文件创建时间（Windows系统） + creation_time = os.path.getctime(filepath) + + # 检查文件是否超过阈值 + if creation_time < threshold_time: + # 删除文件 + os.remove(filepath) + deleted_count += 1 + logger.info(f"已删除: {filename}") + except Exception as e: + logger.info(f"处理文件 {filename} 时出错: {str(e)}", file=sys.stderr) + + logger.info(f"\n操作完成！共删除 {deleted_count} 个日志文件。") + except Exception as e: + logger.info(f"遍历目录时出错: {str(e)}", file=sys.stderr) + + + def __register_and_start_service(self): + # 服务配置 + # 构建NSSM安装命令 + formatted_time = datetime.now().strftime("%Y%m%d%H%M%S") + cmd = [ + str(self.nssm_exe), + "install", + self.service_name, + self.java_path, + f"-jar {self.jar_path} --logging.file.path={self.app_log_dir} --logging.file.name={self.app_log_dir}/app_{formatted_time}.log" + ] + + # 执行安装命令 + print("Installing service...") + result = subprocess.run(cmd, capture_output=True, text=True) + + if result.returncode != 0: + print("Error installing service:") + print(result.stderr) + return False + + print("Service installed successfully!") + + # 设置服务显示名称和描述 + subprocess.run([str(self.nssm_exe), "set", self.service_name, "DisplayName", self.service_name]) + subprocess.run([str(self.nssm_exe), "set", self.service_name, "Description", self.service_description]) + + # 设置启动目录（可选，但推荐） + # subprocess.run([str(nssm_exe), "set", service_name, "AppDirectory", str(datas_dir)]) + + # 设置启动类型为自动（可选） + subprocess.run([str(self.nssm_exe), "set", self.service_name, "Start", "SERVICE_AUTO_START"]) + + print("Service configuration completed.") + print(f"You can now start the service with: net start {self.service_name}") + subprocess.run(rf"net start {self.service_name}") + time.sleep(10) + +if __name__ == '__main__': + InstallMariaDb().start_install_mariadb() + InstallComacDb().start_comac_db() + pass \ No newline at end of file diff --git a/src/install.py b/src/install.py index d72b33a..d47b2da 100644 --- a/src/install.py +++ b/src/install.py @@ -7,6 +7,7 @@ import loguru import elevate import shutil import psutil +import re root_path = r'D:\database' service_name = 'ComacMariaDB' @@ -15,6 +16,8 @@ mariadb_data_path = r'D:\database\mariadb\data' init_sql = r'D:/database/mariadb/data/init.sql' mariadb_port = 3417 init_schema = 'em_data_prod' +db_running_port=12396 +db_running_service_name="ComacDataBase" def ensure_dir(directory): if not os.path.exists(directory): @@ -103,25 +106,66 @@ def start_install(): status = check_service() if status: return - loguru.logger.info("开始安装MariaDB") - ensure_dir(root_path) - loguru.logger.info("开始解压文件") - unzip_file() - loguru.logger.info("开始注册服务") - register_service() - loguru.logger.info("启动数据库") - start_mariadb() - loguru.logger.info("数据库用户初始化") - init_db_user() - loguru.logger.info("数据库初始化") - init_db() - loguru.logger.info("设置Java运行环境") - set_java_env() - loguru.logger.info("生成脚本文件") - set_bat() - loguru.logger.info("安装完成，10秒后自动退出") - time.sleep(10) + else: + loguru.logger.info("开始安装MariaDB") + ensure_dir(root_path) + loguru.logger.info("开始解压文件") + unzip_file() + loguru.logger.info("开始注册服务") + register_service() + loguru.logger.info("启动数据库") + start_mariadb() + loguru.logger.info("数据库用户初始化") + init_db_user() + loguru.logger.info("数据库初始化") + init_db() + loguru.logger.info("设置Java运行环境") + set_java_env() + loguru.logger.info("生成脚本文件") + set_bat() + loguru.logger.info("安装完成，10秒后自动退出") + time.sleep(10) + +def check_service_exists(query_service_name): + try: + output = subprocess.check_output(['sc', 'query', query_service_name], stderr=subprocess.STDOUT) + return True + except subprocess.CalledProcessError as e: + if '1060' in str(e.output): # 错误代码1060表示服务不存在 + return False + raise + +def start_app(): + + # 首先检查有没有数据库运行的服务名，如果没有，则需要创建， + + + pass + + + + +def kill_process_by_port(run_port): + # 执行netstat命令获取端口占用信息 + cmd_netstat = ['netstat', '-ano', '|', 'findstr', fr':{run_port}'] + result = subprocess.run(cmd_netstat, capture_output=True, text=True, shell=True) + out = result.stdout + if out: + arr = re.split(r'\s+', out) + pid = arr[5] + if pid: + try: + subprocess.run(['taskkill', '/F', '/PID', pid], check=True) + loguru.logger.info(f"已终止占用端口 {run_port} 的进程 (PID: {pid})") + except subprocess.CalledProcessError: + loguru.logger.info(f"终止进程 {pid} 失败 (可能权限不足或进程不存在)") + else: + loguru.logger.info(f"端口 {run_port} 未被占用") + + + if __name__ == '__main__': elevate.elevate() start_install() + start_app() diff --git a/src/start.py b/src/start.py index e010e75..dacb64f 100644 --- a/src/start.py +++ b/src/start.py @@ -87,7 +87,7 @@ def start(): creationflags=creation_flags ) - logger.info(f"应用已在后台启动! PID: {process.pid}") + logger.info(f"应用正在启动! PID: {process.pid}") logger.info(f"日志输出: {os.path.abspath(log_file)}") except Exception as e: logger.info(f"启动失败: {str(e)}")