РЕДАКТИРОВАТЬ: см. Ответ @Eric, который создал такое расширение.
Ответ 2
Я построил расширение автоматического импорта, которое поддерживает Python. Это позволяет вам полностью настроить, как импорт записывается в файл, изменяя пути импорта, имена, сортировку относительно других импортов. Плагин Python позволяет даже группировать импорт вместе с дополнительными переносами строк.