42 - User SessionState Object

codingforentrepreneurs · codingforentrepreneurs · commit 4c7d8090e30f · 2024-06-13T17:57:33.000-06:00
diff --git a/full_stack_python/auth/__init__.py b/full_stack_python/auth/__init__.py
@@ -1,8 +1,11 @@
 from .models import UserInfo
 from . import pages
 
+from .state import SessionState
+
 
 __all__ = [
     'pages',
-    'UserInfo'
+    'UserInfo',
+    'SessionState'
 ]
diff --git a/full_stack_python/auth/state.py b/full_stack_python/auth/state.py
@@ -1,12 +1,37 @@
 import reflex as rx
 import reflex_local_auth
 
+import sqlmodel
+
 from .models import UserInfo
 
+
+
+
+class SessionState(reflex_local_auth.LocalAuthState):
+    @rx.cached_var
+    def authenticated_user_info(self) -> UserInfo | None:
+        if self.authenticated_user.id < 0:
+            return
+        with rx.session() as session:
+            return session.exec(
+                sqlmodel.select(UserInfo).where(
+                    UserInfo.user_id == self.authenticated_user.id
+                ),
+            ).one_or_none()
+    
+    def on_load(self):
+        if not self.is_authenticated:
+            return reflex_local_auth.LoginState.redir
+        print(self.is_authenticated)
+        print(self.authenticated_user_info)
+
+
 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)
+        print(self.new_user_id)
         if self.new_user_id >= 0:
             with rx.session() as session:
                 session.add(
diff --git a/full_stack_python/full_stack_python.py b/full_stack_python/full_stack_python.py
@@ -63,6 +63,12 @@ def index() -> rx.Component:
 app.add_page(pages.about_page, 
              route=navigation.routes.ABOUT_US_ROUTE)
 
+app.add_page(
+    pages.protected_page, 
+    route="/protected/",
+    on_load=auth.SessionState.on_load
+)
+
 app.add_page(
     blog.blog_post_list_page, 
     route=navigation.routes.BLOG_POSTS_ROUTE,
diff --git a/full_stack_python/pages/__init__.py b/full_stack_python/pages/__init__.py
@@ -1,9 +1,11 @@
 from .about import about_page
 from .pricing import pricing_page
+from .protected import protected_page
 # from .contact import contact_page
 
 __all__ = [ 
     'about_page',
     # 'contact_page',
-    'pricing_page'
+    'pricing_page',
+    'protected_page'
 ]
diff --git a/full_stack_python/pages/protected.py b/full_stack_python/pages/protected.py
@@ -0,0 +1,20 @@
+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"),
+            rx.text(
+                "Something cool about us.",
+            ),
+            spacing="5",
+            justify="center",
+            align="center",
+            min_height="85vh",
+            id='my-child'
+        )
+    return base_page(my_child)
