def get_contract_info(source_code):
class_types, reformatted_code = pre_parse(source_code)
py_ast = python_ast.parse(reformatted_code)
annotate_ast(py_ast, reformatted_code, class_types)
return py_ast, reformatted_code
def parse_to_ast(source_code: str) -> list:
if '\x00' in source_code:
raise ParserException('No null bytes (\\x00) allowed in the source code.')
class_types, reformatted_code = pre_parse(source_code)
py_ast = python_ast.parse(reformatted_code)
annotate_ast(py_ast, source_code, class_types)
# Convert to Vyper AST.
vyper_ast = parse_python_ast(
source_code=source_code,
node=py_ast,
)
return vyper_ast.body # type: ignore
def parse_to_ast(source_code: str, source_id: int = 0) -> list:
if '\x00' in source_code:
raise ParserException(
'No null bytes (\\x00) allowed in the source code.')
class_types, reformatted_code = pre_parse(source_code)
try:
py_ast = python_ast.parse(reformatted_code)
except SyntaxError as e:
# TODO: Ensure 1-to-1 match of source_code:reformatted_code SyntaxErrors
raise PythonSyntaxException(e, source_code) from e
annotate_ast(py_ast, source_code, class_types)
asttokens.ASTTokens(source_code, tree=py_ast)
# Convert to Vyper AST.
vyper_ast = parse_python_ast(
source_code=source_code,
node=py_ast,
source_id=source_id,
)
return vyper_ast.body # type: ignore