if isinstance(n, ast.Num):
return n.n
elif isinstance(n, ast.Str):
return n.s
return n.value
else:
def get_constant(n: ast.Constant):
return n.value
if not hasattr(_ast, "Starred"):
class Starred:
def __init__(self, *args, **kwargs):
raise NotImplementedError
ast.Starred = Starred
if not hasattr(_ast, "AnnAssign"):
class AnnAssign:
def __init__(self, *args, **kwargs):
raise NotImplementedError
ast.AnnAssign = AnnAssign
else:
ast = _ast
def get_constant(n: ast.Constant):
return n.value
