105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
import flet as ft
|
|
from pages.register_page import Register
|
|
from pages.forgot_password_page import ForgotPassword
|
|
from pages.two_factor_page import TwoFactorAuth
|
|
import requests
|
|
from config import API_BASE_URL
|
|
|
|
class Login:
|
|
def __init__(self, page: ft.Page, auth):
|
|
self.page = page
|
|
self.auth = auth
|
|
self.email = ft.TextField(label="Email")
|
|
self.passwd = ft.TextField(label="Password", password=True, can_reveal_password=True)
|
|
self.error_message = ft.Text("", color=ft.Colors.RED)
|
|
|
|
def on_login_click(self, e):
|
|
self.error_message.value = ""
|
|
self.error_message.update()
|
|
self.page.update()
|
|
|
|
email = self.email.value.strip()
|
|
password = self.passwd.value.strip()
|
|
|
|
if not email or not password:
|
|
self.error_message.value = "Please enter both email and password."
|
|
self.error_message.update()
|
|
return
|
|
|
|
# self.auth.placeholder.content.clean()
|
|
# two_factor = TwoFactorAuth(self.page, email, self, self.auth)
|
|
# self.auth.placeholder.content = two_factor.build()
|
|
# self.auth.placeholder.update()
|
|
|
|
try:
|
|
response = requests.post(
|
|
f"{API_BASE_URL}/auth/login",
|
|
json={"email": email, "password": password},
|
|
timeout=10
|
|
)
|
|
|
|
if response.status_code == 200:
|
|
user_info = requests.get(
|
|
f"{API_BASE_URL}/auth/me",
|
|
headers={"Authorization": f"Bearer {response.json().get('access_token', '')}"}
|
|
)
|
|
if user_info.status_code == 200:
|
|
logo_filename = user_info.json().get("logo_filename", "")
|
|
if logo_filename:
|
|
self.page.client_storage.set("custom_logo", logo_filename)
|
|
|
|
self.auth.placeholder.content.clean()
|
|
two_factor = TwoFactorAuth(self.page, email, self, self.auth)
|
|
self.auth.placeholder.content = two_factor.build()
|
|
self.auth.placeholder.update()
|
|
|
|
else:
|
|
self.error_message.value = response.json().get("error", "Login failed")
|
|
self.error_message.update()
|
|
|
|
except Exception as ex:
|
|
self.error_message.value = f"Login error: {str(ex)}"
|
|
self.error_message.update()
|
|
|
|
def on_forgot_btn_click(self, e):
|
|
self.auth.placeholder.content.clean()
|
|
forgot_passwd = ForgotPassword(self.page, self.auth, self)
|
|
self.auth.placeholder.content = forgot_passwd.build()
|
|
self.auth.placeholder.update()
|
|
|
|
def on_register_btn_click(self, e):
|
|
print('Go to register')
|
|
register = Register(self.page, self.auth, self)
|
|
self.auth.placeholder.content.clean()
|
|
self.auth.placeholder.content = register.build()
|
|
self.auth.placeholder.update()
|
|
|
|
def build(self):
|
|
return ft.Column(
|
|
[
|
|
ft.Text(
|
|
"Login",
|
|
size=30,
|
|
weight="bold"
|
|
),
|
|
self.email,
|
|
self.passwd,
|
|
ft.ElevatedButton(
|
|
"Login",
|
|
width = 150,
|
|
on_click=self.on_login_click
|
|
),
|
|
self.error_message,
|
|
ft.Row(
|
|
[
|
|
ft.TextButton("Forgot Password?", on_click=self.on_forgot_btn_click),
|
|
ft.TextButton("Register", on_click=self.on_register_btn_click)
|
|
],
|
|
alignment=ft.MainAxisAlignment.SPACE_BETWEEN
|
|
)
|
|
],
|
|
horizontal_alignment=ft.CrossAxisAlignment.CENTER,
|
|
alignment=ft.MainAxisAlignment.CENTER,
|
|
spacing=20,
|
|
width=350
|
|
) |