392 lines
14 KiB
Python
392 lines
14 KiB
Python
#!/opt/homebrew/bin/python3
|
||
# -*- coding:utf-8 -*-
|
||
import json
|
||
import shutil
|
||
import datetime
|
||
import uuid
|
||
import os
|
||
from PyQt6 import *
|
||
from PyQt6.QtCore import *
|
||
from logs import log
|
||
from typing import Union
|
||
from config import config
|
||
from common import common
|
||
from userModel.userManager import userManager
|
||
class ProjectManager(QObject):
|
||
sig_projectChanged = pyqtSignal(str)
|
||
def __init__(self, parent=None):
|
||
super().__init__()
|
||
self.currentProId = ""
|
||
self.basePath = config.data["project"]["base_path"]
|
||
if not os.path.exists(self.basePath):
|
||
os.mkdir(self.basePath)
|
||
self.path = ""
|
||
self.historyPath = ""
|
||
self.logPath = ""
|
||
self.cmdHistoryPath = ""
|
||
self.reportPath = ""
|
||
self.projectDict = {}
|
||
self.projectTaskDict = {}
|
||
self.currentProInfo = {}
|
||
self.rowCount = 100
|
||
self.batch_size = 50 # 批量写入的条数
|
||
self.timeout = 5000 # 超时（毫秒）
|
||
self.log_buffer = [] # 日志缓存
|
||
# self.timer = QTimer(self)
|
||
# self.timer.timeout.connect(self.flush_logs) # 超时触发写入
|
||
# self.timer.start(10) # 启动定时器
|
||
|
||
@pyqtSlot(str,str, result=str)
|
||
def create(self, name, remark):
|
||
id = str(uuid.uuid4())
|
||
self.setCurProPath(id)
|
||
userData = userManager.getCurrentUser()
|
||
userName = "unknown"
|
||
if userData:
|
||
userName = userData["name"]
|
||
if not os.path.exists(self.path):
|
||
os.mkdir(self.path)
|
||
if not os.path.exists(self.historyPath):
|
||
os.mkdir(self.historyPath)
|
||
if not os.path.exists(self.logPath):
|
||
os.mkdir(self.logPath)
|
||
if not os.path.exists(self.reportPath):
|
||
os.mkdir(self.reportPath)
|
||
with open(os.path.join(self.path, "info.txt"), "w") as f:
|
||
f.write(json.dumps({
|
||
"id": id,
|
||
"name": name,
|
||
"path": os.path.normpath(self.path),
|
||
"remark": remark,
|
||
"operator": userName,
|
||
"attr": {},
|
||
"createDate": str(QDateTime.currentDateTime().toString("yyyy-MM-dd hh:mm:ss")),
|
||
"lastDate": str(QDateTime.currentDateTime().toString("yyyy-MM-dd hh:mm:ss"))
|
||
}))
|
||
return id
|
||
|
||
def setCurProPath(self,id):
|
||
if id == common.emputyProId:
|
||
self.path = ""
|
||
self.historyPath = ""
|
||
self.logPath = ""
|
||
self.reportPath = ""
|
||
else:
|
||
self.path = os.path.join(self.basePath, id)
|
||
self.historyPath = os.path.join(self.path, "history")
|
||
self.logPath = os.path.join(self.path, "log")
|
||
self.reportPath = os.path.join(self.path, "report")
|
||
|
||
@pyqtSlot(list, result=bool)
|
||
def saveProTask(self, taskInfo, params):
|
||
try:
|
||
proId = self.currentProId
|
||
param = {"task": taskInfo, "params": params}
|
||
return self.updateProTask(proId, param)
|
||
except Exception as e:
|
||
print(f"Error addTask: {e}")
|
||
return False
|
||
|
||
@pyqtSlot(str, QVariant, result=bool)
|
||
def updateProTask(self, id, param = None):
|
||
try:
|
||
if not isinstance(param, dict):
|
||
param = param.toVariant()
|
||
data = {}
|
||
infoPath = os.path.join(self.basePath, id, "task.txt")
|
||
if "task" in param:
|
||
data["task"] = param["task"]
|
||
if "params" in param:
|
||
data["params"] = param["params"]
|
||
with open(infoPath, 'w', encoding="utf-8") as file:
|
||
json.dump(data, file, indent=4)
|
||
return True
|
||
except Exception as e:
|
||
return False
|
||
|
||
@pyqtSlot(str,QVariant, result=bool)
|
||
def addHistory(self, proId, historyInfo):
|
||
try:
|
||
if not isinstance(historyInfo, dict):
|
||
historyInfo = historyInfo.toVariant()
|
||
historyInfo["time"] = str(QDateTime.currentDateTime().toString("yyyy-MM-dd hh:mm:ss"))
|
||
userData = userManager.getCurrentUser()
|
||
userName = "unknown"
|
||
if userData:
|
||
userName = userData["name"]
|
||
historyInfo["operator"] = userName
|
||
return self.updateHistory(proId, historyInfo)
|
||
except Exception as e:
|
||
print(f"Error addHistory: {e}")
|
||
return False
|
||
|
||
@pyqtSlot(str, QVariant, result=bool)
|
||
def updateHistory(self, id, param=None):
|
||
try:
|
||
if not isinstance(param, dict):
|
||
param = param.toVariant()
|
||
|
||
# 设置命令历史文件路径
|
||
self.cmdHistoryPath = os.path.join(self.basePath, id, "cmdhistory.txt")
|
||
self.add_log(param) # 添加日志到缓存
|
||
return True
|
||
except Exception as e:
|
||
log.error(f"更新历史记录失败: {str(e)}") # 记录错误日志
|
||
return False
|
||
|
||
def add_log(self, data):
|
||
"""添加日志到缓存"""
|
||
try:
|
||
self.log_buffer.append(data)
|
||
if len(self.log_buffer) >= self.batch_size:
|
||
self.flush_logs() # 达到条数限制时立即写入
|
||
except Exception as e:
|
||
print(f"添加日志错误: {str(e)}")
|
||
|
||
def flush_logs(self):
|
||
"""将缓存中的日志写入文件"""
|
||
if not self.log_buffer:
|
||
return # 如果缓存为空则不写入
|
||
|
||
log_path = self.cmdHistoryPath # 获取日志文件路径
|
||
if not log_path:
|
||
self.log_buffer.clear()
|
||
return
|
||
|
||
try:
|
||
# 检查文件大小并进行轮换
|
||
if os.path.exists(log_path) and os.path.getsize(log_path) > 100*1024*1024: # 100MB
|
||
self.rotate_history_file(log_path)
|
||
with open(log_path, "a+", encoding='utf-8') as f:
|
||
for entry in self.log_buffer:
|
||
json.dump(entry, f, ensure_ascii=False)
|
||
f.write('\n')
|
||
self.log_buffer.clear() # 清空缓存
|
||
except Exception as e:
|
||
print(f"写入日志失败: {str(e)}")
|
||
|
||
|
||
# def stop(self):
|
||
# """停止日志写入"""
|
||
# # self.timer.stop()
|
||
# self.flush_logs() # 在停止前写入剩余日志
|
||
|
||
def rotate_history_file(self, file_path):
|
||
"""日志轮转实现"""
|
||
base_name = os.path.basename(file_path).split('.')[0]
|
||
log_dir = os.path.dirname(file_path)
|
||
|
||
# 删除最旧日志
|
||
existing_logs = sorted([f for f in os.listdir(log_dir) if f.startswith(base_name)])
|
||
if len(existing_logs) >= 30:
|
||
os.remove(os.path.join(log_dir, existing_logs[0]))
|
||
|
||
# 重命名当前日志
|
||
timestamp = QDateTime.currentDateTime().toString("yyyyMMddHHmmss")
|
||
new_name = f"{base_name}_{timestamp}.bak.txt"
|
||
os.rename(file_path, os.path.join(log_dir, new_name))
|
||
@pyqtSlot(str, QVariant, result=bool)
|
||
def update(self, id, param = None):
|
||
try:
|
||
if not isinstance(param, dict):
|
||
param = param.toVariant()
|
||
data = {}
|
||
infoPath = os.path.join(self.basePath, id, "info.txt")
|
||
with open(infoPath, 'r', encoding="utf-8") as file:
|
||
content = file.read()
|
||
data = json.loads(content)
|
||
if "name" in param:
|
||
data["name"] = param["name"]
|
||
if "remark" in param:
|
||
data["remark"] = param["remark"]
|
||
if "attr" in param:
|
||
data["attr"] = param["attr"]
|
||
data["lastDate"] = str(QDateTime.currentDateTime().toString("yyyy-MM-dd hh:mm:ss"))
|
||
with open(infoPath, 'w', encoding="utf-8") as file:
|
||
json.dump(data, file, indent=4)
|
||
return True
|
||
except Exception as e:
|
||
return False
|
||
|
||
@pyqtSlot(str, result=bool)
|
||
@pyqtSlot(list, result=bool)
|
||
def delete(self, id):
|
||
try:
|
||
if isinstance(id, str):
|
||
proDir = os.path.join(self.basePath, id)
|
||
shutil.rmtree(proDir, ignore_errors=False, onerror=None)
|
||
return True
|
||
else:
|
||
ids = id
|
||
for _id in ids:
|
||
proDir = os.path.join(self.basePath, _id)
|
||
shutil.rmtree(proDir, ignore_errors=False, onerror=None)
|
||
return True
|
||
except Exception as e:
|
||
print(f"Error deletePro: {e}")
|
||
return False
|
||
|
||
@pyqtSlot(str, result=bool)
|
||
def isProjectExist(self, id):
|
||
if isinstance(id, str):
|
||
if id in self.projectDict:
|
||
return True
|
||
else:
|
||
return False
|
||
else:
|
||
return False
|
||
|
||
@pyqtSlot(str, result=bool)
|
||
def updateProId(self, id):
|
||
if id == common.emputyProId:
|
||
self.setCurProPath(id)
|
||
self.sig_projectChanged.emit(id)
|
||
return False
|
||
|
||
self.currentProId = id
|
||
self.currentProInfo = self.projectDict[id]
|
||
self.setCurProPath(id)
|
||
self.sig_projectChanged.emit(id)
|
||
return True
|
||
|
||
@pyqtSlot(str, result=bool)
|
||
def setCurrentProId(self, id):
|
||
if id == common.emputyProId:
|
||
self.setCurProPath(id)
|
||
self.sig_projectChanged.emit(id)
|
||
return False
|
||
if self.currentProId != id:
|
||
self.currentProId = id
|
||
self.currentProInfo = self.projectDict[id]
|
||
self.setCurProPath(id)
|
||
self.sig_projectChanged.emit(id)
|
||
return True
|
||
|
||
@pyqtSlot(result=QVariant)
|
||
def getCurrentProInfo(self):
|
||
return self.currentProInfo
|
||
|
||
@pyqtSlot(result=str)
|
||
def getCurrentProId(self):
|
||
return self.currentProId
|
||
|
||
@pyqtSlot(result=str)
|
||
def getProPath(self):
|
||
if self.currentProId != "":
|
||
if not os.path.exists(self.path):
|
||
os.mkdir(self.path)
|
||
return self.path
|
||
else:
|
||
return ""
|
||
|
||
@pyqtSlot(result=str)
|
||
def getLogPath(self):
|
||
if self.currentProId != "" and self.logPath != "":
|
||
if not os.path.exists(self.logPath):
|
||
os.mkdir(self.logPath)
|
||
return self.logPath
|
||
else:
|
||
return ""
|
||
|
||
@pyqtSlot(result=str)
|
||
def getHistoryPath(self):
|
||
if self.currentProId != "":
|
||
if not os.path.exists(self.historyPath):
|
||
os.mkdir(self.historyPath)
|
||
return self.historyPath
|
||
else:
|
||
return ""
|
||
|
||
@pyqtSlot(result=str)
|
||
def getReportPath(self):
|
||
if self.currentProId != "":
|
||
if not os.path.exists(self.reportPath):
|
||
os.mkdir(self.reportPath)
|
||
return self.reportPath
|
||
else:
|
||
return ""
|
||
|
||
@pyqtSlot(result=str)
|
||
def getBasePath(self):
|
||
return self.basePath
|
||
|
||
@pyqtSlot(str, result=QVariant)
|
||
def getInfo(self, id):
|
||
try:
|
||
if isinstance(id, str) and id == "all":
|
||
self.projectDict = {}
|
||
# 按时间排序
|
||
dirs = sorted([d for d in os.listdir(self.basePath) if os.path.isdir(os.path.join(self.basePath, d))], key=lambda x: os.path.getmtime(os.path.join(self.basePath, x)), reverse=True)
|
||
# 遍历每个子目录
|
||
for dir_name in dirs:
|
||
dir_path = os.path.join(self.basePath, dir_name)
|
||
# 如果当前文件是info.txt，打印其路径
|
||
file = "info.txt"
|
||
# 获取info文件的路径
|
||
file_path = os.path.join(dir_path, file)
|
||
if os.path.exists(file_path):
|
||
# 读取文件内容
|
||
with open(file_path, 'r', encoding="utf-8") as f:
|
||
content = f.read()
|
||
info = json.loads(content)
|
||
self.projectDict[info["id"]] = info
|
||
return self.projectDict
|
||
else:
|
||
file_path = os.path.join(self.basePath, id, "info.txt")
|
||
if os.path.exists(file_path):
|
||
with open(file_path, 'r', encoding="utf-8") as f:
|
||
content = f.read()
|
||
info = json.loads(content)
|
||
return info
|
||
except Exception as e:
|
||
print(f"Error getInfo: {e}")
|
||
return None
|
||
|
||
@pyqtSlot(str, result=QVariant)
|
||
def getProTask(self, id):
|
||
try:
|
||
if id == "":
|
||
self.projectTaskDict[id] = {}
|
||
return {}
|
||
else:
|
||
file_path = os.path.join(self.basePath, id, "task.txt")
|
||
if os.path.exists(file_path):
|
||
with open(file_path, 'r', encoding="utf-8") as f:
|
||
content = f.read()
|
||
task = json.loads(content)
|
||
self.projectTaskDict[id] = task or {}
|
||
return task or {}
|
||
else:
|
||
self.projectTaskDict[id] = {}
|
||
return {}
|
||
|
||
except Exception as e:
|
||
self.projectTaskDict[id] = {}
|
||
print(f"Error getInfo: {e}")
|
||
return {}
|
||
|
||
@pyqtSlot(str, result=list)
|
||
def getCmdHistory(self, id):
|
||
try:
|
||
if id == "":
|
||
return []
|
||
else:
|
||
file_path = os.path.join(self.basePath, id, "cmdhistory.txt")
|
||
if os.path.exists(file_path):
|
||
with open(file_path, 'r', encoding="utf-8") as f:
|
||
content = f.read()
|
||
history_entries = content.strip().split('\n')
|
||
if self.rowCount > len(history_entries):
|
||
self.rowCount = 0
|
||
history = []
|
||
for entry in history_entries[-self.rowCount:]:
|
||
history.append(json.loads(entry))
|
||
return history or []
|
||
else:
|
||
return []
|
||
|
||
except Exception as e:
|
||
print(f"Error getInfo: {e}")
|
||
return []
|
||
|
||
projectManager = ProjectManager() |