✨ Support starred variable for varname() (#96)

Author: pwwang

tests/test_varname.py

@@ -556,3 +556,14 @@ def foo3():
     assert "Skipping (0 more to skip) [In 'foo1'" in msgs[6]
     assert "Ignored by IgnoreModuleQualname('tests.test_varname', '*.wrapper')" in msgs[7]
     assert "Gotcha! [In 'test_internal_debug'" in msgs[8]
+
+
+def test_star_var():
+    def func():
+        v = varname(multi_vars=True)
+        return 1, 2, 3, v
+
+    a, *b, c = func()
+    assert a == 1
+    assert b == [2, 3]
+    assert c == ('a', '*b', 'c')

varname/utils.py

@@ -196,6 +196,8 @@ def node_name(node: ast.AST) -> Union[str, Tuple[Union[str, Tuple], ...]]:
         return node.attr
     if isinstance(node, (ast.List, ast.Tuple)):
         return tuple(node_name(elem) for elem in node.elts)
+    if isinstance(node, ast.Starred):
+        return f"*{node_name(node.value)}"
 
     raise ImproperUseError(
         f"Can only get name of a variable or attribute, "