2b1668206d
- padronizando o nome de get_db_connection e session para get_db_session, para não confundir com session do Flask ou sessoes web
- corrigindo potenciais erros
-- has_permission nao consegue com lazy load carregar permission depois de load_user fechar a conexao, entao joinedLoad com Permission antes de fechar
-- db.rollback não existe caso db = get_db_session() apareça muito depois dentro do try, padronizando antes de try
--- comparar role por nivel (Role.SECRETARIO_GERAL) e nao por nome ("Secretario Geral")
- unificacao de get_otp_qr_code
- mudança de nowutc() para now(UTC) conforme novo padrão
20 lines
577 B
Python
20 lines
577 B
Python
import base64
|
|
import pyotp
|
|
import qrcode
|
|
from io import BytesIO
|
|
|
|
def generate_qr_code(user):
|
|
"""Gera imagem PIL do QR code OTP para o usuário."""
|
|
qr = qrcode.QRCode(version=1, box_size=10, border=5)
|
|
qr.add_data(user.get_otp_uri())
|
|
qr.make(fit=True)
|
|
return qr.make_image(fill_color="black", back_color="white")
|
|
|
|
|
|
def generate_qr_code_base64(user):
|
|
"""Gera QR code OTP codificado em base64 (PNG)."""
|
|
img = generate_qr_code(user)
|
|
buffer = BytesIO()
|
|
img.save(buffer, format="PNG")
|
|
return base64.b64encode(buffer.getvalue()).decode("utf-8")
|