extra_menu.py

import sys
from shutil import copy
from typing import List

from loguru import logger
from PyQt5 import uic
from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import QApplication, QScroller
from qfluentwidgets import (
    ComboBox,
    FluentWindow,
    Flyout,
    FlyoutAnimationType,
    HyperlinkButton,
    InfoBarIcon,
    LineEdit,
    ListWidget,
    PrimaryPushButton,
    SmoothScrollArea,
    ToolButton,
)
from qfluentwidgets import FluentIcon as fIcon

import list_
from conf import CW_HOME, get_week_type
from file import config_center, load_from_json, save_data_to_json, schedule_center
from menu import SettingsMenu
from utils import TimeManagerFactory

settings = None

current_week = TimeManagerFactory.get_instance().get_current_weekday()
temp_schedule = {'schedule': {}, 'schedule_even': {}}


def open_settings(main_window=None) -> None:
    """打开设置窗口"""
    global settings
    try:
        if settings is None or not hasattr(settings, 'isVisible') or not settings.isVisible():
            settings = SettingsMenu(main_window=main_window)
            settings.closed.connect(cleanup_settings)
            settings.show()
            logger.info('打开“设置”')
        else:
            settings.raise_()
            settings.activateWindow()
    except (NameError, AttributeError):
        settings = SettingsMenu(main_window=main_window)
        settings.closed.connect(cleanup_settings)
        settings.show()
        logger.info('打开“设置” (重新初始化)')


def cleanup_settings() -> None:
    global settings
    logger.info('关闭“设置”')
    if 'settings' in globals() and settings is not None:
        settings.close()
        settings = None


class ExtraMenu(FluentWindow):
    def __init__(self) -> None:
        super().__init__()
        self.menu = None
        self.main_window = None
        self.interface = uic.loadUi(f'{CW_HOME}/view/extra_menu.ui')
        self.initUI()
        self.init_interface()

    def init_interface(self) -> None:
        ex_scroll = self.findChild(SmoothScrollArea, 'ex_scroll')
        QScroller.grabGesture(ex_scroll, QScroller.LeftMouseButtonGesture)
        select_temp_week = self.findChild(ComboBox, 'select_temp_week')  # 选择替换日期
        select_temp_week.addItems(list_.week)
        select_temp_week.setCurrentIndex(current_week)
        select_temp_week.currentIndexChanged.connect(self.refresh_schedule_list)  # 日期选择变化

        select_temp_schedule = self.findChild(ComboBox, 'select_temp_schedule')  # 选择替换课表
        select_temp_schedule.addItems(list_.week_type)
        select_temp_schedule.setCurrentIndex(get_week_type())
        select_temp_schedule.currentIndexChanged.connect(self.refresh_schedule_list)  # 日期选择变化

        tmp_schedule_list = self.findChild(ListWidget, 'schedule_list')  # 换课列表
        tmp_schedule_list.addItems(self.load_schedule())
        tmp_schedule_list.itemChanged.connect(self.upload_item)

        class_kind_combo = self.findChild(ComboBox, 'class_combo')  # 课程类型
        class_kind_combo.addItems(list_.class_kind)

        set_button = self.findChild(ToolButton, 'set_button')
        set_button.setIcon(fIcon.EDIT)
        set_button.clicked.connect(self.edit_item)

        save_temp_conf = self.findChild(PrimaryPushButton, 'save_temp_conf')  # 保存设置
        save_temp_conf.clicked.connect(self.save_temp_conf)

        redirect_to_settings = self.findChild(HyperlinkButton, 'redirect_to_settings')
        redirect_to_settings.clicked.connect(lambda: open_settings(self.main_window))

    @staticmethod
    def load_schedule() -> List[str]:
        if get_week_type():
            return schedule_center.schedule_data['schedule_even'][str(current_week)]
        return schedule_center.schedule_data['schedule'][str(current_week)]

    def save_temp_conf(self) -> None:
        try:
            temp_week = self.findChild(ComboBox, 'select_temp_week')
            temp_schedule_set = self.findChild(ComboBox, 'select_temp_schedule')
            if config_center.read_conf('Temp', 'temp_schedule') == '':
                copy(
                    f'{CW_HOME}/config/schedule/{config_center.schedule_name}',
                    f'{CW_HOME}/config/schedule/backup.json',
                )
                logger.success(f'原课表配置已备份：{config_center.schedule_name} --> backup.json')
                config_center.write_conf('Temp', 'temp_schedule', config_center.schedule_name)
            current_full_schedule_data = schedule_center.schedule_data
            adjusted_week = str(temp_week.currentIndex())
            is_even_week = temp_schedule_set.currentIndex() == 1

            if is_even_week:
                if adjusted_week in temp_schedule.get('schedule_even', {}):
                    current_full_schedule_data['schedule_even'] = current_full_schedule_data.get(
                        'schedule_even', {}
                    )
                    current_full_schedule_data['schedule_even'][adjusted_week] = temp_schedule[
                        'schedule_even'
                    ][adjusted_week]
                    current_full_schedule_data['adjusted_classes'] = current_full_schedule_data.get(
                        'adjusted_classes', {}
                    )
                    current_full_schedule_data['adjusted_classes'][f'even_{adjusted_week}'] = True
            elif adjusted_week in temp_schedule.get('schedule', {}):
                current_full_schedule_data['schedule'] = current_full_schedule_data.get(
                    'schedule', {}
                )
                current_full_schedule_data['schedule'][adjusted_week] = temp_schedule['schedule'][
                    adjusted_week
                ]
                current_full_schedule_data['adjusted_classes'] = current_full_schedule_data.get(
                    'adjusted_classes', {}
                )
                current_full_schedule_data['adjusted_classes'][f'odd_{adjusted_week}'] = True

            save_data_to_json(current_full_schedule_data, config_center.schedule_name)
            schedule_center.update_schedule()
            config_center.write_conf('Temp', 'set_week', str(temp_week.currentIndex()))
            config_center.write_conf('Temp', 'set_schedule', str(temp_schedule_set.currentIndex()))

            Flyout.create(
                icon=InfoBarIcon.SUCCESS,
                title='保存成功',
                content="已保存至 ./config.ini \n重启后恢复。",
                target=self.findChild(PrimaryPushButton, 'save_temp_conf'),
                parent=self,
                isClosable=True,
                aniType=FlyoutAnimationType.PULL_UP,
            )
        except Exception as e:
            logger.error(f'保存临时课表时发生错误: {e}')
            Flyout.create(
                icon=InfoBarIcon.ERROR,
                title='保存失败',
                content=f"错误信息：{e}",
                target=self.findChild(PrimaryPushButton, 'save_temp_conf'),
                parent=self,
                isClosable=True,
                aniType=FlyoutAnimationType.PULL_UP,
            )

    def refresh_schedule_list(self) -> None:
        global current_week
        current_week = self.findChild(ComboBox, 'select_temp_week').currentIndex()
        current_schedule = self.findChild(ComboBox, 'select_temp_schedule').currentIndex()
        logger.debug(f'current_week: {current_week}, current_schedule: {current_schedule}')
        tmp_schedule_list = self.findChild(ListWidget, 'schedule_list')  # 换课列表
        tmp_schedule_list.clear()
        tmp_schedule_list.clearSelection()
        if config_center.read_conf('Temp', 'temp_schedule') == '':
            if current_schedule:
                tmp_schedule_list.addItems(
                    schedule_center.schedule_data['schedule_even'][str(current_week)]
                )
            else:
                tmp_schedule_list.addItems(
                    schedule_center.schedule_data['schedule'][str(current_week)]
                )
        elif current_schedule:
            tmp_schedule_list.addItems(
                load_from_json('backup.json')['schedule_even'][str(current_week)]
            )
        else:
            tmp_schedule_list.addItems(load_from_json('backup.json')['schedule'][str(current_week)])

    def upload_item(self) -> None:
        global temp_schedule
        se_schedule_list = self.findChild(ListWidget, 'schedule_list')
        cache_list = []
        for i in range(se_schedule_list.count()):  # 缓存ListWidget数据至列表
            item_text = se_schedule_list.item(i).text()
            cache_list.append(item_text)
        if get_week_type():
            temp_schedule['schedule_even'][str(current_week)] = cache_list
        else:
            temp_schedule['schedule'][str(current_week)] = cache_list

    def edit_item(self) -> None:
        tmp_schedule_list = self.findChild(ListWidget, 'schedule_list')
        class_combo = self.findChild(ComboBox, 'class_combo')
        custom_class = self.findChild(LineEdit, 'custom_class')
        selected_items = tmp_schedule_list.selectedItems()

        if selected_items:
            selected_item = selected_items[0]
            if class_combo.currentIndex() != 0:
                selected_item.setText(class_combo.currentText())
            elif custom_class.text() != '':
                selected_item.setText(custom_class.text())

    def initUI(self) -> None:
        # 修复设置窗口在各个屏幕分辨率DPI下的窗口大小
        screen_geometry = QApplication.primaryScreen().geometry()
        screen_width = screen_geometry.width()
        screen_height = screen_geometry.height()

        width = int(screen_width * 0.55)
        height = int(screen_height * 0.65)

        self.move(int(screen_width / 2 - width / 2), 150)
        self.resize(width, height)

        self.setWindowTitle('Class Widgets - 更多功能')
        self.setWindowIcon(QIcon(str(CW_HOME / 'img/logo/favicon-exmenu.ico')))

        self.addSubInterface(self.interface, fIcon.INFO, '更多设置')

    def closeEvent(self, e) -> None:
        self.deleteLater()
        return super().closeEvent(e)


if __name__ == '__main__':
    from i18n_manager import app

    ex = ExtraMenu()
    ex.show()
    sys.exit(app.exec())