Minor changes to typing. (#672)

Author: Dreamsorcerer

aiohttp_session/__init__.py

@@ -12,7 +12,6 @@
     Callable,
     Dict,
     Iterator,
-    Mapping,
     MutableMapping,
     Optional,
     Union,
@@ -40,7 +39,7 @@ class _CookieParams(TypedDict, total=False):
     expires: str
 
 
-class _SessionData(TypedDict, total=False):
+class SessionData(TypedDict, total=False):
     created: int
     session: Dict[str, Any]
 
@@ -53,7 +52,7 @@ def __init__(
         self,
         identity: Optional[Any],
         *,
-        data: Optional[Mapping[str, Any]],
+        data: Optional[SessionData],
         new: bool,
         max_age: Optional[int] = None,
     ) -> None:
@@ -264,7 +263,7 @@ def max_age(self) -> Optional[int]:
     def cookie_params(self) -> _CookieParams:
         return self._cookie_params
 
-    def _get_session_data(self, session: Session) -> _SessionData:
+    def _get_session_data(self, session: Session) -> SessionData:
         if session.empty:
             return {}
 

examples/postgres_storage.py

@@ -41,7 +41,6 @@ def __init__(  # type: ignore[no-any-unimported]
 
     async def load_session(self, request: web.Request) -> Session:
         cookie = self.load_cookie(request)
-        data = {}
         if cookie is None:
             return Session(None, data={}, new=True, max_age=self.max_age)
         else:

tests/test_session_dict.py

@@ -3,7 +3,7 @@
 
 import pytest
 
-from aiohttp_session import Session
+from aiohttp_session import Session, SessionData
 
 
 def test_create() -> None:
@@ -64,7 +64,7 @@ def test__repr__() -> None:
 
 def test__repr__2() -> None:
     created = int(time.time()) - 1000
-    session_data = {"session": {"key": 123}, "created": created}
+    session_data: SessionData = {"session": {"key": 123}, "created": created}
     s = Session("test_identity", data=session_data, new=False)
     assert str(
         s