Tweak for better performance

Nick-Hall · Oct 24, 2024 · 3fbd62e · 3fbd62e
1 parent 56aa644
commit 3fbd62e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/gramps/gen/lib/baseobj.py b/gramps/gen/lib/baseobj.py
@@ -74,10 +74,9 @@ def get_object_state(self):
                   of the object or None if the object is empty.
         :rtype: dict
         """
-        attr_dict = {"_class": self.__class__.__name__}
-        for key, value in self.__dict__.items():
-            if not key.startswith("_"):
-                attr_dict[key] = value
+        attr_dict = dict((key, value) for key, value in self.__dict__.items()
+                         if not key.startswith("_"))
+        attr_dict[ "_class"] = self.__class__.__name__
         return attr_dict
 
     def set_object_state(self, attr_dict):
@@ -97,8 +96,7 @@ def set_object_state(self, attr_dict):
         :type attr_dict: dict
         """
         for key, value in attr_dict.items():
-            if key != "_class":
-                setattr(self, key, value)
+            setattr(self, key, value)
 
     def matches_string(self, pattern, case_sensitive=False):
         """

diff --git a/gramps/gen/lib/serialize.py b/gramps/gen/lib/serialize.py
@@ -39,6 +39,7 @@
 
 def __object_hook(obj_dict):
     obj = getattr(lib, obj_dict["_class"])()
+    del obj_dict["_class"]
     obj.set_object_state(obj_dict)
     return obj