32 - Edit and Save Database Data

Author: codingforentrepreneurs

full_stack_python/blog/edit.py

@@ -1,52 +1,20 @@
 import reflex as rx 
 
 from ..ui.base import base_page
-# from . import forms
 
-class EditExampleState(rx.State):
-    title: str = "Hello World"
-    content: str = "This is my blog post"
 
-    def handle_submit(self, form_data):
-        print(form_data)
+from . import forms
 
-
-def blog_post_edit_sample_form() -> rx.Component:
-    return rx.form(
-            rx.vstack(
-                rx.hstack(
-                    rx.input(
-                        value=EditExampleState.title,
-                        on_change=EditExampleState.set_title,
-                        name="title",
-                        placeholder="Title",
-                        required=True,
-                        type='text',
-                        width='100%',
-                    ),
-                    width='100%'
-                ),
-                rx.text_area(
-                    value = EditExampleState.content,
-                    on_change = EditExampleState.set_content,
-                    name='content',
-                    placeholder="Your message",
-                    required=True,
-                    height='50vh',
-                    width='100%',
-                ),
-                rx.button("Submit", type="submit"),
-            ),
-            on_submit=EditExampleState.handle_submit,
-    )
+from .state import BlogEditFormState
 
 
 
 
 def blog_post_edit_page() -> rx.Component:
-    my_form = blog_post_edit_sample_form()
+    my_form = forms.blog_post_edit_form()
+    post = BlogEditFormState.post
     my_child = rx.vstack(
-            rx.heading("Edit Blog Post", size="9"),
+            rx.heading("Editing ", post.title, size="9"),
             rx.desktop_only(
                 rx.box(
                     my_form,

full_stack_python/blog/forms.py

@@ -1,7 +1,11 @@
 import reflex as rx 
 
 
-from . import state
+from .state import (
+    BlogAddPostFormState,
+    BlogEditFormState
+)
+
 
 def blog_post_add_form() -> rx.Component:
     return rx.form(
@@ -25,6 +29,48 @@ def blog_post_add_form() -> rx.Component:
                 ),
                 rx.button("Submit", type="submit"),
             ),
-            on_submit=state.BlogAddPostFormState.handle_submit,
+            on_submit=BlogAddPostFormState.handle_submit,
             reset_on_submit=True,
+    )
+
+
+
+
+def blog_post_edit_form() -> rx.Component:
+    post = BlogEditFormState.post
+    title = post.title
+    post_content = BlogEditFormState.post_content
+    return rx.form(
+            rx.box(
+                rx.input(
+                    type='hidden',
+                    name='post_id',
+                    value=post.id
+                ),
+                display='none'
+            ), 
+            rx.vstack(
+                rx.hstack(
+                    rx.input(
+                        default_value=title,
+                        name="title",
+                        placeholder="Title",
+                        required=True,
+                        type='text',
+                        width='100%',
+                    ),
+                    width='100%'
+                ),
+                rx.text_area(
+                    value = post_content,
+                    on_change = BlogEditFormState.set_post_content,
+                    name='content',
+                    placeholder="Your message",
+                    required=True,
+                    height='50vh',
+                    width='100%',
+                ),
+                rx.button("Submit", type="submit"),
+            ),
+            on_submit=BlogEditFormState.handle_submit,
     )

full_stack_python/blog/state.py

@@ -8,6 +8,7 @@
 class BlogPostState(rx.State):
     posts: List['BlogPostModel'] = []
     post: Optional['BlogPostModel'] = None
+    post_content: str = ""
 
     @rx.var
     def blog_post_id(self):
@@ -24,6 +25,7 @@ def get_post_detail(self):
                 )
             ).one_or_none()
             self.post = result
+            self.post_content = self.post.content
         # return
 
 
@@ -45,12 +47,22 @@ def add_post(self, form_data:dict):
             session.refresh(post) # post.id
             # print("added", post)
             self.post = post
-    # def get_post(self):
-    #     with rx.session() as session:
-    #         result = session.exec(
-    #             select(BlogPostModel)
-    #         )
-    #         self.posts = result
+
+    def save_post_edits(self, post_id:int, updated_data:dict):
+        with rx.session() as session:
+            post = session.exec(
+                select(BlogPostModel).where(
+                    BlogPostModel.id == post_id
+                )
+            ).one_or_none()
+            if post is None:
+                return
+            for key, value in updated_data.items():
+                setattr(post, key, value)
+            session.add(post)
+            session.commit()
+            session.refresh(post)
+            # 
 
 
 class BlogAddPostFormState(BlogPostState):
@@ -59,4 +71,16 @@ class BlogAddPostFormState(BlogPostState):
     def handle_submit(self, form_data):
         self.form_data = form_data
         self.add_post(form_data)
+        # redirect
+
+
+class BlogEditFormState(BlogPostState):
+    form_data: dict = {}
+    # post_content: str = ""
+
+    def handle_submit(self, form_data):
+        self.form_data = form_data
+        post_id = form_data.pop('post_id')
+        updated_data = {**form_data}
+        self.save_post_edits(post_id, updated_data)
         # redirect