def _postprocess_pyi(self, ast):
"""Apply all the PYI transformations we need."""
ast = ast.Visit(
visitors.LookupBuiltins(self.builtins, full_names=False))
ast = ast.Visit(visitors.ExpandCompatibleBuiltins(self.builtins))
ast = ast.Visit(visitors.LookupLocalTypes())
return ast
def _postprocess_pyi(self, ast, package_name):
"""Apply all the PYI transformations we need."""
if package_name is not None:
ast = ast.Visit(visitors.QualifyRelativeNames(package_name))
ast = ast.Visit(
visitors.LookupBuiltins(self.builtins, full_names=False))
ast = ast.Visit(visitors.ExpandCompatibleBuiltins(self.builtins))
ast = ast.Visit(visitors.LookupLocalTypes())
return ast
def _postprocess_pyi(self, ast):
"""Apply all the PYI transformations we need."""
package_name = utils.get_pyi_package_name(ast.name, ast.is_package)
if package_name:
ast = ast.Visit(visitors.QualifyRelativeNames(package_name))
ast = ast.Visit(visitors.LookupBuiltins(self.builtins, full_names=False))
ast = ast.Visit(visitors.ExpandCompatibleBuiltins(self.builtins))
dependencies = self._collect_ast_dependencies(ast)
if dependencies:
self._load_ast_dependencies(dependencies, ast)
ast = self._resolve_external_types(ast)
ast = ast.Visit(visitors.LookupLocalTypes())
return ast
def PrepareForExport(module_name, python_version, ast, loader):
"""Prepare an ast as if it was parsed and loaded.
External dependencies will not be resolved, as the ast generated by this
method is supposed to be exported.
Args:
module_name: The module_name as a string for the returned ast.
python_version: A tuple of (major, minor) python version as string
(see config.python_version).
ast: pytd.TypeDeclUnit, is only used if src is None.
loader: A load_pytd.Loader instance.
Returns:
A pytd.TypeDeclUnit representing the supplied AST as it would look after
being written to a file and parsed.
"""
# This is a workaround for functionality which crept into places it doesn't
# belong. Ideally this would call some transformation Visitors on ast to
# transform it into the same ast we get after parsing and loading (compare
# load_pytd.Loader.load_file). Unfortunately parsing has some special cases,
# e.g. '__init__' return type and '__new__' being a 'staticmethod', which
# need to be moved to visitors before we can do this. Printing an ast also
# applies transformations,
# e.g. visitors.PrintVisitor._FormatContainerContents, which need to move to
# their own visitors so they can be applied without printing.
src = pytd_utils.Print(ast)
ast = pytd_builtins.ParsePyTD(src=src,
module=module_name,
python_version=python_version)
ast = ast.Visit(visitors.LookupBuiltins(loader.builtins, full_names=False))
ast = ast.Visit(visitors.ExpandCompatibleBuiltins(loader.builtins))
ast = ast.Visit(visitors.LookupLocalTypes())
ast = ast.Visit(visitors.AdjustTypeParameters())
ast = ast.Visit(visitors.NamedTypeToClassType())
ast = ast.Visit(visitors.FillInLocalPointers({"": ast, module_name: ast}))
ast = ast.Visit(visitors.CanonicalOrderingVisitor())
ast = ast.Visit(
visitors.ClassTypeToLateType(
ignore=[module_name + ".", "__builtin__.", "typing."]))
return ast
def testExpandCompatibleBuiltins(self):
b, _ = parser_builtins.GetBuiltinsAndTyping()
src = textwrap.dedent("""
from typing import Tuple, Union
def f1(a: float) -> None: ...
def f2() -> float: ...
def f3(a: bool) -> None: ...
def f4() -> bool: ...
def f5(a: unicode) -> None: ...
def f6() -> unicode: ...
def f7(a: Union[unicode, int]) -> None: ...
def f8(a: Tuple[unicode, int]) -> None: ...
""")
expected = textwrap.dedent("""
from typing import Tuple, Union
def f1(a: Union[float, int]) -> None: ...
def f2() -> float: ...
def f3(a: Union[bool, None]) -> None: ...
def f4() -> bool: ...
def f5(a: Union[unicode, str, bytes]) -> None: ...
def f6() -> unicode: ...
def f7(a: Union[unicode, bytes, str, int]) -> None: ...
def f8(a: Tuple[Union[unicode, bytes, str], int]) -> None: ...
""")
src_tree, expected_tree = (self.Parse(s)
.Visit(visitors.LookupBuiltins(b))
for s in (src, expected))
new_tree = src_tree.Visit(visitors.ExpandCompatibleBuiltins(b))
self.AssertSourceEquals(new_tree, expected_tree)
def _postprocess_pyi(self, ast):
"""Apply all the PYI transformations we need."""
ast = ast.Visit(visitors.LookupBuiltins(self.builtins))
ast = ast.Visit(visitors.ExpandCompatibleBuiltins(self.builtins))
ast = ast.Visit(visitors.NamedTypeToClassType())
return ast