You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
LC_ModManager/window/main_window.py

186 lines
8.2 KiB
Python

import logging
import os.path
import sys
from typing import Dict, List
from PyQt6 import QtWidgets
from PyQt6.QtCore import Qt
from PyQt6.QtGui import QStandardItemModel, QStandardItem
from window.mod_manager_window_export import Ui_MainWindow
class MainWindow(QtWidgets.QMainWindow):
def __init__(self, parent, parent_logger: logging.Logger, version: str = ""):
super(MainWindow, self).__init__()
self.parent = parent
self.__logger = logging.getLogger("MainWindow")
for handler in parent_logger.handlers:
self.__logger.addHandler(handler)
self.__logger.setLevel(parent_logger.level)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.setWindowTitle(f"LC Mod Manager {version}")
self.ui.actionAdd_new_mod.triggered.connect(self.on_action_add_new_mod)
self.ui.actionrefresh_mods.triggered.connect(self.on_action_refresh_mods)
self.ui.actionCheck_for_Updates.triggered.connect(self.on_action_check_for_updates)
self.ui.actionSet_game_folder.triggered.connect(self.on_action_set_game_folder)
self.ui.actionRemove_ALL_manager_files.triggered.connect(self.on_action_remove_all_files)
self.ui.DeleteModFilesButton.pressed.connect(self.on_pressed_delete_mod_files_button)
self.ui.ApplyChangesButton.pressed.connect(self.on_pressed_apply_changes_button)
self.ui.DisacrdChangesButton.pressed.connect(self.on_pressed_discard_changes_button)
self.ui.TODOButton.hide()
self.ui.ApplyChangesButton.hide()
self.ui.DisacrdChangesButton.hide()
self.ui.DeleteModFilesButton.hide()
self.ui.actionCheck_for_Updates.setDisabled(True)
def set_available_mods(self, available_mods: Dict[str, str]):
"""
Sets the given mods as the list of available mods
:param available_mods:
:return:
"""
item_model = QStandardItemModel(self.ui.AvailableModsList)
for mod_name in available_mods.keys():
item = QStandardItem(mod_name + f" ({available_mods[mod_name]['version']})")
item.setFlags(Qt.ItemFlag.ItemIsUserCheckable | Qt.ItemFlag.ItemIsEnabled)
item.setData(Qt.CheckState.Unchecked, Qt.ItemDataRole.CheckStateRole)
item.setData(mod_name, Qt.ItemDataRole.UserRole)
item_model.appendRow(item)
if self.parent.is_mod_installed(mod_name):
item.setCheckState(Qt.CheckState.Checked)
item_model.itemChanged.connect(self.on_available_mod_item_changed)
self.ui.AvailableModsList.setModel(item_model)
def set_installed_mods(self, installed_mods: Dict[str, str]):
"""
Sets the given mods as the list of installed mods
:param installed_mods: string list of all the mods as to be written to the list
:return:
"""
item_model = QStandardItemModel(self.ui.InstalledModsListView)
for mod_name in installed_mods.keys():
if mod_name in self.parent.available_mods.keys():
mod_version = self.parent.available_mods[mod_name]['version']
else:
mod_version = "Not Tracked"
item = QStandardItem(mod_name + f" ({mod_version})")
item.setFlags(Qt.ItemFlag.ItemIsUserCheckable | Qt.ItemFlag.ItemIsEnabled)
item.setData(Qt.CheckState.Unchecked, Qt.ItemDataRole.CheckStateRole)
item.setData(mod_name, Qt.ItemDataRole.UserRole)
item_model.appendRow(item)
item.setCheckState(Qt.CheckState.Checked)
item_model.itemChanged.connect(self.on_installed_mod_item_changed)
self.ui.InstalledModsListView.setModel(item_model)
# UI Callback functions
## Actions
def on_action_add_new_mod(self):
self.__logger.debug("Action: \"add new mod\" triggered!")
dialog = QtWidgets.QFileDialog(self, "Select Lethal Company mod")
dialog.setFileMode(QtWidgets.QFileDialog.FileMode.ExistingFiles)
result = dialog.getOpenFileName(filter='ZIP (*.zip)')
self.__logger.debug(f"user selected \"{result[0]}\"")
if not os.path.isfile(result[0]):
dialog = QtWidgets.QMessageBox()
dialog.setWindowTitle("Not a file")
dialog.setInformativeText(f"The given file \"{result}\" did not look like a file!")
dialog.setIcon(QtWidgets.QMessageBox.Icon.Warning)
dialog.exec()
if not self.parent.is_valid_mod_file(result[0]):
dialog = QtWidgets.QMessageBox()
dialog.setWindowTitle("Not a valid mod file")
dialog.setInformativeText(
f"The given file \"{result}\" did not look like a mod file. Is the manifest.json present?")
dialog.setIcon(QtWidgets.QMessageBox.Icon.Warning)
dialog.exec()
self.parent.add_mod_file(result[0])
def on_action_refresh_mods(self):
self.__logger.debug("Action: \"refresh mods\" triggered!")
self.parent.index_stored_mods()
self.parent.index_installed_mods()
def on_action_check_for_updates(self):
self.__logger.debug("Action: \"check for updates\" triggered!")
def on_action_set_game_folder(self):
self.__logger.debug("Action: \"set game folder\" triggered!")
dialog = QtWidgets.QFileDialog(self, "Select Lethal Company folder")
dialog.setFileMode(QtWidgets.QFileDialog.FileMode.Directory)
dialog.setOptions(QtWidgets.QFileDialog.Option.ShowDirsOnly)
result = dialog.getExistingDirectory()
dir_accepted = self.parent.set_game_folder(result)
attempt = 0
while not dir_accepted:
attempt += 1
dialog = QtWidgets.QMessageBox()
dialog.setWindowTitle("Invalid game path")
dialog.setInformativeText(f"The given path \"{result}\" did not look like the Lethal Company game folder!\n"
f"If you can't find it. Try using steam and select \"browse local files\"!")
dialog.setIcon(QtWidgets.QMessageBox.Icon.Warning)
dialog.exec()
dialog = QtWidgets.QFileDialog(self, "Select Lethal Company folder")
dialog.setFileMode(QtWidgets.QFileDialog.FileMode.Directory)
dialog.setOptions(QtWidgets.QFileDialog.Option.ShowDirsOnly)
result = dialog.getExistingDirectory()
dir_accepted = self.parent.set_game_folder(result)
if attempt > 2:
return
def on_action_remove_all_files(self):
self.__logger.debug("Action: \"remove all manager files\" triggered!")
self.parent.nuke_manager_files()
## Buttons
def on_pressed_delete_mod_files_button(self):
self.__logger.debug("Pressed button: \"Delete Mod Files\"")
raise NotImplementedError
def on_pressed_apply_changes_button(self):
self.__logger.debug("Pressed button: \"Apply Changes\"")
raise NotImplementedError
def on_pressed_discard_changes_button(self):
self.__logger.debug("Pressed button: \"Discard Changes\"")
raise NotImplementedError
def on_available_mod_item_changed(self, item: QStandardItem):
self.__logger.debug(f"Available Mod list item \"{item.text()}\" changed to {item.checkState()}")
mod_name = item.data(Qt.ItemDataRole.UserRole)
if item.checkState() == Qt.CheckState.Checked:
self.parent.install_mod(mod_name)
elif item.checkState() == Qt.CheckState.Unchecked:
self.parent.uninstall_mod(mod_name)
def on_installed_mod_item_changed(self, item: QStandardItem):
self.__logger.debug(f"Installed Mod list item \"{item.text()}\" changed to {item.checkState()}")
mod_name = item.data(Qt.ItemDataRole.UserRole)
if item.checkState() == Qt.CheckState.Unchecked: # mod should be uninstalled
accepted = QtWidgets.QMessageBox.question(self, "Really uninstall mod?",
f"Do you really want to uninstall the mod \"{mod_name}\"?\n"
"This could lead to permanent data loss if it wasn't tracked!")
if accepted:
self.parent.uninstall_mod(mod_name)
else:
return