add iter & next support to Variable

Signed-off-by: Zhiyuan Chen <“[email protected]”>
ZhiyuanChen · Feb 10, 2023 · e370169 · e370169
1 parent 6233496
commit e370169
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/chanfig/flat_dict.py b/chanfig/flat_dict.py
@@ -1099,3 +1099,6 @@ def __getstate__(self, *args, **kwargs):
     def __setstate__(self, states, *args, **kwargs):
         for name, value in states.items():
             self.setattr(name, value)
+
+    def __wrapped__(self, *args, **kwargs):
+        pass
diff --git a/chanfig/variable.py b/chanfig/variable.py
@@ -393,5 +393,11 @@ def __deepcopy__(self, memo: Optional[Mapping] = None):
     def __format__(self, format_spec):
         return self.value if isinstance(self, str) else format(self.value, format_spec)
 
+    def __iter__(self):
+        return iter(self.value)
+
+    def __next__(self):
+        return next(self.value)
+
     def __repr__(self):
         return repr(self.value)