def _load_builtin(self, subdir, module_name, third_party_only=False):
"""Load a pytd/pyi that ships with pytype or typeshed."""
# Try our own type definitions first.
if not third_party_only:
builtin_dir = utils.get_versioned_path(subdir, self.python_version)
mod = builtins.ParsePredefinedPyTD(
builtin_dir, module_name, self.python_version)
if not mod:
mod = builtins.ParsePredefinedPyTD(
builtin_dir, module_name, self.python_version,
as_package=True)
if mod:
return self.load_file(filename=self.PREFIX + module_name,
module_name=module_name,
ast=mod)
if self.use_typeshed:
return self._load_typeshed_builtin(subdir, module_name)
return None
def _load_builtin(self, subdir, module_name):
"""Load a pytd/pyi that ships with pytype or typeshed."""
version = self.options.python_version
# Try our own type definitions first.
mod = builtins.ParsePredefinedPyTD(subdir, module_name, version)
if not mod and self.options.typeshed:
# Fall back to typeshed.
mod = typeshed.parse_type_definition(subdir, module_name, version)
if mod:
log.debug("Found %s entry for %r", subdir, module_name)
return self._load_file(filename=self.PREFIX + module_name,
module_name=module_name,
ast=mod)
return None
def testParsePredefinedPyTD(self):
"""Test ParsePredefinedPyTD()."""
ast = builtins.ParsePredefinedPyTD("builtins",
"__builtin__",
python_version=(2, 7, 6))
self.assertIsNotNone(ast.Lookup("__builtin__.int"))
def testParsePredefinedPyTD(self):
"""Test ParsePredefinedPyTD()."""
builtin_dir = "builtins/%d" % self.PYTHON_VERSION[0]
ast = builtins.ParsePredefinedPyTD(
builtin_dir, "__builtin__", python_version=self.PYTHON_VERSION)
self.assertIsNotNone(ast.Lookup("__builtin__.int"))
def testParsePredefinedPyTD(self):
"""Test ParsePredefinedPyTD()."""
ast = builtins.ParsePredefinedPyTD("builtins",
"sys",
python_version=(2, 7, 6))
self.assertIsNotNone(ast.Lookup("sys.stderr"))
