70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
import flet as ft
|
|
import requests
|
|
from config import API_BASE_URL
|
|
|
|
class TwoFactorAuth:
|
|
def __init__(self, page: ft.Page, email: str, login, auth):
|
|
self.page = page
|
|
self.email = email
|
|
self.auth = auth
|
|
self.login = login
|
|
self.code_field = ft.TextField(label="Verification Code")
|
|
self.error_text = ft.Text(value="", color=ft.Colors.RED)
|
|
self.success_text = ft.Text(value="", color=ft.Colors.GREEN)
|
|
self.verify_button = ft.ElevatedButton(text="Verify", on_click=self.on_verify_click, width=150)
|
|
self.back_button = ft.TextButton(text="Back to Login", on_click=self.on_back_clicked)
|
|
|
|
def on_verify_click(self, e):
|
|
code = self.code_field.value
|
|
|
|
if not code:
|
|
self.error_text.value = "Please enter the verification code."
|
|
self.page.update()
|
|
return
|
|
|
|
try:
|
|
response = requests.post(f"{API_BASE_URL}/auth/verify_code", json={"email": self.email, "code": code})
|
|
if response.status_code == 200:
|
|
token = response.json().get("access_token")
|
|
self.page.client_storage.set("token", token)
|
|
user_info = requests.get(f"{API_BASE_URL}/auth/me", headers={"Authorization": f"Bearer {token}"}).json()
|
|
self.page.session.set("user_id", user_info.get("id"))
|
|
self.page.session.set("first_order_number", user_info['first_order_number'])
|
|
#print(user_info.get("user_role"))
|
|
if user_info.get("user_role") == 'admin':
|
|
print('Admin Logged In')
|
|
self.page.go("/admin")
|
|
else:
|
|
self.success_text.value = "Verification successful. You are now logged in."
|
|
self.error_text.value = ""
|
|
self.page.update()
|
|
self.page.go("/dashboard") # Change this to your main page
|
|
else:
|
|
self.error_text.value = "Invalid or expired code."
|
|
self.success_text.value = ""
|
|
self.page.update()
|
|
except Exception as err:
|
|
self.error_text.value = f"Error: {err}"
|
|
self.success_text.value = ""
|
|
self.page.update()
|
|
|
|
def on_back_clicked(self, e):
|
|
self.auth.placeholder.content.clean()
|
|
self.auth.placeholder.content = self.login.build()
|
|
self.auth.placeholder.update()
|
|
|
|
def build(self):
|
|
return ft.Column(
|
|
[
|
|
ft.Text(value=f"Enter the code sent to {self.email}", size=18),
|
|
self.code_field,
|
|
self.verify_button,
|
|
self.error_text,
|
|
self.success_text,
|
|
self.back_button
|
|
],
|
|
horizontal_alignment=ft.CrossAxisAlignment.CENTER,
|
|
alignment=ft.MainAxisAlignment.CENTER,
|
|
spacing=20,
|
|
width=350
|
|
) |