43 - Fixing Registration Bug with Extended Model Data

codingforentrepreneurs · codingforentrepreneurs · commit 9748c94f07f9 · 2024-06-13T18:20:10.000-06:00
diff --git a/full_stack_python/auth/state.py b/full_stack_python/auth/state.py
@@ -28,11 +28,30 @@ def on_load(self):
 
 
 class MyRegisterState(reflex_local_auth.RegistrationState):
-    # This event handler must be named something besides `handle_registration`!!!
+    def handle_registration(
+        self, form_data
+    ) -> rx.event.EventSpec | list[rx.event.EventSpec]:
+        """Handle registration form on_submit.
+
+        Set error_message appropriately based on validation results.
+
+        Args:
+            form_data: A dict of form fields and values.
+        """
+        username = form_data["username"]
+        password = form_data["password"]
+        validation_errors = self._validate_fields(
+            username, password, form_data["confirm_password"]
+        )
+        if validation_errors:
+            self.new_user_id = -1
+            return validation_errors
+        self._register_user(username, password)
+        return self.new_user_id
+    
     def handle_registration_email(self, form_data):
-        registration_result = super().handle_registration(form_data)
-        print(self.new_user_id)
-        if self.new_user_id >= 0:
+        new_user_id = self.handle_registration(form_data)
+        if new_user_id >= 0:
             with rx.session() as session:
                 session.add(
                     UserInfo(
@@ -41,4 +60,4 @@ def handle_registration_email(self, form_data):
                     )
                 )
                 session.commit()
-        return registration_result
+        return type(self).successful_registration
diff --git a/full_stack_python/pages/protected.py b/full_stack_python/pages/protected.py
@@ -1,10 +1,7 @@
 import reflex as rx 
-import reflex_local_auth
 
 from ..ui.base import base_page
 
-# @rx.page(route='/about')
-@reflex_local_auth.require_login
 def protected_page() -> rx.Component:
     my_child = rx.vstack(
             rx.heading("Protect Page", size="9"),
