python-auto-import: handle multiline imports