Now exchanging data with viewmodels.

Author: mikeckennedy

code/ch5-viewmodels/services/package_service.py

@@ -0,0 +1,26 @@
+from typing import List
+
+from starlette.requests import Request
+
+from viewmodels.shared.viewmodel import ViewModelBase
+
+
+class IndexViewModel(ViewModelBase):
+    def __init__(self, request: Request):
+        super().__init__(request)
+
+        self.release_count: int = 1
+        self.user_count: int = 2
+        self.package_count: int = 3
+        self.packages: List = []
+
+        # {
+        #     'package_count': 274_000,
+        #     'release_count': 2_234_847,
+        #     'user_count': 73_874,
+        #     'packages': [
+        #         {'id': 'fastapi', 'summary': "A great web framework"},
+        #         {'id': 'uvicorn', 'summary': "Your favorite ASGI server"},
+        #         {'id': 'httpx', 'summary': "Requests for an async world"},
+        #     ]
+        # }

code/ch5-viewmodels/services/user_service.py

@@ -0,0 +1,14 @@
+from typing import Optional
+
+from starlette.requests import Request
+
+
+class ViewModelBase:
+
+    def __init__(self, request: Request):
+        self.request: Request = request
+        self.error: Optional[str] = None
+        self.user_id: Optional[int] = None
+
+    def to_dict(self) -> dict:
+        return self.__dict__

code/ch5-viewmodels/viewmodels/home/indexviewmodel.py

@@ -1,22 +1,17 @@
 import fastapi
 from fastapi_chameleon import template
+from starlette.requests import Request
+
+from viewmodels.home.indexviewmodel import IndexViewModel
 
 router = fastapi.APIRouter()
 
 
 @router.get('/')
 @template()
-def index():
-    return {
-        'package_count': 274_000,
-        'release_count': 2_234_847,
-        'user_count': 73_874,
-        'packages': [
-            {'id': 'fastapi', 'summary': "A great web framework"},
-            {'id': 'uvicorn', 'summary': "Your favorite ASGI server"},
-            {'id': 'httpx', 'summary': "Requests for an async world"},
-        ]
-    }
+def index(request: Request):
+    vm = IndexViewModel(request)
+    return vm.to_dict()
 
 
 @router.get('/about')