Add game image support to preview generator

joaorb64 · Nov 7, 2024 · 7fdcc14 · 7fdcc14
1 parent bf13fca
commit 7fdcc14
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 8 deletions.
diff --git a/layout/scoreboard_roa2/index.css b/layout/scoreboard_roa2/index.css
@@ -27,7 +27,6 @@ body {
     drop-shadow(0 -1px 2px rgba(0, 0, 0, 0.1));
   letter-spacing: 1;
   background-size: cover;
-  background-image: url("https://cdn.discordapp.com/attachments/1016678165764120666/1299114778681282660/Sample_RoA2_1.png?ex=6725e9ab&is=6724982b&hm=04b8e06bfc5faf1b7e2aa9c51f3cfaae59e7debaf16d4710181d93b93f221443&");
 }
 
 .container {

diff --git a/layout/scoreboard_roa2/index.png b/layout/scoreboard_roa2/index.png
diff --git a/layout/scoreboard_roa2/index_preview.png b/layout/scoreboard_roa2/index_preview.png
diff --git a/scripts/render_webpage.py b/scripts/render_webpage.py
@@ -17,6 +17,11 @@ def __init__(self, parent=None):
                      QtWebEngineCore.QWebEngineSettings.WebAttribute.LocalContentCanAccessFileUrls):
             settings.setAttribute(attr, True)
 
+        self.setAttribute(QtCore.Qt.WA_TranslucentBackground, True)
+        self.setStyleSheet("background: transparent;")
+
+        self.page().setBackgroundColor(QtCore.Qt.transparent)
+
         self.loadFinished.connect(self.save)
         self.setAttribute(QtCore.Qt.WA_DontShowOnScreen, True)
         self.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
@@ -49,9 +54,35 @@ def saveScreenshot(self):
         print(u"width: %d,hight: %d" % (size.width(), size.height()))
         img = QtGui.QImage(size.width(), size.height(),
                            QtGui.QImage.Format_ARGB32)
+        img.fill(QtGui.QColor(255, 192, 203))
         painter = QtGui.QPainter(img)
+
+        # Check for a background image
+        bg_image = None
+
+        game_screenshot = self.current[1].rsplit("_preview")[0]+".png"
+        if os.path.isfile(game_screenshot):
+            bg_image = game_screenshot
+            print("Found background image")
+        else:
+            default_game_screenshot = os.path.join(
+                os.path.dirname(self.current[1]), "default.png")
+
+            if os.path.isfile(default_game_screenshot):
+                bg_image = default_game_screenshot
+                print("Found default background image")
+
+        if bg_image:
+            bg = QtGui.QImage(bg_image)
+            painter.drawImage(
+                QtCore.QRect(0, 0, 1920, 1080),
+                bg
+            )
+
+        # Render html
         self.render(painter, QtCore.QPoint(0, 0),
                     QtCore.QRect(0, 0, 1920, 1080))
+
         painter.end()
         filename = self.current[1]
         if img.save(filename):
@@ -73,16 +104,13 @@ def saveScreenshot(self):
     in_out = []
 
     for path, subdirs, files in os.walk(os.path.abspath("../layout/")):
-        if not os.path.isdir("renders"):
-            os.mkdir("renders")
         for name in files:
             if name.endswith(".html"):
                 print(os.path.join(path, name))
-                in_out.append([
-                    f"file:///{os.path.join(path, name).replace(os.path.sep, '/')}",
-                    path.replace(os.path.sep, '/')+"/" +
-                    name.split(".")[0]+"_preview.png"
-                ])
+                in_out.append([f"file:///{os.path.join(path, name).replace(os.path.sep, '/')}",
+                               path.replace(os.path.sep, '/')+"/" +
+                               name.split(".")[0]+"_preview.png"
+                               ])
     print(in_out)
 
     shotter = PageShotter()