flightinfoV3_QT/Functions/QtFunctions/workloadinfos.py

import datetime
import json
import os
import time

import openpyxl
from PyQt6 import QtCore
from PyQt6.QtCore import Qt , QTimer
from PyQt6.QtGui import QTextDocument , QTextCursor , QTextCharFormat , QColor , QTextOption , QPalette
from PyQt6.QtWidgets import QWidget , QMessageBox , QTableWidgetItem , QFileDialog , QStyledItemDelegate , QStyle , \
    QMenu

from UI.workloadinfos import Ui_workloadinfos
from .LogerinTxt import app_logger
from Functions.apiFunction import api_fc

filename = "D:\\flightinfo\\logs\\peoples.json"
class Workloadinfos(Ui_workloadinfos, QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setupUi(self)
        self.ipinfo = None
        self.user = None
        self.selectedbc=None
        self.selectedtime=None
        self.doubleclickLock = 0
        self.headerLabel = ["序号", "交", "姓名", "备", "角色", "备", "授权", "备", "工作量", "备", "当前工作量", "备", "总工作量", "备",
                            "总工时", "备", "当前工作", "备", "修正工时", "备", "修正详情", "备注信息", "备", "备", "备", "备", "备", "备",
                            "1","2","3","4","5","6","7","8","9","10","11","12","13","14","15","16","17","18","19","20"]
        self.tableWidgetWorkload.setSortingEnabled(True)
        self.tableWidgetWorkload.setColumnCount(len(self.headerLabel))
        self.tableWidgetWorkload.verticalHeader().setVisible(False)
        self.tableWidgetWorkload.setHorizontalHeaderLabels(self.headerLabel)
        self.refresh.clicked.connect(self.refreshnow)
        self.changeupdate.clicked.connect(self.update)
        self.clear.clicked.connect(self.clearselect)
        self.autoupdateMode=False
        self.autoupdate_check.stateChanged.connect(self.on_checkbox_changed)
        # 保存当前排序状态
        self.current_sort_column = -1
        self.current_sort_order = Qt.SortOrder.AscendingOrder
        self.searchinfo.returnPressed.connect(self.search_table)
        self.searchinfo.textChanged.connect(self.search_table_empty)
        self.selectall.clicked.connect(self.selectall_funciton)
        self.load_pushbutton.clicked.connect(self.get_load)
        self.tableWidgetWorkload.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
        self.tableWidgetWorkload.customContextMenuRequested.connect(self.generateMenu)
        self.tableWidgetWorkload.horizontalHeader().sortIndicatorChanged.connect(self.on_sort_changed)
        self.changeupdate.clicked.connect(self.changeupdate_Function)
        self.tableWidgetWorkload.cellDoubleClicked.connect(self.itemclick2)


    def start(self):
        self.show()
        self.peopleresold = self.read_json_to_list()
        try:
            # ip = "http://" + str(self.ipinfo) + "/static/getPglistInDatabase"
            # self.peopleres = requests.get(url=ip, timeout=30).json()['返回值']
            self.peopleres = api_fc.request_get("/static/getPglistInDatabase")['返回值']
            #print(self.peopleres)
        except Exception as e:
            QMessageBox.warning(self, "提示", "服务器连接超时，请联系管理员检查服务器！")
            app_logger.log_error(e)
            self.peopleres = []
        self.peopleselect=self.peopleresold if self.peopleresold else self.peopleres
        self.comboBox_peoplelist.addItems(self.peopleres)
        self.set_initial_selection()
        self.refreshnow()
        self.serchTimer = QTimer()
        self.serchTimer.start(120 * 1000)
        self.serchTimer.timeout.connect(self.autoupdate)

    def itemclick2(self, row, column):
        try:
            # print(self.doubleclickLock)
            if column == 21 and self.doubleclickLock == 0:
                self.doubleclickLockUpdate=1
                self.clickrow = row
                self.clickcolumn = column
                self.olditem = self.tableWidgetWorkload.item(row, column).text()
                item = self.tableWidgetWorkload.item(row, column)
                # 如果单元格对象不存在，则返回默认的背景色
                if not item:
                    self.color=self.tableWidgetWorkload.palette().color(QPalette.Base)
                else:
                    # 获取并返回单元格的背景色
                    self.color = item.background()
                self.tableWidgetWorkload.cellChanged.connect(self.cellchanged)
            else:
                self.doubleclickLockUpdate = 0
        except Exception as e:
            app_logger.log_error(e)
    def cellchanged(self, row, column):
        if self.clickrow == row and self.clickcolumn == column :
            self.clickrow = None
            self.clickcolumn = None
            try:
                self.doubleclickLock = 1
                self.newitem = self.tableWidgetWorkload.item(row, column).text().replace("\"","*").replace("\\",";").replace("\'","*").replace("/","、").replace("#","*")
                name = self.tableWidgetWorkload.item(row, 2).text()
                if column == 21 and self.olditem != self.newitem:
                    item = self.tableWidgetWorkload.item(row, column)
                    if item.text() != "":
                        text = item.text().replace("\"", "*").replace("\\", ";").replace("\'", "*").replace("/", "、").replace("#", "*")
                    else:
                        text = "清空项目12345678987654321"
                    # ip = "http://" + str(self.ipinfo) + "/static/updateNoteItem/" + "%s/" % str(text) + "%s" % str(name)
                    # # print(ip)
                    # ip2 = "http://" + str(self.ipinfo) + "/static/insertLogs2/" + "排班辅助%s的备注模块：由%s变更为%s/" % (name,self.olditem, self.newitem) + "%s/" % str(self.user) + "%s" % str(datetime.datetime.now())
                    try:
                        # requests.get(url=ip, timeout=30).json()
                        # requests.get(url=ip2, timeout=30).json()
                        api_fc.request_post('/static/updateNoteItem', text=str(text), name=str(name))
                        api_fc.request_post('/static/insertLogs2',
                                            text="排班辅助%s的备注模块：由%s变更为%s" % (name,self.olditem, self.newitem),
                                            name=str(self.user),
                                            time=str(datetime.datetime.now()))
                        self.tableWidgetWorkload.setItem(int(row), column, QTableWidgetItem(str(self.newitem)))
                        self.tableWidgetWorkload.item(int(row), column).setBackground(QColor("yellow"))
                        self.tableWidgetWorkload.item(int(row), column).setTextAlignment(Qt.AlignmentFlag.AlignCenter)
                    except Exception as e:
                        app_logger.log_error(e)
                        QMessageBox.warning(self, "提示", "服务器连接超时，请联系管理员检查服务器！")
                self.doubleclickLockUpdate = 0
                self.doubleclickLock = 0
                self.tableWidgetWorkload.cellChanged.disconnect(self.cellchanged)
            except Exception as e:
                self.doubleclickLockUpdate = 0
                self.clickrow = None
                self.clickcolumn = None
                app_logger.log_error(e)
                try:
                    self.tableWidgetWorkload.cellChanged.disconnect(self.cellchanged)
                except:
                    pass
                self.tableWidgetWorkload.setItem(int(row), column, QTableWidgetItem(str(self.olditem)))
                self.tableWidgetWorkload.item(int(row), column).setBackground(QColor(self.color))
                self.tableWidgetWorkload.item(int(row), column).setTextAlignment(Qt.AlignmentFlag.AlignCenter)
                app_logger.log_error(e)
                self.doubleclickLock = 0


    def changeupdate_Function(self):
        day=datetime.datetime.now().strftime("%Y%m%d")
        info = self.changeinfo.text().strip().lower()
        people = self.changepeople.text().strip()
        costtime = self.changevalue.text().strip().lower()
        try:
            costtime = float(costtime)
            if info and people and costtime:
                timestr=str(int(time.time()*1000))
                id= f"人工修正-{info}-{timestr}-{people}"
                data0={
                    "id": id,
                    'people':people,
                    'costtime':costtime,
                    'info':info,
                    'day':day
                }
                try:
                    # ip = "http://" + str(self.ipinfo) + "/static/manchangecosttime"
                    # data = json.dumps(data0)
                    # res = requests.post(url=ip, data=data, timeout=30).json()
                    res = api_fc.request_post('/static/manchangecosttime', data=data0)
                    if res["返回值"] == "ok":
                        QMessageBox.information(self, "提示", "操作成功！")
                        self.refreshnow()
                    else:
                        QMessageBox.warning(self, "提示", "操作失败！")
                except Exception as e:
                    QMessageBox.warning(self, "提示", "服务器连接超时，请联系管理员检查服务器！")
                    app_logger.log_error(e)

            else:
                QMessageBox.warning(self,"警告", "请填写完整信息！！！")
        except ValueError:
            QMessageBox.warning(self,"警告", "工时内请填写数字！！！")

    def generateMenu(self, pos):
        try:
            nowDay = datetime.date.today().strftime("%Y%m%d")
            a_str = nowDay + " 09:00:00"
            b_str = nowDay + " 20:30:00"
            a = datetime.datetime.strptime(a_str, "%Y%m%d %H:%M:%S")
            b = datetime.datetime.strptime(b_str, "%Y%m%d %H:%M:%S")
            nowtime = datetime.datetime.now()
            passday = (nowtime - datetime.timedelta(days=4)).strftime("%Y%m%d")
            if a < nowtime < b:
                for i in self.tableWidgetWorkload.selectionModel().selection().indexes():
                    menu = QMenu()
                    item = menu.addAction('上个班未交班')
                    item2 = menu.addAction('上个班交班')

                    screenPos = self.tableWidgetWorkload.mapToGlobal(pos)
                    action = menu.exec(screenPos)
                    rowIndex = i.row()
                    taskids = self.tableWidgetWorkload.item(rowIndex, 3).text()
                    handstsNow=self.tableWidgetWorkload.item(rowIndex, 1).text()
                    name=self.tableWidgetWorkload.item(rowIndex, 2).text()
                    data={
                        "ids":taskids,
                        "type":handstsNow,
                        "name":name,
                        "date":passday,
                    }
                    if action == item:
                        if handstsNow == "Y":
                            self.handoverflight(data)
                            self.refreshnow()
                        else:
                            QMessageBox.warning(self, "警告", f"未查询到{name}交班记录，无需取消交班")
                        return
                    elif action == item2:
                        if handstsNow != "Y":
                            self.handoverflight(data)
                            self.refreshnow()
                        else:
                            QMessageBox.warning(self, "警告", f"经查询{name}已被记录交班，不用重复记录")
                        return
                    else:
                        return
        except Exception as e:
            app_logger.log_error(e)

    def get_load(self):
        try:
            peoples = []
            fileName, _ = QFileDialog.getOpenFileName(self, '打开文件', '/')
            ext = os.path.splitext(fileName)[1].lower()
            if ext == ".xlsx":
                workbook = openpyxl.load_workbook(filename=fileName)
                mysheet = workbook.active
                myrows = list(mysheet.values)
                if myrows[0][0] != ""and myrows[0][0] != None:
                    for i in myrows:
                        if i[0] != ""and i[0] != None:
                            peoples.append(i[0])
                if peoples:
                    self.peopleselect=peoples
                    self.write_list_to_json(peoples)
                    self.set_initial_selection()
                    self.refreshnow()
                else:
                    QMessageBox.information(self, "提示", "人员名单导入为空，请重新录入")
            else:
                QMessageBox.warning(self, "警告", "请使用xlsx文件，在第一列输入人名再导入，人名需要带字母")
                return
        except Exception as e:
            app_logger.log_error(e)
    def selectall_funciton(self):
        self.peopleselect=self.peopleres
        self.write_list_to_json(self.peopleselect)
        self.set_initial_selection()
    def search_table_empty(self):
        """根据搜索框内容过滤表格数据"""
        search_text = self.searchinfo.text().strip().lower()
        if not search_text:
            # 如果搜索框为空，显示所有行
            for row in range(self.tableWidgetWorkload.rowCount()):
                self.tableWidgetWorkload.setRowHidden(row, False)
            return

    def search_table(self):
        """根据搜索框内容过滤表格数据"""
        search_text = self.searchinfo.text().strip().lower()
        if not search_text:
            # 如果搜索框为空，显示所有行
            for row in range(self.tableWidgetWorkload.rowCount()):
                self.tableWidgetWorkload.setRowHidden(row, False)
            return
        # 遍历表格所有单元格进行搜索
        for row in range(self.tableWidgetWorkload.rowCount()):
            row_hidden = True
            for col in range(self.tableWidgetWorkload.columnCount()):
                item = self.tableWidgetWorkload.item(row, col)
                if item and search_text in item.text().lower():
                    row_hidden = False
                    break
            self.tableWidgetWorkload.setRowHidden(row, row_hidden)

    def on_sort_changed(self, logicalIndex, order):
        """记录当前的排序状态"""
        self.current_sort_column = logicalIndex
        self.current_sort_order = order

    def on_checkbox_changed(self, state):
        if state != 0:
            self.autoupdateMode = True
        else:
            self.autoupdateMode = False

    def ensure_file_exists(self):
        """确保文件存在，不存在则创建空文件"""
        directory = os.path.dirname(filename)
        # 如果目录不存在，则创建目录
        if directory and not os.path.exists(directory):
            os.makedirs(directory)
        # 如果文件不存在，则创建空文件
        if not os.path.exists(filename):
            with open(filename, 'w', encoding='utf-8') as f:
                f.write("[]")  #

    def write_list_to_json(self,data_list):
        """将 Python 列表写入 JSON 文件"""
        try:
            self.ensure_file_exists()  # 确保文件存在
            with open(filename, 'w', encoding='utf-8') as f:
                json.dump(data_list, f, ensure_ascii=False)
            #print(f"数据已成功写入 {filename}")
        except Exception as e:
            app_logger.log_error(e)
            #print(f"写入文件时出错: {e}")
    def read_json_to_list(self):
        """从 JSON 文件读取数据并转换为 Python 列表"""
        try:
            self.ensure_file_exists()  # 确保文件存在
            with open(filename, 'r', encoding='utf-8') as f:
                data = json.load(f)
            return data
        except json.JSONDecodeError:
            #print(f"文件 {filename} 格式不是有效的 JSON，将重置为空列表")
            # 重置文件为空白列表
            self.write_list_to_json([])
            return None
        except Exception as e:
            return None
    def set_initial_selection(self):
        """设置多选ComboBox的初始选中值"""
        for i in range(self.comboBox_peoplelist.model().rowCount()):
            item = self.comboBox_peoplelist.model().item(i)
            if item.text() in self.peopleselect:
                item.setCheckState(QtCore.Qt.CheckState.Checked)
            else:
                item.setCheckState(QtCore.Qt.CheckState.Unchecked)
        self.comboBox_peoplelist.update_text()
    def getdatas(self):
        try:
            # ip = "http://" + str(self.ipinfo) + "/static/getworkload"
            # data = json.dumps(self.peopleselect)
            # res = requests.post(url=ip, data=data, timeout=30).json()
            res = api_fc.request_post('/static/getworkload', names=self.peopleselect)["data"]
            return res
        except Exception as e:
            QMessageBox.warning(self, "提示", "服务器连接超时，请联系管理员检查服务器！")
            app_logger.log_error(e)
    def handoverflight(self,data0):
        try:
            # ip = "http://" + str(self.ipinfo) + "/static/handoverflight"
            # data = json.dumps(data0)
            # res = requests.post(url=ip, data=data, timeout=30).json()
            res = api_fc.request_post('/static/handoverflight', data=data0)
            if res["返回值"] == "ok":
                QMessageBox.information(self, "提示", "操作成功！")
            else:
                QMessageBox.warning(self, "提示", "操作失败！")
        except Exception as e:
            QMessageBox.warning(self, "提示", "服务器连接超时，请联系管理员检查服务器！")
            app_logger.log_error(e)
    def autoupdate(self):
        if self.autoupdateMode:
            res = self.getdatas()
            self.tableDisplay(res)
            if self.searchinfo.text():
                self.search_table()

    def tableDisplay(self,displaydata):
        try:
            if displaydata:
                self.tableWidgetWorkload.setSortingEnabled(False)
                # 清除原有数据
                self.tableWidgetWorkload.setRowCount(0)
                # 设置列数（假设数据是二维列表，第一行为表头）
                self.tableWidgetWorkload.setRowCount(len(displaydata))
                for row in range(len(displaydata)):
                    for col in range(len(self.headerLabel)):
                        item = QTableWidgetItem(str(displaydata[str(row)][str(col)]))
                        self.tableWidgetWorkload.setItem(row, col, item)
                        if col == 16 and "无" not in str(displaydata[str(row)][str(col)]):
                            item.setForeground(QColor("red"))
                            font = item.font()
                            item.setFont(font)
                        elif col == 8 :
                            item.setForeground(QColor("green"))
                            font = item.font()
                            font.setBold(True)
                            item.setFont(font)
                        elif col == 21 and str(displaydata[str(row)][str(col)]) != "" :
                            item.setBackground(QColor("yellow"))
                        elif col == 1 and str(displaydata[str(row)][str(col)]) == "Y":
                            item.setBackground(QColor("yellow"))
                        if col not in [5]:
                            self.tableWidgetWorkload.item(int(row), col).setTextAlignment(Qt.AlignmentFlag.AlignCenter)
                        else:
                            self.tableWidgetWorkload.item(int(row), col).setTextAlignment(Qt.AlignmentFlag.AlignTop | Qt.AlignmentFlag.AlignCenter)
                            self.tableWidgetWorkload.setRowHeight(int(row), 35)

                    settips = {1: "3", 14: "20"}
                    for key, value in settips.items():
                        if self.tableWidgetWorkload.item(int(row), key):
                            self.tableWidgetWorkload.item(int(row), key).setToolTip("%s" % (str(displaydata[str(row)][str(value)])))

                self.tableWidgetWorkload.setItemDelegate(ColoredTextDelegate())
                self.tableWidgetWorkload.resizeColumnsToContents()
                self.tableWidgetWorkload.setWordWrap(True)
                self.tableWidgetWorkload.setColumnWidth(6, 100)

                for i in range(len(self.headerLabel)):
                    if i in [3,5,7,9,11,13,15,17,19,22,23,24,25,26,27,10,12,18,20]:
                        self.tableWidgetWorkload.setColumnHidden(i, True)
                    else:
                        self.tableWidgetWorkload.setColumnHidden(i, False)

                self.tableWidgetWorkload.setSortingEnabled(True)
        except Exception as e:
            app_logger.log_error(e)
            self.tableWidgetWorkload.setSortingEnabled(True)

    def refreshnow(self):
        self.peopleselect = self.comboBox_peoplelist.get_selected_items()
        self.write_list_to_json(self.peopleselect)
        res=self.getdatas()
        self.tableDisplay(res)
        if self.searchinfo.text():
            self.search_table()

    def clearselect(self):
        self.peopleselect=[]
        self.write_list_to_json(self.peopleselect)
        self.set_initial_selection()

class ColoredTextDelegate(QStyledItemDelegate):

    def paint(self, painter, option, index):# 保存painter状态
        painter.save()

        # 设置交替行颜色
        palette = option.palette
        if index.row() % 2 == 1:
            bg_color = palette.color(QPalette.ColorRole.AlternateBase)
        else:
            bg_color = palette.color(QPalette.ColorRole.Base)

        # 绘制背景（考虑选中状态）
        if option.state & QStyle.StateFlag.State_Selected:
            painter.fillRect(option.rect, palette.highlight())
        else:
            painter.fillRect(option.rect, bg_color)

        if index.column() in [28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47]:
            # 获取单元格文本

            text = index.data(Qt.ItemDataRole.DisplayRole) or ""
            if "预排" in text:
                painter.fillRect(option.rect, QColor("yellow"))
                # 设置文本绘制选项
                painter.save()
            text_option = QTextOption()
            text_option.setAlignment(Qt.AlignmentFlag.AlignCenter)
            doc = QTextDocument()
            doc.setDefaultTextOption(text_option)
            doc.setTextWidth(option.rect.width())

            default_format = QTextCharFormat()
            default_format.setForeground(QColor("gray"))
            font = default_format.font()
            font.setPointSize(8)
            default_format.setFont(font)

            balck_format = QTextCharFormat()
            balck_format.setForeground(QColor("balck"))
            font = balck_format.font()
            font.setPointSize(10)
            balck_format.setFont(font)
            # 插入带格式文本
            cursor = QTextCursor(doc)
            cursor.insertText(text[:5], balck_format)
            cursor.insertText(text[5:], default_format)

            # 绘制文档
            painter.save()
            painter.translate(option.rect.topLeft())
            doc.drawContents(painter)
            painter.restore()
        elif index.column() == 8:
            text = index.data(Qt.ItemDataRole.DisplayRole) or ""
            start = text.find("(")
            end = text.find(")")
            # 设置文本绘制选项
            painter.save()
            text_option = QTextOption()
            text_option.setAlignment(Qt.AlignmentFlag.AlignCenter)

            doc = QTextDocument()
            doc.setDefaultTextOption(text_option)
            doc.setTextWidth(option.rect.width())

            default_format = QTextCharFormat()
            default_format.setForeground(QColor("red"))
            font = default_format.font()
            font.setBold(True)
            default_format.setFont(font)

            balck_format = QTextCharFormat()
            balck_format.setForeground(QColor("balck"))

            # 插入带格式文本
            cursor = QTextCursor(doc)
            cursor.insertText(text[:start+1], balck_format)
            cursor.insertText(text[start+1:end], default_format)
            cursor.insertText(text[end:], balck_format)
            # 计算垂直居中位置
            text_height = doc.size().height()
            y_offset = (option.rect.height() - text_height) / 2
            # 绘制文本
            painter.translate(option.rect.x(), option.rect.y() + y_offset)
            doc.drawContents(painter)
            painter.restore()

        else:
            # 其他单元格使用默认绘制
            super().paint(painter, option, index)


    def sizeHint(self, option, index):
        # 确保单元格有合适的大小
        if index.column() == 8:
            text = index.data(Qt.ItemDataRole.DisplayRole) or ""
            doc = QTextDocument()
            doc.setPlainText(text)
            doc.setTextWidth(option.rect.width())
            return doc.size().toSize()
        return super().sizeHint(option, index)