From de6c06906e24cde483849c2605695e696015892a Mon Sep 17 00:00:00 2001
From: Dummerle <44114474+Dummerle@users.noreply.github.com>
Date: Sun, 23 Jan 2022 01:45:16 +0100
Subject: [PATCH] Some fixes for installing_game_widget.py

---
 rare/components/tabs/games/__init__.py        | 37 +++++++++++++++----
 .../game_widgets/installing_game_widget.py    | 13 +++++--
 2 files changed, 39 insertions(+), 11 deletions(-)

diff --git a/rare/components/tabs/games/__init__.py b/rare/components/tabs/games/__init__.py
index de7417ac..b3a6fe44 100644
--- a/rare/components/tabs/games/__init__.py
+++ b/rare/components/tabs/games/__init__.py
@@ -118,7 +118,7 @@ class GamesTab(QStackedWidget, Ui_GamesTab):
         self.signals.installation_started.connect(self.installation_started)
         self.signals.update_gamelist.connect(self.update_list)
         self.signals.installation_finished.connect(
-            lambda x: self.installing_widget.setVisible(False)
+            self.installation_finished
         )
         self.signals.game_uninstalled.connect(lambda name: self.update_list([name]))
 
@@ -126,6 +126,11 @@ class GamesTab(QStackedWidget, Ui_GamesTab):
 
         self.game_list_scroll_area.horizontalScrollBar().setDisabled(True)
 
+    def installation_finished(self, app_name: str):
+        self.installing_widget.setVisible(False)
+        self.installing_widget.set_game("")
+        self.filter_games("")
+
     def installation_started(self, app_name: str):
         game = self.core.get_game(app_name, False)
         if game.is_dlc:
@@ -133,6 +138,12 @@ class GamesTab(QStackedWidget, Ui_GamesTab):
         self.installing_widget.set_game(app_name)
         self.installing_widget.setVisible(True)
 
+        i_widget, l_widget = self.widgets.get(app_name, (None, None))
+        if not i_widget or not l_widget:
+            return
+        i_widget.setVisible(False)
+        l_widget.setVisible(False)
+
     def verification_finished(self, igame: InstalledGame):
         # only if igame needs verification
         i_widget, l_widget = self.widgets[igame.app_name]
@@ -256,14 +267,17 @@ class GamesTab(QStackedWidget, Ui_GamesTab):
         if not filter_name and (t := self.active_filter):
             filter_name = t
 
-        for t in self.widgets.values():
-            app_name = t[0].game.app_name
-            # icon and list widget
-            if filter_name == "installed":
+        def get_visibility(widget):
+            app_name = widget.game.app_name
+
+            if not isinstance(widget,
+                              InstallingGameWidget) and self.installing_widget.game and self.installing_widget.game.app_name == app_name:
+                visible = False
+            elif filter_name == "installed":
                 visible = self.core.is_installed(app_name)
             elif filter_name == "offline":
-                if self.core.is_installed(app_name):
-                    visible = t[0].igame.can_run_offline
+                if self.core.is_installed(app_name) and not isinstance(widget, InstallingGameWidget):
+                    visible = widget.igame.can_run_offline
                 else:
                     visible = False
             elif filter_name == "32bit" and self.bit32:
@@ -285,12 +299,19 @@ class GamesTab(QStackedWidget, Ui_GamesTab):
 
             if (
                     search_text.lower() not in app_name.lower()
-                    and search_text.lower() not in t[0].game.app_title.lower()
+                    and search_text.lower() not in widget.game.app_title.lower()
             ):
                 visible = False
+            return visible
+
+        for t in self.widgets.values():
+            visible = get_visibility(t[0])
             for w in t:
                 w.setVisible(visible)
 
+        if self.installing_widget.game:
+            self.installing_widget.setVisible(get_visibility(self.installing_widget))
+
     def update_list(self, app_names: list = None):
         logger.debug("Updating list for " + str(app_names))
         if app_names:
diff --git a/rare/components/tabs/games/game_widgets/installing_game_widget.py b/rare/components/tabs/games/game_widgets/installing_game_widget.py
index 5f77a5c3..61a3e56b 100644
--- a/rare/components/tabs/games/game_widgets/installing_game_widget.py
+++ b/rare/components/tabs/games/game_widgets/installing_game_widget.py
@@ -2,6 +2,7 @@ from PyQt5.QtCore import Qt, QRect
 from PyQt5.QtGui import QPaintEvent, QPainter, QPixmap, QPen, QFont, QColor
 from PyQt5.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget
 
+from legendary.models.game import Game
 from rare import shared
 from rare.utils.utils import (
     get_pixmap,
@@ -12,6 +13,8 @@ from rare.utils.utils import (
 
 
 class InstallingGameWidget(QWidget):
+    game: Game = None
+
     def __init__(self):
         super(InstallingGameWidget, self).__init__()
         self.setObjectName("game_widget_icon")
@@ -36,10 +39,13 @@ class InstallingGameWidget(QWidget):
         self.layout().addLayout(minilayout)
 
     def set_game(self, app_name):
-        game = shared.core.get_game(app_name)
-        self.title_label.setText(f"<h4>{game.app_title}</h4>")
+        if not app_name:
+            self.game = None
+            return
+        self.game = shared.core.get_game(app_name)
+        self.title_label.setText(f"<h4>{self.game.app_title}</h4>")
 
-        self.image_widget.set_game(game.app_name)
+        self.image_widget.set_game(self.game.app_name)
 
     def set_status(self, s: int):
         self.image_widget.progress = s
@@ -81,6 +87,7 @@ class PaintWidget(QWidget):
     def paintEvent(self, a0: QPaintEvent) -> None:
         painter = QPainter()
         painter.begin(self)
+        painter.setRenderHint(QPainter.Antialiasing)
         painter.drawPixmap(self.rect(), self.bw_image)
 
         w = self.bw_image.width() * self.progress // 100