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:
Keyan Pishdadian 2015-02-25 19:48:29 -05:00
parent bbd2b83d98
commit a7bf11468d
2 changed files with 9 additions and 1 deletions

View file

@ -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: