eric6/Graphics/PackageDiagramBuilder.py

changeset 8291
3d79b1e5bf3c
parent 8289
871b40c5a77a
child 8295
3f5e8b0a338e
--- a/eric6/Graphics/PackageDiagramBuilder.py	Wed May 05 17:29:41 2021 +0200
+++ b/eric6/Graphics/PackageDiagramBuilder.py	Wed May 05 18:17:24 2021 +0200
@@ -45,6 +45,12 @@
         
         self.package = os.path.abspath(package)
         self.noAttrs = noAttrs
+        
+        self.__relPackage = (
+            self.project.getRelativePath(self.package)
+            if self.project.isProjectSource(self.package) else
+            ""
+        )
     
     def initialize(self):
         """
@@ -502,3 +508,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(),
+            "no_attributes": self.noAttrs,
+        }
+        
+        data["package"] = (
+            Utilities.fromNativeSeparators(self.__relPackage)
+            if self.__relPackage else
+            Utilities.fromNativeSeparators(self.package)
+        )
+        
+        return data

eric ide

mercurial