typing: fix all type issues

Author: sumnerevans

linkedin_messaging/api_objects.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 from dataclasses import dataclass, field
-from typing import Any, List, Optional
+from typing import Any, Callable, Dict, List, Optional
 
 import dataclasses_json
 from dataclasses_json import (
@@ -41,7 +41,7 @@ def __repr__(self):
     datetime: (lambda s: datetime.fromtimestamp(int(s) / 1000) if s else None),
     URN: (lambda s: URN(s) if s else None),
 }
-encoder_functions = {
+encoder_functions: Dict[Any, Callable[[Any], Any]] = {
     datetime: (lambda d: int(d.timestamp() * 1000) if d else None),
     URN: (lambda u: str(u) if u else None),
 }

linkedin_messaging/linkedin.py

@@ -80,13 +80,15 @@ def __init__(self):
     @staticmethod
     def from_pickle(pickle_str: bytes) -> "LinkedInMessaging":
         linkedin = LinkedInMessaging()
+        assert isinstance(linkedin.session.cookie_jar, aiohttp.CookieJar)
         linkedin.session.cookie_jar._cookies = pickle.loads(pickle_str)
         for c in linkedin.session.cookie_jar:
             if c.key == "JSESSIONID":
                 linkedin.session.headers["csrf-token"] = c.value.strip('"')
         return linkedin
 
     def to_pickle(self) -> bytes:
+        assert isinstance(self.session.cookie_jar, aiohttp.CookieJar)
         return pickle.dumps(self.session.cookie_jar._cookies)
 
     async def close(self):