Mercurial Repositories > eric / file diff

--- a/eric6/Graphics/ImportsDiagramBuilder.py	Wed May 05 17:29:41 2021 +0200
+++ b/eric6/Graphics/ImportsDiagramBuilder.py	Wed May 05 18:17:24 2021 +0200
@@ -50,6 +50,12 @@
         
         self.showExternalImports = showExternalImports
         self.packagePath = os.path.abspath(package)
+        
+        self.__relPackagePath = (
+            self.project.getRelativePath(self.packagePath)
+            if self.project.isProjectSource(self.packagePath) else
+            ""
+        )
     
     def initialize(self):
         """
@@ -358,3 +364,23 @@
         self.initialize()
         
         return True
+    
+    def toDict(self):
+        """
+        Public method to collect data to be persisted.
+        
+        @return dictionary containing data to be persisted
+        @rtype dict
+        """
+        data = {
+            "project_name": self.project.getProjectName(),
+            "show_external": self.showExternalImports,
+        }
+        
+        data["package"] = (
+            Utilities.fromNativeSeparators(self.__relPackagePath)
+            if self.__relPackagePath else
+            Utilities.fromNativeSeparators(self.packagePath)
+        )
+        
+        return data