From 1ee90447d7df8f591d787a078182a49ad2d4ebad Mon Sep 17 00:00:00 2001
From: FrankAbagnaleJr <saxplayer1216+github@gmail.com>
Date: Thu, 17 Dec 2020 14:56:55 -0600
Subject: [PATCH] Generate QR code images

Now generates QR images instead of attempting to display the QR code in the console. Places them in a subdirectory of the current directory, entitled "qr"
---
 extract_otp_secret_keys.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/extract_otp_secret_keys.py b/extract_otp_secret_keys.py
index 365d31d..e754321 100644
--- a/extract_otp_secret_keys.py
+++ b/extract_otp_secret_keys.py
@@ -48,6 +48,7 @@ import sys
 from urllib.parse import parse_qs, urlencode, urlparse, quote
 
 import generated_python.google_auth_pb2
+import os,re,string
 
 arg_parser = argparse.ArgumentParser()
 arg_parser.add_argument('--verbose', '-v', help='verbose output', action='store_true')
@@ -68,10 +69,12 @@ def convert_secret_from_bytes_to_base32_str(bytes):
     return str(base64.b32encode(otp.secret), 'utf-8').replace('=', '')
 
 
-def print_qr(data):
+def print_qr(data, name):
     qr = QRCode()
     qr.add_data(data)
-    qr.print_tty()
+    img = qr.make_image(fill_color="black", back_color="white")
+    img.save(name)
+        
 
 for line in (line.strip() for line in fileinput.input(args.infile)):
     if verbose: print(line)
@@ -85,7 +88,9 @@ for line in (line.strip() for line in fileinput.input(args.infile)):
     if verbose: print(payload)
 
     # pylint: disable=no-member
+    i = int(0)
     for otp in payload.otp_parameters:
+        i+=1
         print('\nName:   {}'.format(otp.name))
         secret = convert_secret_from_bytes_to_base32_str(otp.secret)
         print('Secret: {}'.format(secret))
@@ -97,4 +102,9 @@ for line in (line.strip() for line in fileinput.input(args.infile)):
         otp_url = 'otpauth://{}/{}?'.format('totp' if otp.type == 2 else 'hotp', quote(otp.name)) + urlencode(url_params)
         if args.qr:
             if verbose: print(otp_url)
-            print_qr(otp_url)
+            if not(os.path.exists("qr")):os.mkdir("qr")
+            pattern = re.compile('[\W_]+')
+            file_otp_name = pattern.sub('', otp.name)
+            file_otp_issuer = pattern.sub('',otp.issuer)
+            if not(file_otp_issuer):print_qr(otp_url, "qr/{}-{}.png".format(i,file_otp_name))
+            if file_otp_issuer: print_qr(otp_url, "qr/{}-{}-{}.png".format(i,file_otp_name,file_otp_issuer))