Add trailing comma when a single import doesn't fit on a line. (#504)

[etc/vim.git] / black.py

diff --git a/black.py b/black.py
index 56765313aa85b0571c7831f209573bd2b3fbb955..f76f0ffc24cfcf237d13fcc19f5492635158c4c4 100644 (file)
--- a/black.py
+++ b/black.py
@@ -2209,7 +2209,7 @@ def left_hand_split(line: Line, py36: bool = False) -> Iterator[Line]:
             yield result
 
 
-def right_hand_split(
+def right_hand_split(  # noqa C901
     line: Line, line_length: int, py36: bool = False, omit: Collection[LeafID] = ()
 ) -> Iterator[Line]:
     """Split line into many lines, starting with the last matching bracket pair.
@@ -2250,6 +2250,9 @@ def right_hand_split(
         # the matching `opening_bracket` wasn't available on `line` anymore.
         raise CannotSplit("No brackets found")
 
+    if line.is_import and len(body_leaves) == 1:
+        body_leaves.append(Leaf(token.COMMA, ","))
+
     # Build the new lines.
     for result, leaves in (head, head_leaves), (body, body_leaves), (tail, tail_leaves):
         for leaf in leaves: