generate_qr_codes.py

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

Usage:
  generate_qr_codes.py [-o|--old] [-p|--plain] ANDOTP_AES_BACKUP_FILE

Options:
  -o --old      Use old encryption (andOTP <= 0.6.2)
  -p --plain    Load plain json file without encryption
  -h --help     Show this screen.
  --version     Show version.

"""
from docopt import docopt
import sys
import pyotp
import pyqrcode
import json
import andotp_decrypt
import os.path


def valid_filename_char(c):
    return c.isalpha() or c.isdigit() or c in "@_-"


def gen_filename(entry):
    parts = []
    label = entry.get('label')
    if label:
        parts.append(label)
    issuer = entry.get('issuer')
    if issuer:
        parts.append(issuer)

    key = '@'.join(parts)

    prefix = "".join([c for c in key if valid_filename_char(c)]).strip()

    candidate = f"{prefix}.svg"
    counter = 0
    while os.path.exists(candidate):
        counter += 1
        candidate = f"{prefix}_{counter}.svg"

    return candidate


def main():
    arguments = docopt(__doc__, version='generate_qr_codes 0.1')

    if not arguments['--plain']:
        password = andotp_decrypt.get_password()

    text = None

    if arguments['--old']:
        text = andotp_decrypt.decrypt_aes(password, arguments['ANDOTP_AES_BACKUP_FILE'])
    elif arguments['--plain']:
        with open(arguments['ANDOTP_AES_BACKUP_FILE']) as plain_file:
            text = plain_file.read()
    else:
        text = andotp_decrypt.decrypt_aes_new_format(password, arguments['ANDOTP_AES_BACKUP_FILE'])

    if not text:
        print("Something went wrong while loading %s. Maybe the passphrase was wrong?"
              " or the input file is empty!" % arguments['ANDOTP_AES_BACKUP_FILE'])
        sys.exit(1)
    entries = json.loads(text)
    for entry in entries:
        url = None
        issuer = entry.get('issuer')
        label = entry.get('label')
        if not issuer and " - " 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 = pyqrcode.create(url)
            save_filename = gen_filename(entry)
            img.svg(save_filename, scale=4, background='#fff')
            print("Code saved as: %s" % save_filename)


if __name__ == '__main__':
    main()