Returning just a smaller page size of limited videos in feed.

mikeckennedy · mikeckennedy · commit 7a5ea2aea5bf · 2021-09-22T10:54:30.000-07:00
diff --git a/code/ch7_infinite_scroll/ch7_final_video_collector/viewmodels/feed/feed_viewmodel.py b/code/ch7_infinite_scroll/ch7_final_video_collector/viewmodels/feed/feed_viewmodel.py
@@ -6,7 +6,14 @@
 
 
 class FeedViewModel(ViewModelBase):
-    def __init__(self):
+    def __init__(self, page_size: int, page: int = 1):
         super().__init__()
 
-        self.videos: List[Video] = video_service.all_videos()
+        self.page_size = page_size
+        self.page = page
+
+        all_videos = video_service.all_videos()
+        start = (page - 1) * page_size
+        end = start + page_size
+
+        self.videos: List[Video] = all_videos[start:end]
diff --git a/code/ch7_infinite_scroll/ch7_final_video_collector/views/feed.py b/code/ch7_infinite_scroll/ch7_final_video_collector/views/feed.py
@@ -1,16 +1,15 @@
-import time
-
 import flask
 
 from infrastructure.view_modifiers import response
 from viewmodels.feed.feed_viewmodel import FeedViewModel
 
 blueprint = flask.Blueprint('feed', __name__, template_folder='templates')
 
+VIDEOS_PER_PAGE = 3
+
 
 @blueprint.get('/feed')
 @response(template_file='feed/index.html')
 def index():
-    time.sleep(5)
-    vm = FeedViewModel()
+    vm = FeedViewModel(page_size=VIDEOS_PER_PAGE)
     return vm.to_dict()
