feat: support vars_only to keep parents of an attribute for helpers.jsobj() (#110)

Author: pwwang

tests/test_helpers.py

@@ -84,6 +84,19 @@ def test_jsobj():
     obj = jsobj(a, b, c=3)
     assert obj == {"a": 1, "b": 2, "c": 3}
 
+    x = lambda: None
+    x.b = 2
+    obj = jsobj(a, x.b)
+    assert obj == {"a": 1, "b": 2}
+    obj1 = jsobj(a, x.b, vars_only=False)
+    assert obj1 == {"a": 1, "x.b": 2}
+
+    vars_only = 3
+    obj2 = jsobj(vars_only, x.b)
+    assert obj2 == {"vars_only": 3, "b": 2}
+    obj3 = jsobj(vars_only, x.b, vars_only=False)
+    assert obj3 == {"vars_only": 3, "x.b": 2}
+
 
 def test_register_to_function():
     @register

varname/helpers.py

@@ -153,7 +153,7 @@ def __repr__(self) -> str:
         )
 
 
-def jsobj(*args: Any, **kwargs: Any) -> Dict[str, Any]:
+def jsobj(*args: Any, vars_only: bool = True, **kwargs: Any) -> Dict[str, Any]:
     """A wrapper to create a JavaScript-like object
 
     When an argument is passed as positional argument, the name of the variable
@@ -172,11 +172,12 @@ def jsobj(*args: Any, **kwargs: Any) -> Dict[str, Any]:
     Args:
         *args: The positional arguments
         **kwargs: The keyword arguments
+        vars_only: Whether to only include variables in the output
 
     Returns:
         A dict-like object
     """
-    argnames: Tuple[str, ...] = argname("args")  # type: ignore
+    argnames: Tuple[str, ...] = argname("args", vars_only=vars_only)  # type: ignore
     out = dict(zip(argnames, args))
     out.update(kwargs)
     return out