User ID should be typed as bson.ObjectID. :)

Author: mikeckennedy

src/ch16-mongodb-version/final/pypi_nosql/pypi/services/user_service.py

@@ -1,10 +1,11 @@
-import bson
 from typing import Optional
 
-from pypi import DbSession
-from pypi.nosql.users import User
+# noinspection PyPackageRequirements
+import bson
 from passlib.handlers.sha2_crypt import sha512_crypt
 
+from pypi.nosql.users import User
+
 
 def user_count() -> int:
     return User.objects().count()

src/ch16-mongodb-version/final/pypi_nosql/pypi/viewmodels/shared/viewmodel_base.py

@@ -1,3 +1,7 @@
+from typing import Optional
+
+# noinspection PyPackageRequirements
+import bson
 from pyramid.request import Request
 
 from pypi.infrastructure import request_dict, cookie_auth
@@ -7,8 +11,8 @@ class ViewModelBase:
     def __init__(self, request: Request):
         self.request = request
         self.request_dict = request_dict.create(request)
-        self.error: str = None
-        self.user_id: int = cookie_auth.get_user_id_via_auth_cookie(request)
+        self.error: Optional[str] = None
+        self.user_id: bson.ObjectId = cookie_auth.get_user_id_via_auth_cookie(request)
 
     def to_dict(self):
         return self.__dict__