Improve type safety for latest bs4 version

PlaidWeb · Feb 23, 2025 · cfb257d · cfb257d
1 parent 85357f8
commit cfb257d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 14 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/publ/tokens.py b/publ/tokens.py
@@ -9,6 +9,7 @@
 from typing import Optional
 
 import arrow
+import bs4
 import flask
 import itsdangerous
 import requests
@@ -150,9 +151,10 @@ def get_ticket_endpoint(me_url: str):
     if req.links and 'canonical' in req.links:
         canonical_url = req.links['canonical']['url']
     else:
-        link = content.find('link', rel='canonical')
+        link = typing.cast(bs4.Tag, content.find('link', rel='canonical'))
         if link:
-            canonical_url = urllib.parse.urljoin(me_url, link.get('href'))
+            canonical_url = urllib.parse.urljoin(me_url,
+                                                 typing.cast(str, link.get('href')))
         else:
             canonical_url = me_url