diff options
| author | Xuehai Pan <XuehaiPan@pku.edu.cn> | 2022-12-07 02:47:55 +0800 |
|---|---|---|
| committer | Xuehai Pan <XuehaiPan@pku.edu.cn> | 2022-12-07 02:53:07 +0800 |
| commit | eed5d059a8ea5442266be892322f8815074800a5 (patch) | |
| tree | 895b8218293419eb3db5604fd6f3b3609428a815 | |
| parent | 12cc5fbd67eebf92eb2213b03c07b138ae1fb448 (diff) | |
| download | isort-eed5d059a8ea5442266be892322f8815074800a5.tar.gz | |
Fix `black` compatibility for `.pyi` type stub files
| -rw-r--r-- | isort/output.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/isort/output.py b/isort/output.py index c59be936..3cb3c08b 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 = [] |