Now the add video view model and view work together to capture the form data.

mikeckennedy · mikeckennedy · commit 1267ec78ed0b · 2021-09-20T10:09:38.000-07:00
diff --git a/code/ch4_app/ch4_final_video_collector/viewmodels/videos/add_video_viewmodel.py b/code/ch4_app/ch4_final_video_collector/viewmodels/videos/add_video_viewmodel.py
@@ -0,0 +1,21 @@
+from typing import Optional
+
+from viewmodels.shared.viewmodelbase import ViewModelBase
+
+
+class AddViewViewModel(ViewModelBase):
+    def __init__(self, cat_name: str):
+        super().__init__()
+
+        self.cat_name = cat_name
+        self.id: Optional[str] = None
+        self.title: Optional[str] = None
+        self.author: Optional[str] = None
+        self.view_count: Optional[int] = None
+
+    def restore_from_form(self):
+        d = self.request_dict
+        self.title = d.get('title')
+        self.author = d.get('author')
+        self.id = d.get('id')
+        self.view_count = int(d.get('view_count', 0))
diff --git a/code/ch4_app/ch4_final_video_collector/views/videos.py b/code/ch4_app/ch4_final_video_collector/views/videos.py
@@ -1,6 +1,7 @@
 import flask
 
 from infrastructure.view_modifiers import response
+from viewmodels.videos.add_video_viewmodel import AddViewViewModel
 from viewmodels.videos.category_viewmodel import CategoryViewModel
 from viewmodels.videos.play_viewmodel import PlayViewModel
 
@@ -19,3 +20,15 @@ def category(cat_name: str):
 def play(video_id: str):
     vm = PlayViewModel(video_id)
     return vm.to_dict()
+
+
+@blueprint.post('/videos/add/<cat_name>')
+def add_post(cat_name: str):
+    vm = AddViewViewModel(cat_name)
+    vm.restore_from_form()
+
+    # TODO: Add video here
+
+    return flask.redirect(f'/videos/category/{cat_name}')
+
+
