Take a screenshot in case of exception in CodeExecutor

src/tribler-gui/tribler_gui/code_executor.py

@@ -7,7 +7,7 @@
 
 from PyQt5.QtNetwork import QTcpServer
 
-from tribler_gui.utilities import connect
+from tribler_gui.utilities import connect, take_screenshot
 
 
 class CodeExecutor:
@@ -58,6 +58,7 @@ def run_code(self, code, task_id):
             pass
 
         if self.shell.last_traceback:
+            self.take_screenshot()
             self.on_crash(self.shell.last_traceback)
             return
 
@@ -67,6 +68,13 @@ def run_code(self, code, task_id):
         for socket in self.sockets:
             socket.write(b"result %s %s\n" % (return_value, task_id))
 
+    def take_screenshot(self):
+        window = self.shell.locals.get('window')
+        if not window:
+            self.logger.warning("Cannot take screenshot, window is not found in locals")
+        else:
+            take_screenshot(window)
+
     def on_crash(self, exception_text):
         self.logger.error(f"Crash in CodeExecutor:\n{exception_text}")
 

src/tribler-gui/tribler_gui/utilities.py

@@ -3,6 +3,7 @@
 import math
 import os
 import sys
+import time
 import traceback
 import types
 from datetime import datetime, timedelta
@@ -11,7 +12,8 @@
 from urllib.parse import quote_plus
 from uuid import uuid4
 
-from PyQt5.QtCore import QCoreApplication, QLocale, QTranslator, pyqtSignal
+from PyQt5.QtCore import QCoreApplication, QLocale, QPoint, QTranslator, pyqtSignal
+from PyQt5.QtGui import QPixmap, QRegion
 from PyQt5.QtWidgets import QApplication
 
 import tribler_gui
@@ -434,3 +436,13 @@ def get_translator(language=None):
     filename = ""
     translator.load(locale, filename, directory=TRANSLATIONS_DIR)
     return translator
+
+
+def take_screenshot(window):
+    timestamp = int(time.time())
+    pixmap = QPixmap(window.rect().size())
+    window.render(pixmap, QPoint(), QRegion(window.rect()))
+    screenshots_dir = Path.cwd() / "screenshots"
+    screenshots_dir.mkdir(exist_ok=True)
+    img_name = 'screenshot_%d.jpg' % timestamp
+    pixmap.save(str(screenshots_dir / img_name))