Simplify Line.contains_multiline_strings method (#1267)

This commit is contained in:
Yurii Karabas 2020-02-10 00:21:46 +03:00 committed by GitHub
parent 383857273c
commit ce14fa8b49
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -760,11 +760,9 @@ def format_str(src_contents: str, *, mode: FileMode) -> FileContent:
if supports_feature(versions, feature)
}
for current_line in lines.visit(src_node):
for _ in range(after):
dst_contents.append(str(empty_line))
dst_contents.append(str(empty_line) * after)
before, after = elt.maybe_empty_lines(current_line)
for _ in range(before):
dst_contents.append(str(empty_line))
dst_contents.append(str(empty_line) * before)
for line in split_line(
current_line, line_length=mode.line_length, features=split_line_features
):
@ -1449,11 +1447,7 @@ def contains_unsplittable_type_ignore(self) -> bool:
return False
def contains_multiline_strings(self) -> bool:
for leaf in self.leaves:
if is_multiline_string(leaf):
return True
return False
return any(is_multiline_string(leaf) for leaf in self.leaves)
def maybe_remove_trailing_comma(self, closing: Leaf) -> bool:
"""Remove trailing comma if there is one and it's safe."""