def visit_ImportFrom(self, n: ast27.ImportFrom) -> ImportBase:
i = None # type: ImportBase
if len(n.names) == 1 and n.names[0].name == '*':
i = ImportAll(n.module, n.level)
else:
i = ImportFrom(
self.translate_module_id(n.module) if n.module is not None else
'', n.level, [(a.name, a.asname) for a in n.names])
self.imports.append(i)
return i
def visit_ImportFrom(self, n: ast27.ImportFrom) -> ImportBase:
assert n.level is not None
if len(n.names) == 1 and n.names[0].name == '*':
mod = n.module if n.module is not None else ''
i = ImportAll(mod, n.level) # type: ImportBase
else:
i = ImportFrom(
self.translate_module_id(n.module) if n.module is not None else
'', n.level, [(a.name, a.asname) for a in n.names])
self.imports.append(i)
return self.set_line(i, n)
def visit_ImportFrom(self, n: ast27.ImportFrom) -> ImportBase:
assert n.level is not None
if len(n.names) == 1 and n.names[0].name == '*':
mod = n.module if n.module is not None else ''
i = ImportAll(mod, n.level) # type: ImportBase
else:
module_id = self.translate_module_id(n.module) if n.module is not None else ''
i = ImportFrom(module_id, n.level, [(a.name, a.asname) for a in n.names])
# See comments in the constructor for more information about this field.
if module_id == '__future__' and any(a.name == 'unicode_literals' for a in n.names):
self.unicode_literals = True
self.imports.append(i)
return self.set_line(i, n)
def visit_import_all(self, node: ImportAll) -> Node:
return ImportAll(node.id, node.relative)
