generate_qr_codes.py

#!/usr/bin/env python3
"""generate_qr_codes.py

Usage:
  generate_qr_codes.py ANDOTP_AES_BACKUP_FILE

Options:
  -h --help     Show this screen.
  --version     Show version.

"""
from docopt import docopt
import sys
import pyotp
import qrcode
import fileinput
import json
import andotp_decrypt

if __name__ == '__main__':
    arguments = docopt(__doc__, version='generate_qr_codes 0.1')

    text = andotp_decrypt.decrypt_aes(arguments['ANDOTP_AES_BACKUP_FILE'])
    if not text:
        print("Something went wrong while loading %s. Maybe the passphrase was wrong?" % arguments['ANDOTP_AES_BACKUP_FILE'])
        sys.exit(1)
    entries = json.loads(text)
    for entry in entries:
        url = None
        issuer = None
        label = entry['label']
        if " - " in label:
            issuer, label = label.split(" - ", 1)
        if entry['type'] == 'TOTP':
            totp = pyotp.TOTP(entry['secret'], interval=entry['period'])
            url = totp.provisioning_uri(label, issuer_name = issuer)
        elif entry['type'] == 'HOTP':
            totp = pyotp.TOTP(entry['secret'])
            url = totp.provisioning_uri(label, issuer_name = issuer)
        if url:
            img = qrcode.make(url)
            safe_filename = "".join([c for c in label if c.isalpha() or c.isdigit() or c in "@_-"]).strip() + ".png"
            img.save(safe_filename)
            print("Code saved as: %s" % safe_filename)