trent/flask_photo_scaling_app
1
0
Fork 0
mirror of https://github.com/TrentSPalmer/flask_photo_scaling_app.git synced 2025-02-18 13:40:55 -08:00
Code Issues Projects Releases Wiki Activity

fix style app/auth/totp.py

Browse Source
This commit is contained in:
Trent Palmer 2025-02-09 17:13:58 -08:00
parent 5eae85ca4a
commit bab12d33e9
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/auth/totp.py
View File

@ -32,7 +32,12 @@ def enable_totp():
else: else:
flash("TOTP Code didn't validate, rescan and try again") flash("TOTP Code didn't validate, rescan and try again")
return(redirect(url_for('prof.edit_profile'))) return(redirect(url_for('prof.edit_profile')))
return render_template('qr.html', qr=qr, form=form, title="Aunthentication Code") return render_template(
'qr.html',
qr=qr,
form=form,
title="Aunthentication Code",
)
def get_totp_qr(contributor): def get_totp_qr(contributor):
@ -40,7 +45,9 @@ def get_totp_qr(contributor):
contributor.totp_key = pyotp.random_base32() contributor.totp_key = pyotp.random_base32()
db.session.commit() db.session.commit()
totp_uri = pyotp.totp.TOTP(contributor.totp_key).provisioning_uri(name=contributor.email, issuer_name='Photo App') totp_uri = pyotp.totp.TOTP(
contributor.totp_key,
).provisioning_uri(name=contributor.email, issuer_name='Photo App')
img = qrcode.make(totp_uri, image_factory=qrcode.image.svg.SvgPathImage) img = qrcode.make(totp_uri, image_factory=qrcode.image.svg.SvgPathImage)
f = BytesIO() f = BytesIO()
img.save(f) img.save(f)