51 - Saving UserInfo ID to Related Models

codingforentrepreneurs · codingforentrepreneurs · commit 362716695fe3 · 2024-06-18T13:07:40.000-06:00
diff --git a/full_stack_python/auth/state.py b/full_stack_python/auth/state.py
@@ -9,6 +9,11 @@
 
 
 class SessionState(reflex_local_auth.LocalAuthState):
+    @rx.cached_var
+    def my_userinfo_id(self) -> str | None:
+        if self.authenticated_user_info is None:
+            return None
+        return self.authenticated_user_info.id
 
     @rx.cached_var
     def my_user_id(self) -> str | None:
@@ -35,7 +40,7 @@ def authenticated_user_info(self) -> UserInfo | None:
             if result is None:
                 return None
             # database lookup
-            result.user
+            # result.user
             # user_obj = result.user
             # print(result.user)
             return result
diff --git a/full_stack_python/blog/detail.py b/full_stack_python/blog/detail.py
@@ -19,6 +19,9 @@ def blog_post_detail_page() -> rx.Component:
                 edit_link_el,
                 align='end'
             ),
+            rx.text("User info id ", state.BlogPostState.post.userinfo_id),
+            rx.text("User info: ", state.BlogPostState.post.userinfo.to_string()),
+            rx.text("User: ", state.BlogPostState.post.userinfo.user.to_string()),
             rx.text(state.BlogPostState.post.publish_date),
             rx.text(
                 state.BlogPostState.post.content,
diff --git a/full_stack_python/blog/state.py b/full_stack_python/blog/state.py
@@ -5,13 +5,14 @@
 from sqlmodel import select
 
 from .. import navigation
+from ..auth.state import SessionState
 from ..models import BlogPostModel
 
 BLOG_POSTS_ROUTE = navigation.routes.BLOG_POSTS_ROUTE
 if BLOG_POSTS_ROUTE.endswith("/"):
     BLOG_POSTS_ROUTE = BLOG_POSTS_ROUTE[:-1]
 
-class BlogPostState(rx.State):
+class BlogPostState(SessionState):
     posts: List['BlogPostModel'] = []
     post: Optional['BlogPostModel'] = None
     post_content: str = ""
@@ -43,6 +44,9 @@ def get_post_detail(self):
                     (BlogPostModel.id == self.blog_post_id)
                 )
             ).one_or_none()
+            if result.userinfo: # db lookup
+                print('working')
+                # result.userinfo.user
             self.post = result
             if result is None:
                 self.post_content = ""
@@ -106,8 +110,11 @@ class BlogAddPostFormState(BlogPostState):
     form_data: dict = {}
 
     def handle_submit(self, form_data):
-        self.form_data = form_data
-        self.add_post(form_data)
+        data = form_data.copy()
+        if self.my_userinfo_id is not None:
+            data['userinfo_id'] = self.my_userinfo_id
+        self.form_data = data
+        self.add_post(data)
         return self.to_blog_post(edit_page=True)
 
 
diff --git a/full_stack_python/contact/state.py b/full_stack_python/contact/state.py
@@ -29,7 +29,8 @@ async def handle_submit(self, form_data: dict):
             data[k] = v
         if self.my_user_id is not None:
             data['user_id'] = self.my_user_id
-        print("contact data", data)
+        if self.my_userinfo_id is not None:
+            data['userinfo_id'] = self.my_userinfo_id
         with rx.session() as session:
             db_entry = ContactEntryModel(
                 **data
