Login finalizado, admin funcionando corretamente e sendo gerado oQRcode na raiz do projeto
This commit is contained in:
@@ -4,6 +4,36 @@ import os
|
||||
from pathlib import Path
|
||||
import pyotp
|
||||
|
||||
def generate_qr_code(user):
|
||||
"""
|
||||
Gera o QR code para um usuário específico
|
||||
|
||||
Args:
|
||||
user: Instância do modelo Usuario
|
||||
|
||||
Returns:
|
||||
Path: Caminho do arquivo QR code gerado
|
||||
"""
|
||||
import qrcode
|
||||
|
||||
# Gerar QR Code apenas na raiz do projeto
|
||||
qr_path = Path('admin_qr.png')
|
||||
|
||||
# Remover arquivo antigo se existir
|
||||
if qr_path.exists():
|
||||
os.remove(str(qr_path))
|
||||
|
||||
# Gerar e salvar QR Code
|
||||
qr = qrcode.QRCode(version=1, box_size=10, border=5)
|
||||
qr.add_data(user.get_otp_uri())
|
||||
qr.make(fit=True)
|
||||
img = qr.make_image(fill_color="black", back_color="white")
|
||||
img.save(str(qr_path))
|
||||
|
||||
print(f"\nQR Code gerado em: {os.path.abspath(qr_path)}")
|
||||
|
||||
return qr_path
|
||||
|
||||
def create_admin_user():
|
||||
try:
|
||||
# Inicializar o banco de dados
|
||||
@@ -17,6 +47,8 @@ def create_admin_user():
|
||||
|
||||
if admin:
|
||||
print("\n=== Usuário Admin Encontrado ===")
|
||||
# Atualizar QR code mesmo se o usuário já existir
|
||||
qr_path = generate_qr_code(admin)
|
||||
else:
|
||||
print("\n=== Criando Novo Usuário Admin ===")
|
||||
# Criar usuário admin com novo segredo OTP
|
||||
@@ -45,21 +77,11 @@ def create_admin_user():
|
||||
totp = pyotp.TOTP(admin.otp_secret)
|
||||
otp_uri = totp.provisioning_uri(
|
||||
name=admin.username,
|
||||
issuer_name="Sistema de Gestão"
|
||||
issuer_name="Sistema de Controles"
|
||||
)
|
||||
|
||||
# Configurar diretório para o QR Code
|
||||
home = Path.home()
|
||||
qr_dir = home / '.local' / 'share' / 'controles' / 'qrcodes'
|
||||
qr_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Gerar e salvar QR Code
|
||||
qr_path = qr_dir / 'admin_qr.png'
|
||||
qr = qrcode.QRCode(version=1, box_size=10, border=5)
|
||||
qr.add_data(otp_uri)
|
||||
qr.make(fit=True)
|
||||
img = qr.make_image(fill_color="black", back_color="white")
|
||||
img.save(str(qr_path))
|
||||
# Usar a função extraída para gerar o QR code
|
||||
qr_path = generate_qr_code(admin)
|
||||
|
||||
print("\n=== QR Code Gerado ===")
|
||||
print(f"QR Code salvo em: {qr_path}")
|
||||
|
||||
Reference in New Issue
Block a user