forked from orbit-oss/flask
Add skip to fix unnoticed bug with good imports
Fixes logic so that imports that should not be changed are skipped, which was not happening correctly before.
This commit is contained in:
parent
818c42cfea
commit
f6c45afb6f
2 changed files with 9 additions and 1 deletions
|
|
@ -40,6 +40,8 @@ def fix_from_imports(red):
|
|||
from_imports = red.find_all("FromImport")
|
||||
for x, node in enumerate(from_imports):
|
||||
values = node.value
|
||||
if len(values) < 2:
|
||||
continue
|
||||
if (values[0].value == 'flask') and (values[1].value == 'ext'):
|
||||
# Case 1
|
||||
if len(node.value) == 3:
|
||||
|
|
|
|||
|
|
@ -45,7 +45,7 @@ def test_named_module_import():
|
|||
assert output == "import flask_foo as foobar"
|
||||
|
||||
|
||||
def test__named_from_import():
|
||||
def test_named_from_import():
|
||||
red = RedBaron("from flask.ext.foo import bar as baz")
|
||||
output = migrate.fix_tester(red)
|
||||
assert output == "from flask_foo import bar as baz"
|
||||
|
|
@ -69,3 +69,9 @@ def test_nested_function_call_migration():
|
|||
output = migrate.fix_tester(red)
|
||||
assert output == ("import flask_foo\n\n"
|
||||
"flask_foo.bar(var)")
|
||||
|
||||
|
||||
def test_no_change_to_import():
|
||||
red = RedBaron("from flask import Flask")
|
||||
output = migrate.fix_tester(red)
|
||||
assert output == "from flask import Flask"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue