hgext/beautifygraph.py

--- a/hgext/beautifygraph.py	Sat Oct 05 10:29:34 2019 -0400
+++ b/hgext/beautifygraph.py	Sun Oct 06 09:45:02 2019 -0400
@@ -28,55 +28,64 @@
 # leave the attribute unspecified.
 testedwith = 'ships-with-hg-core'
 
+
 def prettyedge(before, edge, after):
     if edge == '~':
-        return '\xE2\x95\xA7' # U+2567 ╧
+        return '\xE2\x95\xA7'  # U+2567 ╧
     if edge == '/':
-        return '\xE2\x95\xB1' # U+2571 ╱
+        return '\xE2\x95\xB1'  # U+2571 ╱
     if edge == '-':
-        return '\xE2\x94\x80' # U+2500 ─
+        return '\xE2\x94\x80'  # U+2500 ─
     if edge == '|':
-        return '\xE2\x94\x82' # U+2502 │
+        return '\xE2\x94\x82'  # U+2502 │
     if edge == ':':
-        return '\xE2\x94\x86' # U+2506 ┆
+        return '\xE2\x94\x86'  # U+2506 ┆
     if edge == '\\':
-        return '\xE2\x95\xB2' # U+2572 ╲
+        return '\xE2\x95\xB2'  # U+2572 ╲
     if edge == '+':
-        if before == ' ' and not after  == ' ':
-            return '\xE2\x94\x9C' # U+251C ├
-        if after  == ' ' and not before == ' ':
-            return '\xE2\x94\xA4' # U+2524 ┤
-        return '\xE2\x94\xBC' # U+253C ┼
+        if before == ' ' and not after == ' ':
+            return '\xE2\x94\x9C'  # U+251C ├
+        if after == ' ' and not before == ' ':
+            return '\xE2\x94\xA4'  # U+2524 ┤
+        return '\xE2\x94\xBC'  # U+253C ┼
     return edge
 
+
 def convertedges(line):
     line = ' %s ' % line
     pretty = []
     for idx in pycompat.xrange(len(line) - 2):
-        pretty.append(prettyedge(line[idx:idx + 1],
-                                 line[idx + 1:idx + 2],
-                                 line[idx + 2:idx + 3]))
+        pretty.append(
+            prettyedge(
+                line[idx : idx + 1],
+                line[idx + 1 : idx + 2],
+                line[idx + 2 : idx + 3],
+            )
+        )
     return ''.join(pretty)
 
+
 def getprettygraphnode(orig, *args, **kwargs):
     node = orig(*args, **kwargs)
     if node == 'o':
-        return '\xE2\x97\x8B' # U+25CB ○
+        return '\xE2\x97\x8B'  # U+25CB ○
     if node == '@':
-        return '\xE2\x97\x8D' # U+25CD ◍
+        return '\xE2\x97\x8D'  # U+25CD ◍
     if node == '*':
-        return '\xE2\x88\x97' # U+2217 ∗
+        return '\xE2\x88\x97'  # U+2217 ∗
     if node == 'x':
-        return '\xE2\x97\x8C' # U+25CC ◌
+        return '\xE2\x97\x8C'  # U+25CC ◌
     if node == '_':
-        return '\xE2\x95\xA4' # U+2564 ╤
+        return '\xE2\x95\xA4'  # U+2564 ╤
     return node
 
+
 def outputprettygraph(orig, ui, graph, *args, **kwargs):
     (edges, text) = zip(*graph)
     graph = zip([convertedges(e) for e in edges], text)
     return orig(ui, graph, *args, **kwargs)
 
+
 def extsetup(ui):
     if ui.plain('graph'):
         return
@@ -86,8 +95,12 @@
         return
 
     if r'A' in encoding._wide:
-        ui.warn(_('beautifygraph: unsupported terminal settings, '
-                  'monospace narrow text required\n'))
+        ui.warn(
+            _(
+                'beautifygraph: unsupported terminal settings, '
+                'monospace narrow text required\n'
+            )
+        )
         return
 
     extensions.wrapfunction(graphmod, 'outputgraph', outputprettygraph)