Auto-format the Python scripts

We have a few Python scripts which over the years ended up with a
variety of formatting.

This patch auto-formats them using `black -l 79` to make them more
uniform, and easier to read and write.

internal/tlsconst/generate-ciphers.py

@@ -14,7 +14,7 @@ URL = "https://www.iana.org/assignments/tls-parameters/tls-parameters-4.csv"
 
 def getCiphers():
     req = urllib.request.urlopen(URL)
-	data = req.read().decode('utf-8')
+    data = req.read().decode("utf-8")
 
     ciphers = []
     reader = csv.DictReader(data.splitlines())
@@ -37,8 +37,9 @@ def getCiphers():
 
 ciphers = getCiphers()
 
-out = open(sys.argv[1], 'w')
+out = open(sys.argv[1], "w")
-out.write("""\
+out.write(
+    """\
 package tlsconst
 
 // AUTOGENERATED - DO NOT EDIT
@@ -46,9 +47,10 @@ package tlsconst
 // This file was autogenerated by generate-ciphers.py.
 
 var cipherSuiteName = map[uint16]string{
-""")
+"""
+)
 
 for ver, desc in ciphers:
     out.write('\t%s: "%s",\n' % (ver, desc))
 
-out.write('}\n')
+out.write("}\n")

test/util/chamuyero

@@ -21,15 +21,15 @@ import time
 
 # Command-line flags.
 ap = argparse.ArgumentParser()
-ap.add_argument("script", type=argparse.FileType('r', encoding='utf8'))
+ap.add_argument("script", type=argparse.FileType("r", encoding="utf8"))
 args = ap.parse_args()
 
 # Make sure stdout is open in utf8 mode, as we will print our input, which is
 # utf8, and want it to work regardless of the environment.
-sys.stdout = open(sys.stdout.fileno(), mode='w', encoding='utf8', buffering=1)
+sys.stdout = open(sys.stdout.fileno(), mode="w", encoding="utf8", buffering=1)
 
 
-class Process (object):
+class Process(object):
     def __init__(self, cmd, **kwargs):
         self.cmd = subprocess.Popen(cmd, **kwargs)
 
@@ -45,13 +45,15 @@ class Process (object):
     def close(self):
         return self.cmd.stdin.close()
 
-class Sock (object):
+
+class Sock(object):
     """A (generic) socket.
 
     This class implements the common code for socket support.
     Subclasses will implement the behaviour specific to different socket
     types.
     """
+
     def __init__(self, addr):
         self.addr = addr
         self.sock = NotImplemented
@@ -84,7 +86,8 @@ class Sock (object):
         self.connw.close()
         self.sock.close()
 
-class UnixSock (Sock):
+
+class UnixSock(Sock):
     def __init__(self, addr):
         Sock.__init__(self, addr)
         self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
@@ -95,7 +98,7 @@ class UnixSock (Sock):
         Sock.listen(self)
 
 
-class TCPSock (Sock):
+class TCPSock(Sock):
     def __init__(self, addr):
         host, port = addr.rsplit(":", 1)
         Sock.__init__(self, (host, int(port)))
@@ -108,7 +111,7 @@ class TCPSock (Sock):
         self.has_conn.set()
 
 
-class TLSSock (Sock):
+class TLSSock(Sock):
     def __init__(self, addr):
         host, port = addr.rsplit(":", 1)
         Sock.__init__(self, (host, int(port)))
@@ -125,8 +128,9 @@ class TLSSock (Sock):
         self.has_conn.set()
 
 
-class Interpreter (object):
+class Interpreter(object):
     """Interpreter for chamuyero scripts."""
+
     def __init__(self):
         # Processes and sockets we have spawn. Indexed by the id provided by
         # the user.
@@ -176,9 +180,14 @@ class Interpreter (object):
 
             # =   Launch a process.
             if op == "=":
-                cmd = Process(params, shell=True, universal_newlines=True,
+                cmd = Process(
-                        stdin=subprocess.PIPE, stdout=subprocess.PIPE,
+                    params,
-                        stderr=subprocess.STDOUT)
+                    shell=True,
+                    universal_newlines=True,
+                    stdin=subprocess.PIPE,
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.STDOUT,
+                )
                 self.procs[proc] = cmd
 
             # |=   Launch a process, do not capture stdout.
@@ -224,16 +233,20 @@ class Interpreter (object):
             elif op == "<-":
                 read = self.procs[proc].readline()
                 if read != params + "\n":
-                    self.runtime_error("data different that expected:\n"
+                    self.runtime_error(
+                        "data different that expected:\n"
                         + "  expected: %s\n" % repr(params)
-                            + "       got: %s" % repr(read))
+                        + "       got: %s" % repr(read)
+                    )
             elif op == "<~":
                 read = self.procs[proc].readline()
                 m = re.match(params, read)
                 if m is None:
-                    self.runtime_error("data did not match regexp:\n"
+                    self.runtime_error(
+                        "data did not match regexp:\n"
                         + "  regexp: %s\n" % repr(params)
-                            + "     got: %s" % repr(read))
+                        + "     got: %s" % repr(read)
+                    )
             elif op == "<...":
                 while True:
                     read = self.procs[proc].readline()
@@ -249,8 +262,10 @@ class Interpreter (object):
             elif op == "wait":
                 retcode = self.procs[proc].wait()
                 if params and retcode != int(params):
-                    self.runtime_error("return code did not match:\n"
+                    self.runtime_error(
-                            + "  expected %s, got %d" % (params, retcode))
+                        "return code did not match:\n"
+                        + "  expected %s, got %d" % (params, retcode)
+                    )
 
             # close  Close the process.
             elif op == "close":

test/util/mail_diff

@@ -24,11 +24,11 @@ def flexible_eq(expected, got):
         if posG >= len(got):
             return False
 
-        if c == '?':
+        if c == "?":
             posG += 1
             continue
-        if c == '*':
+        if c == "*":
-            while posG < len(got) and got[posG] != '\t':
+            while posG < len(got) and got[posG] != "\t":
                 posG += 1
                 continue
             continue
@@ -54,7 +54,7 @@ def msg_equals(expected, msg):
             diff = True
             continue
 
-		if expected[h] == '*':
+        if expected[h] == "*":
             continue
 
         if not flexible_eq(val, msg[h]):
@@ -67,12 +67,16 @@ def msg_equals(expected, msg):
         return False
 
     if expected.is_multipart() != msg.is_multipart():
-		print("Multipart differs, expected %s, got %s" % (
+        print(
-			expected.is_multipart(), msg.is_multipart()))
+            "Multipart differs, expected %s, got %s"
+            % (expected.is_multipart(), msg.is_multipart())
+        )
         return False
 
     if expected.is_multipart():
-		for exp, got in itertools.zip_longest(expected.get_payload(), msg.get_payload()):
+        for exp, got in itertools.zip_longest(
+            expected.get_payload(), msg.get_payload()
+        ):
             if not msg_equals(exp, got):
                 return False
     else:
@@ -92,10 +96,8 @@ if __name__ == "__main__":
 
     # We use a custom strict policy to do more strict content validation.
     policy = email.policy.EmailPolicy(
-			utf8=True,
+        utf8=True, linesep="\r\n", refold_source="none", raise_on_defect=True
-			linesep="\r\n",
+    )
-			refold_source='none',
-			raise_on_defect=True)
 
     expected = email.parser.Parser(policy=policy).parse(open(f1))
     msg = email.parser.Parser(policy=policy).parse(open(f2))