import_mod = results.get("module_name")
if import_mod:
mod_name = import_mod.value
- new_name = str(self.mapping[mod_name])
+ new_name = self.mapping[mod_name]
import_mod.replace(Name(new_name, prefix=import_mod.get_prefix()))
if "name_import" in results:
# If it's not a "from x import x, y" or "import x as y" import,
def transform(self, node, results):
attr = results["attr"][0]
- new = str(MAP[attr.value])
+ new = MAP[attr.value]
attr.replace(Name(new, prefix=attr.get_prefix()))
#import_mod = results.get("module")
if mod_name and attr_name:
- new_attr = str(LOOKUP[(mod_name.value, attr_name.value)])
+ new_attr = LOOKUP[(mod_name.value, attr_name.value)]
attr_name.replace(Name(new_attr, prefix=attr_name.get_prefix()))
PATTERN = '|'.join(_pats)
def transform(self, node, results):
- new_value = str(_TYPE_MAPPING.get(results["name"].value))
+ new_value = _TYPE_MAPPING.get(results["name"].value)
if new_value:
return Name(new_value, prefix=node.get_prefix())
return None