2022-12-24 12:09:19 +13:00
|
|
|
from typing import Optional
|
|
|
|
|
|
|
|
from PyQt5.QtCore import pyqtSignal, pyqtSlot
|
2021-09-11 02:36:26 +12:00
|
|
|
from PyQt5.QtWidgets import QFrame, QWidget, QMessageBox
|
2022-06-19 06:45:36 +12:00
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
from rare.models.game import RareGame
|
2022-12-24 12:09:19 +13:00
|
|
|
from rare.models.install import InstallOptionsModel
|
2022-12-25 15:21:23 +13:00
|
|
|
from rare.shared import LegendaryCoreSingleton, GlobalSignalsSingleton
|
2022-12-27 02:11:23 +13:00
|
|
|
from rare.shared.game_utils import GameUtils
|
2022-09-05 05:38:24 +12:00
|
|
|
from rare.shared.image_manager import ImageSize
|
2021-09-11 02:36:26 +12:00
|
|
|
from rare.ui.components.tabs.games.game_info.game_dlc import Ui_GameDlc
|
|
|
|
from rare.ui.components.tabs.games.game_info.game_dlc_widget import Ui_GameDlcWidget
|
2022-06-20 05:42:49 +12:00
|
|
|
from rare.widgets.image_widget import ImageWidget
|
2021-09-11 02:36:26 +12:00
|
|
|
|
|
|
|
|
2022-12-24 12:09:19 +13:00
|
|
|
class GameDlc(QWidget):
|
2021-09-11 02:36:26 +12:00
|
|
|
install_dlc = pyqtSignal(str, bool)
|
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
def __init__(self, game_utils: GameUtils, parent=None):
|
2021-09-11 02:36:26 +12:00
|
|
|
super(GameDlc, self).__init__(parent=parent)
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui = Ui_GameDlc()
|
|
|
|
self.ui.setupUi(self)
|
2022-02-26 06:43:27 +13:00
|
|
|
self.core = LegendaryCoreSingleton()
|
|
|
|
self.signals = GlobalSignalsSingleton()
|
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
self.rgame: Optional[RareGame] = None
|
2021-11-02 10:53:04 +13:00
|
|
|
self.game_utils = game_utils
|
2021-10-08 06:42:27 +13:00
|
|
|
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.available_dlc_scroll.setFrameStyle(QFrame.NoFrame)
|
|
|
|
self.ui.installed_dlc_scroll.setFrameStyle(QFrame.NoFrame)
|
2021-10-08 06:42:27 +13:00
|
|
|
|
2021-09-11 02:36:26 +12:00
|
|
|
self.installed_dlc_widgets = list()
|
|
|
|
self.available_dlc_widgets = list()
|
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
def update_dlcs(self, rgame: RareGame):
|
|
|
|
self.rgame = rgame
|
|
|
|
self.title.setTitle(self.rgame.app_title)
|
2021-09-11 02:36:26 +12:00
|
|
|
|
|
|
|
if self.installed_dlc_widgets:
|
|
|
|
for dlc_widget in self.installed_dlc_widgets:
|
2022-06-24 02:54:26 +12:00
|
|
|
dlc_widget.uninstall.disconnect()
|
2021-09-11 02:36:26 +12:00
|
|
|
dlc_widget.deleteLater()
|
|
|
|
self.installed_dlc_widgets.clear()
|
|
|
|
if self.available_dlc_widgets:
|
|
|
|
for dlc_widget in self.available_dlc_widgets:
|
|
|
|
dlc_widget.install.disconnect()
|
|
|
|
dlc_widget.deleteLater()
|
|
|
|
self.available_dlc_widgets.clear()
|
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
for dlc in sorted(self.rgame.owned_dlcs, key=lambda x: x.app_title):
|
|
|
|
if dlc.is_installed:
|
2021-09-11 02:36:26 +12:00
|
|
|
dlc_widget = GameDlcWidget(dlc, True)
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.installed_dlc_contents_layout.addWidget(dlc_widget)
|
2021-10-11 08:52:11 +13:00
|
|
|
dlc_widget.uninstall.connect(self.uninstall)
|
|
|
|
self.installed_dlc_widgets.append(dlc_widget)
|
|
|
|
|
2021-09-11 02:36:26 +12:00
|
|
|
else:
|
|
|
|
dlc_widget = GameDlcWidget(dlc, False)
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.available_dlc_contents_layout.addWidget(dlc_widget)
|
2021-09-11 02:36:26 +12:00
|
|
|
dlc_widget.install.connect(self.install)
|
|
|
|
self.available_dlc_widgets.append(dlc_widget)
|
|
|
|
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.installed_dlc_label.setVisible(not self.installed_dlc_widgets)
|
|
|
|
self.ui.installed_dlc_scroll.setVisible(bool(self.installed_dlc_widgets))
|
2021-09-11 02:36:26 +12:00
|
|
|
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.available_dlc_label.setVisible(not self.available_dlc_widgets)
|
|
|
|
self.ui.available_dlc_scroll.setVisible(bool(self.available_dlc_widgets))
|
2021-09-11 02:36:26 +12:00
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
@pyqtSlot(RareGame)
|
|
|
|
def uninstall(self, rgame: RareGame):
|
|
|
|
if self.game_utils.uninstall_game(rgame.app_name):
|
|
|
|
self.update_dlcs(self.rgame)
|
2021-10-11 08:52:11 +13:00
|
|
|
|
2021-09-11 02:36:26 +12:00
|
|
|
def install(self, app_name):
|
2022-12-25 15:21:23 +13:00
|
|
|
if not self.core.is_installed(self.rgame.app_name):
|
2021-12-24 22:09:50 +13:00
|
|
|
QMessageBox.warning(
|
|
|
|
self,
|
|
|
|
"Error",
|
2022-12-25 15:21:23 +13:00
|
|
|
self.tr("Base Game is not installed. Please install {} first").format(self.rgame.app_title),
|
2021-12-24 22:09:50 +13:00
|
|
|
)
|
2021-09-11 02:36:26 +12:00
|
|
|
return
|
2021-10-08 06:42:27 +13:00
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
self.signals.game.install.emit(InstallOptionsModel(app_name=app_name, update=True))
|
2021-09-11 02:36:26 +12:00
|
|
|
|
|
|
|
|
2022-12-24 12:09:19 +13:00
|
|
|
class GameDlcWidget(QFrame):
|
2022-12-25 15:21:23 +13:00
|
|
|
install = pyqtSignal(RareGame)
|
|
|
|
uninstall = pyqtSignal(RareGame)
|
2021-09-11 02:36:26 +12:00
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
def __init__(self, dlc: RareGame, installed: bool, parent=None):
|
2021-09-11 02:36:26 +12:00
|
|
|
super(GameDlcWidget, self).__init__(parent=parent)
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui = Ui_GameDlcWidget()
|
|
|
|
self.ui.setupUi(self)
|
2021-09-11 02:36:26 +12:00
|
|
|
self.dlc = dlc
|
2021-09-12 23:23:40 +12:00
|
|
|
|
2022-06-20 05:42:49 +12:00
|
|
|
self.image = ImageWidget(self)
|
|
|
|
self.image.setFixedSize(ImageSize.Smaller)
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.dlc_layout.insertWidget(0, self.image)
|
2022-06-19 06:45:36 +12:00
|
|
|
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.dlc_name.setText(dlc.app_title)
|
2022-12-25 15:21:23 +13:00
|
|
|
self.ui.version.setText(dlc.version)
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.app_name.setText(dlc.app_name)
|
2021-09-11 02:36:26 +12:00
|
|
|
|
2022-12-25 15:21:23 +13:00
|
|
|
self.image.setPixmap(dlc.pixmap)
|
2021-12-05 12:31:36 +13:00
|
|
|
|
2021-10-11 08:52:11 +13:00
|
|
|
if installed:
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.action_button.setProperty("uninstall", 1)
|
|
|
|
self.ui.action_button.clicked.connect(self.uninstall_dlc)
|
|
|
|
self.ui.action_button.setText(self.tr("Uninstall DLC"))
|
2021-09-11 02:36:26 +12:00
|
|
|
else:
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.action_button.setProperty("install", 1)
|
|
|
|
self.ui.action_button.clicked.connect(self.install_game)
|
|
|
|
self.ui.action_button.setText(self.tr("Install DLC"))
|
2021-10-11 08:52:11 +13:00
|
|
|
|
|
|
|
def uninstall_dlc(self):
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.action_button.setDisabled(True)
|
|
|
|
self.ui.action_button.setText(self.tr("Uninstalling"))
|
2022-12-25 15:21:23 +13:00
|
|
|
self.uninstall.emit(self.dlc)
|
2021-09-11 02:36:26 +12:00
|
|
|
|
|
|
|
def install_game(self):
|
2022-12-24 12:09:19 +13:00
|
|
|
self.ui.action_button.setDisabled(True)
|
|
|
|
self.ui.action_button.setText(self.tr("Installing"))
|
2022-12-25 15:21:23 +13:00
|
|
|
self.install.emit(self.dlc)
|