all: allow to add fields to classes

samuel-gauthier · rjarry · commit 30120a39aeda · 2023-07-12T13:19:35.000+02:00
It can be handy to add custom fields to classes, let's add support for
it.

As a side-node, this does not degrade performance for access / write for
the fields that are in the slots.

Signed-off-by: Samuel Gauthier &lt;samuel.gauthier@6wind.com&gt;
Acked-by: Robin Jarry &lt;robin@jarry.cc&gt;
diff --git a/libyang/context.py b/libyang/context.py
@@ -21,7 +21,7 @@
 
 # -------------------------------------------------------------------------------------
 class Context:
-    __slots__ = ("cdata",)
+    __slots__ = ("cdata", "__dict__")
 
     def __init__(
         self,
diff --git a/libyang/data.py b/libyang/data.py
@@ -193,7 +193,7 @@ class DNode:
     Data tree node.
     """
 
-    __slots__ = ("context", "cdata", "free_func")
+    __slots__ = ("context", "cdata", "free_func", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata):
         """
diff --git a/libyang/diff.py b/libyang/diff.py
@@ -81,7 +81,7 @@ def flatten(node, dic):
 
 # -------------------------------------------------------------------------------------
 class SNodeDiff:
-    pass
+    __slots__ = ("__dict__",)
 
 
 # -------------------------------------------------------------------------------------
diff --git a/libyang/schema.py b/libyang/schema.py
@@ -44,7 +44,7 @@ def printer_flags(
 
 # -------------------------------------------------------------------------------------
 class Module:
-    __slots__ = ("context", "cdata")
+    __slots__ = ("context", "cdata", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata):
         self.context = context
@@ -227,7 +227,7 @@ def parse_data_dict(
 
 # -------------------------------------------------------------------------------------
 class Revision:
-    __slots__ = ("context", "cdata", "module")
+    __slots__ = ("context", "cdata", "module", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata, module):
         self.context = context
@@ -270,7 +270,7 @@ def __str__(self):
 
 # -------------------------------------------------------------------------------------
 class Extension:
-    __slots__ = ("context", "cdata")
+    __slots__ = ("context", "cdata", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata, module_parent: Module = None):
         self.context = context
@@ -331,7 +331,7 @@ def module(self) -> Module:
 
 # -------------------------------------------------------------------------------------
 class _EnumBit:
-    __slots__ = ("context", "cdata")
+    __slots__ = ("context", "cdata", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata):
         self.context = context
@@ -382,7 +382,7 @@ class Bit(_EnumBit):
 
 # -------------------------------------------------------------------------------------
 class Type:
-    __slots__ = ("context", "cdata", "cdata_parsed")
+    __slots__ = ("context", "cdata", "cdata_parsed", "__dict__")
 
     UNKNOWN = lib.LY_TYPE_UNKNOWN
     BINARY = lib.LY_TYPE_BINARY
@@ -609,7 +609,7 @@ def __str__(self):
 
 # -------------------------------------------------------------------------------------
 class Feature:
-    __slots__ = ("context", "cdata")
+    __slots__ = ("context", "cdata", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata):
         self.context = context
@@ -652,7 +652,7 @@ def __str__(self):
 
 # -------------------------------------------------------------------------------------
 class IfFeatureExpr:
-    __slots__ = ("context", "cdata", "module_features", "compiled")
+    __slots__ = ("context", "cdata", "module_features", "compiled", "__dict__")
 
     def __init__(self, context: "libyang.Context", cdata, module_features=None):
         """
@@ -875,7 +875,7 @@ def __str__(self):
 
 # -------------------------------------------------------------------------------------
 class SNode:
-    __slots__ = ("context", "cdata", "cdata_parsed")
+    __slots__ = ("context", "cdata", "cdata_parsed", "__dict__")
 
     CONTAINER = lib.LYS_CONTAINER
     LEAF = lib.LYS_LEAF
