Fix second part of #1275, when add_imports is used

PyCQA · Jul 15, 2020 · fd87001 · fd87001
1 parent fafe61d
commit fd87001
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/isort/api.py b/isort/api.py
@@ -456,6 +456,9 @@ def _sort_imports(
                     indent = line[: -len(line.lstrip())]
                 elif not (stripped_line or contains_imports):
                     if add_imports and not indent:
+                        if not import_section:
+                            output_stream.write(line)
+                            line = ""
                         import_section += line_separator.join(add_imports) + line_separator
                         contains_imports = True
                         add_imports = []

diff --git a/tests/test_regressions.py b/tests/test_regressions.py
@@ -56,6 +56,26 @@ def test_blank_lined_removed_issue_1275():
 My docstring
 """
 
+from a import other_thing
+from b import thing
+'''
+    )
+
+    assert (
+        isort.code(
+            '''"""
+My docstring
+"""
+
+from b import thing
+from a import other_thing
+''',
+            add_imports=["from b import thing"],
+        )
+        == '''"""
+My docstring
+"""
+
 from a import other_thing
 from b import thing
 '''