Changeset d0ce666f in sasview for src/sas/sasgui/guiframe/report_image_handler.py

Timestamp:

Sep 6, 2018 2:11:25 PM (6 years ago)

Author:

krzywon

Branches:

master, magnetic_scatt, release-4.2.2, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, unittest-saveload

Children:

44e8f48

Parents:

fa412df

Message:

Remove image references when report is closed and point to correct file locations when saving reports to PDF.

File:

• src/sas/sasgui/guiframe/report_image_handler.py (modified) (3 diffs)

src/sas/sasgui/guiframe/report_image_handler.py

@@ -1 +1 @@
 import wx
+import logging
+
+logger = logging.getLogger(__name__)
 
 
@@ -9 +12 @@
 
         def __init__(self):
+            wx.FileSystem.AddHandler(wx.MemoryFSHandler())
             self.img_holder = wx.MemoryFSHandler()
-            wx.FileSystem.AddHandler(wx.MemoryFSHandler())
             self.refs = {}
             self.indices = []
@@ -60 +63 @@
 
     @staticmethod
+    def check_for_empty_instance():
+        if ReportImageHandler.instance is None:
+            ReportImageHandler.instance = \
+                ReportImageHandler._ReportImageHandler()
+
+    @staticmethod
     def set_figs(figs, bitmaps, perspective):
-        if not ReportImageHandler.instance:
-            ReportImageHandler.instance = ReportImageHandler._ReportImageHandler()
+        ReportImageHandler.check_for_empty_instance()
         return ReportImageHandler.instance.set_figs(figs, bitmaps, perspective)
+
+    @staticmethod
+    def remove_figure(fig_url):
+        try:
+            ReportImageHandler.check_for_empty_instance()
+            ReportImageHandler.instance.refs.pop(fig_url)
+        except Exception as e:
+            logger.warn(e)