Support for Attributes and Decorators in _NameUtil

Instagram · Sep 15, 2023 · fc54cfb · fc54cfb
1 parent 94dd20e
commit fc54cfb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libcst/metadata/scope_provider.py b/libcst/metadata/scope_provider.py
@@ -362,12 +362,16 @@ def get_name_for(node: Union[str, cst.CSTNode]) -> Optional[str]:
  return node.value
  elif isinstance(node, str):
  return node
+ elif isinstance(node, cst.Attribute):
+ return f"{_NameUtil.get_name_for(node.value)}.{node.attr.value}"
  elif isinstance(node, cst.Call):
  return _NameUtil.get_name_for(node.func)
  elif isinstance(node, cst.Subscript):
  return _NameUtil.get_name_for(node.value)
  elif isinstance(node, (cst.FunctionDef, cst.ClassDef)):
  return _NameUtil.get_name_for(node.name)
+ elif isinstance(node, cst.Decorator):
+ return get_full_name_for_node(node.decorator)
  return None