init commit

transportmanager/client/pages/two_factor_page.py

@@ -0,0 +1,70 @@
+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
+            )