summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTimothy Crosley <timothy.crosley@gmail.com>2019-12-24 00:56:37 -0800
committerTimothy Crosley <timothy.crosley@gmail.com>2019-12-24 00:56:37 -0800
commita6a32c4814df0453c468db0075dd3190f3244351 (patch)
tree828158c29d21109b02fc868cf5dccfe665ace5ec
parent7b67b7f7cca70549d452c603f4d6c199f5831402 (diff)
downloadisort-a6a32c4814df0453c468db0075dd3190f3244351.tar.gz
Fix stripping of import sections, to avoid stripping new lines before comments
-rw-r--r--isort/api.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/isort/api.py b/isort/api.py
index 8ff8ec87..cca2f5bb 100644
--- a/isort/api.py
+++ b/isort/api.py
@@ -249,7 +249,9 @@ def sort_imports(
if not contains_imports:
output_stream.write(import_section)
else:
- if first_import_section:
+ if first_import_section and not import_section.lstrip(
+ line_separator
+ ).startswith(COMMENT_INDICATORS):
import_section = import_section.lstrip(line_separator)
first_import_section = False
output_stream.write(
@@ -268,7 +270,9 @@ def sort_imports(
if not contains_imports:
output_stream.write(import_section)
else:
- if first_import_section:
+ if first_import_section and not import_section.lstrip(line_separator).startswith(
+ COMMENT_INDICATORS
+ ):
import_section = import_section.lstrip(line_separator)
output_stream.write(
output.sorted_imports(