41 - Customize User Registration for Related Model

codingforentrepreneurs · codingforentrepreneurs · commit 6a2c167765b3 · 2024-06-13T17:44:17.000-06:00
diff --git a/full_stack_python/auth/forms.py b/full_stack_python/auth/forms.py
@@ -0,0 +1,43 @@
+import reflex as rx
+import reflex_local_auth
+from reflex_local_auth.pages.components import input_100w, MIN_WIDTH
+
+from .state import MyRegisterState
+
+def register_error() -> rx.Component:
+    """Render the registration error message."""
+    return rx.cond(
+        reflex_local_auth.RegistrationState.error_message != "",
+        rx.callout(
+            reflex_local_auth.RegistrationState.error_message,
+            icon="triangle_alert",
+            color_scheme="red",
+            role="alert",
+            width="100%",
+        ),
+    )
+
+
+def my_register_form() -> rx.Component:
+    """Render the registration form."""
+    return rx.form(
+        rx.vstack(
+            rx.heading("Create an account", size="7"),
+            register_error(),
+            rx.text("Username"),
+            input_100w("username"),
+            rx.text("Email"),
+            input_100w("email", type='email'),
+            rx.text("Password"),
+            input_100w("password", type="password"),
+            rx.text("Confirm Password"),
+            input_100w("confirm_password", type="password"),
+            rx.button("Sign up", width="100%"),
+            rx.center(
+                rx.link("Login", on_click=lambda: rx.redirect(reflex_local_auth.routes.LOGIN_ROUTE)),
+                width="100%",
+            ),
+            min_width=MIN_WIDTH,
+        ),
+        on_submit=MyRegisterState.handle_registration_email,
+    )
diff --git a/full_stack_python/auth/pages.py b/full_stack_python/auth/pages.py
@@ -5,6 +5,8 @@
 
 from ..ui.base import base_page
 
+from .forms import my_register_form
+
 def my_login_page()->rx.Component:
     return base_page(
         rx.center(
@@ -25,7 +27,7 @@ def my_register_page()->rx.Component:
                 rx.vstack(
                     rx.text("Registration successful!"),
                 ),
-                rx.card(register_form()),
+                rx.card(my_register_form()),
                 
             ),
              min_height="85vh",
diff --git a/full_stack_python/auth/state.py b/full_stack_python/auth/state.py
@@ -0,0 +1,19 @@
+import reflex as rx
+import reflex_local_auth
+
+from .models import UserInfo
+
+class MyRegisterState(reflex_local_auth.RegistrationState):
+    # This event handler must be named something besides `handle_registration`!!!
+    def handle_registration_email(self, form_data):
+        registration_result = super().handle_registration(form_data)
+        if self.new_user_id >= 0:
+            with rx.session() as session:
+                session.add(
+                    UserInfo(
+                        email=form_data["email"],
+                        user_id=self.new_user_id,
+                    )
+                )
+                session.commit()
+        return registration_result
