Ran black on license_message.py

bin/license_message.py

@@ -28,53 +28,61 @@
 MESSAGE_TIMEOUT = 60 * 60 * 24  # Messages are valid for 1 day
 NONCE_BYTES = 12
 
+
 class DecryptionTimeout(Exception):
     # Decrypting a message that is too old
     pass
 
+
 def generate_key():
     key = secrets.token_bytes(32)
     return key
 
+
 def handle_generate_key(args):
     key = generate_key()
     key_b64 = base64.b64encode(key)
     print(key_b64.decode("utf-8"), file=args.outfile)
 
+
 def encrypt_message(key, message):
     nonce = secrets.token_bytes(NONCE_BYTES)
     timestamp = calendar.timegm(dt.now().utctimetuple())
-    data = timestamp.to_bytes(10, byteorder='big') + b"__" + message
+    data = timestamp.to_bytes(10, byteorder="big") + b"__" + message
     ciphertext = nonce + AESGCM(key).encrypt(nonce, data, b"")
     return ciphertext
 
+
 def handle_encrypt_message(args):
     key = base64.b64decode(args.key.encode("utf-8"))
     message = args.message.encode("utf-8")
     ciphertext = encrypt_message(key, message)
     ciphertext_b64 = base64.b64encode(ciphertext)
     print(ciphertext_b64.decode("utf-8"), file=args.outfile)
 
+
 def decrypt_message(key, ciphertext, timeout=MESSAGE_TIMEOUT):
     nonce, ciphertext = ciphertext[:NONCE_BYTES], ciphertext[NONCE_BYTES:]
     message = AESGCM(key).decrypt(nonce, ciphertext, b"")
 
     msg_timestamp, message = re.split(b"__", message, maxsplit=1)
-    msg_timestamp = int.from_bytes(msg_timestamp, byteorder='big')
+    msg_timestamp = int.from_bytes(msg_timestamp, byteorder="big")
     timestamp = calendar.timegm(dt.now().utctimetuple())
     if (timestamp - msg_timestamp) > timeout:
         raise DecryptionTimeout("The message has an expired timeout")
     return message.decode("utf-8")
 
+
 def handle_decrypt_message(args):
     key = base64.b64decode(args.key.encode("utf-8"))
     ciphertext = base64.b64decode(args.message.encode("utf-8"))
     message = decrypt_message(key, ciphertext, timeout=args.timeout)
     print(str(message), file=args.outfile)
 
+
 def parse_args(argv=None):
     parser = argparse.ArgumentParser(description=__doc__)
-    parser.add_argument("--outfile", default=sys.stdout, type=argparse.FileType('w'), help="The output file")
+    parser.add_argument("--outfile", default=sys.stdout, type=argparse.FileType("w"), help="The output file")
 
     subparsers = parser.add_subparsers(help="Available sub-commands")
 
@@ -89,11 +97,17 @@ def parse_args(argv=None):
     decrypt_parser = subparsers.add_parser("decrypt", help="Decyrpt a message")
     decrypt_parser.add_argument("--key", required=True, help="The encryption key")
     decrypt_parser.add_argument("--message", required=True, help="Message to decrypt")
-    decrypt_parser.add_argument("--timeout", default=MESSAGE_TIMEOUT, type=int, help="A message timeout. Decryption will fail for older messages")
+    decrypt_parser.add_argument(
+        "--timeout",
+        default=MESSAGE_TIMEOUT,
+        type=int,
+        help="A message timeout. Decryption will fail for older messages",
+    )
     decrypt_parser.set_defaults(func=handle_decrypt_message)
 
     return parser.parse_args(argv)
 
+
 if __name__ == "__main__":
     args = parse_args()
     args.func(args)