cut off the project root from the beginning of the filename in breakpoints/stacktraces viewers

Author: robertbasic

pugdebug/gui/breakpoints.py

@@ -12,6 +12,8 @@
 from PyQt5.QtCore import pyqtSignal
 from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem
 
+from pugdebug.models.settings import get_setting
+
 
 class PugdebugBreakpointViewer(QTreeWidget):
 
@@ -31,7 +33,8 @@ def set_breakpoints(self, breakpoints):
         self.clear()
 
         for breakpoint in breakpoints:
-            args = [breakpoint['filename'], str(breakpoint['lineno'])]
+            filename = self.__cut_filename(breakpoint['filename'])
+            args = [filename, str(breakpoint['lineno'])]
 
             item = QTreeWidgetItem(args)
 
@@ -42,3 +45,12 @@ def handle_item_double_clicked(self, item, column):
         line = int(item.text(1))
 
         self.item_double_clicked_signal.emit(file, line)
+
+    def __cut_filename(self, filename):
+        path_map = get_setting('path/path_mapping')
+        if len(path_map) > 0:
+            filename = filename[len(path_map):]
+        else:
+            root = get_setting('path/project_root')
+            filename = filename[len(root):]
+        return "~%s" % filename

pugdebug/gui/stacktraces.py

@@ -12,6 +12,8 @@
 from PyQt5.QtCore import pyqtSignal
 from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem
 
+from pugdebug.models.settings import get_setting
+
 
 class PugdebugStacktraceViewer(QTreeWidget):
 
@@ -32,8 +34,9 @@ def set_stacktraces(self, stacktraces):
         self.clear()
 
         for stacktrace in stacktraces:
+            filename = self.__cut_filename(stacktrace['filename'])
             args = [
-                stacktrace['filename'],
+                filename,
                 stacktrace['lineno'],
                 stacktrace['where']
             ]
@@ -46,3 +49,12 @@ def handle_item_double_clicked(self, item, column):
         line = int(item.text(1))
 
         self.item_double_clicked_signal.emit(file, line)
+
+    def __cut_filename(self, filename):
+        path_map = get_setting('path/path_mapping')
+        if len(path_map) > 0:
+            filename = filename[len(path_map):]
+        else:
+            root = get_setting('path/project_root')
+            filename = filename[len(root):]
+        return "~%s" % filename