From eed5d059a8ea5442266be892322f8815074800a5 Mon Sep 17 00:00:00 2001 From: Xuehai Pan Date: Wed, 7 Dec 2022 02:47:55 +0800 Subject: [PATCH] Fix `black` compatibility for `.pyi` type stub files --- isort/output.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/isort/output.py b/isort/output.py index c59be936d..3cb3c08b0 100644 --- a/isort/output.py +++ b/isort/output.py @@ -209,16 +209,20 @@ def sorted_imports( break if config.lines_after_imports != -1: - formatted_output[imports_tail:0] = [ - "" for line in range(config.lines_after_imports) - ] + lines_after_imports = config.lines_after_imports + if config.profile == "black" and extension == "pyi": # special case for black + lines_after_imports = 1 + formatted_output[imports_tail:0] = ["" for line in range(lines_after_imports)] elif extension != "pyi" and next_construct.startswith(STATEMENT_DECLARATIONS): formatted_output[imports_tail:0] = ["", ""] else: formatted_output[imports_tail:0] = [""] if config.lines_before_imports != -1: - formatted_output[:0] = ["" for line in range(config.lines_before_imports)] + lines_before_imports = config.lines_before_imports + if config.profile == "black" and extension == "pyi": # special case for black + lines_before_imports = 1 + formatted_output[:0] = ["" for line in range(lines_before_imports)] if parsed.place_imports: new_out_lines = []