34 - Blog Post Redirect and Urls

codingforentrepreneurs · codingforentrepreneurs · commit 5a8352752716 · 2024-06-13T14:33:36.000-06:00
diff --git a/full_stack_python/blog/detail.py b/full_stack_python/blog/detail.py
@@ -7,7 +7,7 @@
 # @rx.page(route='/about')
 def blog_post_detail_page() -> rx.Component:
     can_edit = True
-    edit_link = rx.link("Edit", href=f"/blog/{state.BlogPostState.blog_post_id}/edit")
+    edit_link = rx.link("Edit", href=f"{state.BlogPostState.blog_post_edit_url}")
     edit_link_el = rx.cond(
         can_edit,
         edit_link,
diff --git a/full_stack_python/blog/state.py b/full_stack_python/blog/state.py
@@ -3,8 +3,13 @@
 
 from sqlmodel import select
 
+from .. import navigation
 from .model 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):
     posts: List['BlogPostModel'] = []
     post: Optional['BlogPostModel'] = None
@@ -14,6 +19,18 @@ class BlogPostState(rx.State):
     def blog_post_id(self):
         return self.router.page.params.get("blog_id", "")
 
+    @rx.var
+    def blog_post_url(self):
+        if not self.post:
+            return f"{BLOG_POSTS_ROUTE}"
+        return f"{BLOG_POSTS_ROUTE}/{self.post.id}"
+
+    @rx.var
+    def blog_post_edit_url(self):
+        if not self.post:
+            return f"{BLOG_POSTS_ROUTE}"
+        return f"{BLOG_POSTS_ROUTE}/{self.post.id}/edit"
+
     def get_post_detail(self):
         with rx.session() as session:
             if self.blog_post_id == "":
@@ -65,7 +82,14 @@ def save_post_edits(self, post_id:int, updated_data:dict):
             session.add(post)
             session.commit()
             session.refresh(post)
-            # 
+            self.post = post
+    
+    def to_blog_post(self, edit_page=False):
+        if not self.post:
+            return rx.redirect(BLOG_POSTS_ROUTE)
+        if edit_page:
+             return rx.redirect(f"{self.blog_post_edit_url}")
+        return rx.redirect(f"{self.blog_post_url}")
 
 
 class BlogAddPostFormState(BlogPostState):
@@ -74,7 +98,7 @@ class BlogAddPostFormState(BlogPostState):
     def handle_submit(self, form_data):
         self.form_data = form_data
         self.add_post(form_data)
-        # redirect
+        return self.to_blog_post(edit_page=True)
 
 
 class BlogEditFormState(BlogPostState):
@@ -86,4 +110,4 @@ def handle_submit(self, form_data):
         post_id = form_data.pop('post_id')
         updated_data = {**form_data}
         self.save_post_edits(post_id, updated_data)
-        # redirect
+        return self.to_blog_post()
