screenshots panel improved

bauh/view/qt/screenshots.py

@@ -1,21 +1,32 @@
+import logging
+from threading import Thread
 from typing import List
 
+from PyQt5.QtCore import Qt
 from PyQt5.QtGui import QIcon, QPixmap
 from PyQt5.QtWidgets import QDialog, QLabel, QGridLayout, QPushButton
 
 from bauh.api.abstract.cache import MemoryCache
+from bauh.api.http import HttpClient
 from bauh.view.qt import qt_utils
 from bauh.view.qt.view_model import PackageView
 
 
 class ScreenshotsDialog(QDialog):
 
-    def __init__(self, pkg: PackageView, icon_cache: MemoryCache, i18n: dict, screenshots: List[QPixmap]):
+    MAX_HEIGHT = 600
+    MAX_WIDTH = 800
+
+    def __init__(self, pkg: PackageView, http_client: HttpClient, icon_cache: MemoryCache, i18n: dict, screenshots: List[QPixmap], logger: logging.Logger):
         super(ScreenshotsDialog, self).__init__()
         self.setWindowTitle(str(pkg))
         self.screenshots = screenshots
+        self.logger = logger
+        self.loaded_imgs = []
+        self.download_threads = []
         self.resize(1280, 720)
         self.i18n = i18n
+        self.http_client = http_client
 
         icon_data = icon_cache.get(pkg.model.icon_url)
 
@@ -36,30 +47,73 @@ class ScreenshotsDialog(QDialog):
 
         self.img = QLabel()
         self.layout().addWidget(self.img, 1, 1)
+        self.img_label = QLabel()
+        self.img_label.setStyleSheet('QLabel { font-weight: bold; text-align: center }')
+        self.layout().addWidget(self.img_label, 2, 1)
+
+        self.img_idx = 0
+
+        for idx, s in enumerate(self.screenshots):
+            t = Thread(target=self._download_img, args=(idx, s))
+            t.start()
 
-        self.idx = 0
         self._load_img()
 
     def _load_img(self):
-        pixmap = self.screenshots[self.idx]
-        self.img.setPixmap(pixmap)
+        if len(self.loaded_imgs) > self.img_idx:
+            img = self.loaded_imgs[self.img_idx]
+
+            if isinstance(img, QPixmap):
+                self.img_label.setText('')
+                self.img.setPixmap(img)
+            else:
+                self.img_label.setText(img)
+                self.img.setPixmap(QPixmap())
+        else:
+            self.img.setPixmap(QPixmap())
+            self.img_label.setText('...{}...'.format(self.i18n['screenshots,download.running']))
 
         if len(self.screenshots) == 1:
             self.bt_back.setVisible(False)
             self.bt_next.setVisible(False)
         else:
-            self.bt_back.setVisible(self.idx != 0)
-            self.bt_next.setVisible(self.idx != len(self.screenshots) - 1)
+            self.bt_back.setEnabled(self.img_idx != 0)
+            self.bt_next.setEnabled(self.img_idx != len(self.screenshots) - 1)
 
         self.adjustSize()
         qt_utils.centralize(self)
 
+    def _download_img(self, idx: int, url: str):
+        self.logger.info('Downloading image [{}] from {}'.format(idx, url))
+        res = self.http_client.get(url)
+
+        if res:
+            if not res.content:
+                self.logger.warning("Image [{}] from {} has no content".format(idx, url))
+                self.screenshots.append(self.i18n['screenshots.download.no_content'])
+                return
+            else:
+                self.logger.info('Image [{}] successfully downloaded'.format(idx))
+                pixmap = QPixmap()
+                pixmap.loadFromData(res.content)
+
+                if pixmap.size().height() > self.MAX_HEIGHT or pixmap.size().width() > self.MAX_WIDTH:
+                    pixmap = pixmap.scaled(self.MAX_WIDTH, self.MAX_HEIGHT, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+
+                self.loaded_imgs.append(pixmap)
+
+                if self.img_idx == idx:
+                    self._load_img()
+        else:
+            self.logger.info("Could not retrieve image [{}] from {}".format(idx, url))
+            self.screenshots.append(self.i18n['screenshots.download.no_response'])
+
     def back(self):
-        self.idx -= 1
+        self.img_idx -= 1
         self._load_img()
 
     def next(self):
-        self.idx += 1
+        self.img_idx += 1
         self._load_img()