header and broken type

Author: AbstractUmbra

mystbin/client.py

@@ -68,7 +68,8 @@ def _generate_sync_session(self, session: Type["requests.Session"]) -> Type["req
         """ We will update a :class:`requests.Session` instance with the auth we require. """
         # the passed session was found to be 'sync'.
         if self.api_key:
-            session.headers.update({"Authorization": self.api_key})
+            session.headers.update(
+                {"Authorization": self.api_key, "User-Agent": "Mystbin.py"})
         return session
 
     async def _generate_async_session(self, session: Optional[aiohttp.ClientSession] = None) -> aiohttp.ClientSession:
@@ -77,7 +78,7 @@ async def _generate_async_session(self, session: Optional[aiohttp.ClientSession]
             session = aiohttp.ClientSession(raise_for_status=False)
         if self.api_key:
             session._default_headers.update(
-                {"Authorization": self.api_key})
+                {"Authorization": self.api_key, "User-Agent": "Mystbin.py"})
         session._timeout = aiohttp.ClientTimeout(CLIENT_TIMEOUT)
         return session
 
@@ -148,7 +149,7 @@ def get(self, paste_id: str) -> Union[PasteData, Awaitable]:
 
     def _perform_sync_get(self, paste_id: str, syntax: str = None) -> PasteData:
         """ Sync get request. """
-        response: requests.Response = self.session.get(
+        response: Type["requests.Response"] = self.session.get(
             f"{API_BASE_URL}/{paste_id}", timeout=CLIENT_TIMEOUT)
         if response.status_code not in (200, ):
             raise BadPasteID("This is an invalid Mystb.in paste ID.")