48 - Multiple Ways to Associate Users to Data Models

Author: codingforentrepreneurs

full_stack_python/auth/state.py

@@ -10,6 +10,12 @@
 
 class SessionState(reflex_local_auth.LocalAuthState):
 
+    @rx.cached_var
+    def my_user_id(self) -> str | None:
+        if self.authenticated_user.id < 0:
+            return None
+        return self.authenticated_user.id
+    
     @rx.cached_var
     def authenticated_username(self) -> str | None:
         if self.authenticated_user.id < 0:

full_stack_python/contact/form.py

@@ -1,10 +1,22 @@
 import reflex as rx 
 
-
+from ..auth.state import SessionState
 from .state import ContactState
 
 def contact_form() -> rx.Component:
     return rx.form(
+            # rx.cond(
+            #     SessionState.my_user_id,
+            #     rx.box(
+            #         rx.input(
+            #             type='hidden',
+            #             name='user_id',
+            #             value=SessionState.my_user_id
+            #         ),
+            #         display='none'
+            #     ),
+            #     rx.fragment('')
+            # ),
             rx.vstack(
                 rx.hstack(
                     rx.input(

full_stack_python/contact/page.py

@@ -1,5 +1,6 @@
 import reflex as rx 
 
+from ..auth.state import SessionState
 from .. import navigation
 from ..ui.base import base_page
 
@@ -31,6 +32,7 @@ def contact_entries_list_page() ->rx.Component:
     ) 
 
 def contact_page() -> rx.Component:
+    
     my_child = rx.vstack(
             rx.heading("Contact Us", size="9"),
             rx.cond(state.ContactState.did_submit, state.ContactState.thank_you, ""),

full_stack_python/contact/state.py

@@ -5,7 +5,9 @@
 from sqlmodel import select
 from .model import ContactEntryModel
 
-class ContactState(rx.State):
+from ..auth.state import SessionState
+
+class ContactState(SessionState):
     form_data: dict = {}
     entries: List['ContactEntryModel'] = []
     did_submit: bool = False
@@ -24,6 +26,9 @@ async def handle_submit(self, form_data: dict):
             if v == "" or v is None:
                 continue
             data[k] = v
+        if self.my_user_id is not None:
+            data['user_id'] = self.my_user_id
+        print("contact data", data)
         with rx.session() as session:
             db_entry = ContactEntryModel(
                 **data