From d04c472600644620a68a4d3a7fd784f37a1f4c53 Mon Sep 17 00:00:00 2001 From: Solderpunk Date: Sun, 10 May 2020 14:17:35 +0200 Subject: [PATCH] Reduce code duplication. --- av98.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/av98.py b/av98.py index 2600e0e..f86b89f 100755 --- a/av98.py +++ b/av98.py @@ -610,6 +610,12 @@ Slow internet connection? Use 'set timeout' to be more patient.""") debug_text = "\x1b[0;32m[DEBUG] " + debug_text + "\x1b[0m" print(debug_text) + def _activate_client_cert(self, certfile, keyfile): + self.client_certs["active"] = (certfile, keyfile) + self.active_cert_domains = [] + self.prompt = self.cert_prompt + self._debug("Using ID {} / {}.".format(*self.client_certs["active"])) + def _deactivate_client_cert(self): self.client_certs["active"] = None self.active_cert_domains = [] @@ -702,9 +708,7 @@ Slow internet connection? Use 'set timeout' to be more patient.""") certfile = input("Certfile path: ") print("Loading private key file, in PEM format (blank line to cancel)") keyfile = input("Keyfile path: ") - self.client_certs["active"] = (certfile, keyfile) - self.active_cert_domains = [] - self.prompt = self.cert_prompt + self._activate_client_cert(certfile, keyfile) @restricted def do_handler(self, line): @@ -1125,8 +1129,7 @@ def main(): # Act on args if args.tls_cert: # If tls_key is None, python will attempt to load the key from tls_cert. - gc.client_certs["active"] = (args.tls_cert, args.tls_key) - gc.prompt = gc.cert_prompt + self._activate_client_cert(args.tls_cert, args.tls_key) if args.bookmarks: gc.cmdqueue.append("bookmarks") elif args.url: