smokephil/extract_otp_secret_keys
1
0
Fork 0
mirror of https://github.com/scito/extract_otp_secret_keys.git synced 2025-12-06 06:44:57 +01:00
Code Issues Projects Releases Wiki Activity

fix optional QRCode import

Browse source
This commit is contained in:
scito 2022-09-03 23:46:05 +02:00
parent aaef496d4f
commit eac342c0cc
2 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.flake8
View file

@ -4,5 +4,3 @@ ignore =
exclude = exclude =
protobuf_generated_python protobuf_generated_python
__pycache__ __pycache__
per-file-ignores =
extract_otp_secret_keys.py: F821, F401

11
extract_otp_secret_keys.py
View file

@ -63,7 +63,8 @@ def convert_secret_from_bytes_to_base32_str(bytes):
return str(base64.b32encode(bytes), 'utf-8').replace('=', '') return str(base64.b32encode(bytes), 'utf-8').replace('=', '')
def save_qr(data, name): def save_qr(args, data, name):
from qrcode import QRCode
global verbose global verbose
qr = QRCode() qr = QRCode()
qr.add_data(data) qr.add_data(data)
@ -72,7 +73,8 @@ def save_qr(data, name):
img.save(name) img.save(name)
def print_qr(data): def print_qr(args, data):
from qrcode import QRCode
qr = QRCode() qr = QRCode()
qr.add_data(data) qr.add_data(data)
qr.print_ascii() qr.print_ascii()
@ -101,7 +103,6 @@ def sys_main():
def main(sys_args): def main(sys_args):
global verbose, quiet global verbose, quiet
args = parse_args(sys_args) args = parse_args(sys_args)
if args.saveqr or args.printqr: from qrcode import QRCode
verbose = args.verbose verbose = args.verbose
quiet = args.quiet quiet = args.quiet
@ -149,13 +150,13 @@ def extract_otps(args):
otp_url = 'otpauth://{}/{}?'.format('totp' if otp.type == 2 else 'hotp', quote(otp.name)) + urlencode(url_params) otp_url = 'otpauth://{}/{}?'.format('totp' if otp.type == 2 else 'hotp', quote(otp.name)) + urlencode(url_params)
if verbose: print(otp_url) if verbose: print(otp_url)
if args.printqr: if args.printqr:
print_qr(otp_url) print_qr(args, otp_url)
if args.saveqr: if args.saveqr:
if not (path.exists('qr')): mkdir('qr') if not (path.exists('qr')): mkdir('qr')
pattern = rcompile(r'[\W_]+') pattern = rcompile(r'[\W_]+')
file_otp_name = pattern.sub('', otp.name) file_otp_name = pattern.sub('', otp.name)
file_otp_issuer = pattern.sub('', otp.issuer) file_otp_issuer = pattern.sub('', otp.issuer)
save_qr(otp_url, 'qr/{}-{}{}.png'.format(j, file_otp_name, '-' + file_otp_issuer if file_otp_issuer else '')) save_qr(args, otp_url, 'qr/{}-{}{}.png'.format(j, file_otp_name, '-' + file_otp_issuer if file_otp_issuer else ''))
if not quiet: print() if not quiet: print()
otps.append({ otps.append({