@music-league-eras/local-runner 0.1.0

package/vendor/python/app/services/viewer.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+import re
+from typing import Optional
+
+from bs4 import BeautifulSoup
+
+USER_HREF_RE = re.compile(r"^/user/([0-9a-f]{32})/?$", re.I)
+VIEW_PROFILE_LABEL = "view profile"
+
+
+def extract_viewer_user_id(html: str) -> Optional[str]:
+    soup = BeautifulSoup(html or "", "html.parser")
+
+    a_tag = soup.find(
+        "a",
+        href=lambda h: bool(h and USER_HREF_RE.match(h)),
+        string=lambda s: bool(s and s.strip().lower() == VIEW_PROFILE_LABEL),
+    )
+    if a_tag and a_tag.get("href"):
+        match = USER_HREF_RE.match(a_tag.get("href").strip())
+        return match.group(1) if match else None
+
+    for link in soup.select('a[href^="/user/"]'):
+        href = (link.get("href") or "").strip()
+        match = USER_HREF_RE.match(href)
+        if match:
+            return match.group(1)
+
+    return None
+
+
+def extract_viewer_avatar_url(html: str) -> Optional[str]:
+    soup = BeautifulSoup(html or "", "html.parser")
+    viewer_id = extract_viewer_user_id(html)
+
+    def is_user_link(tag) -> bool:
+        href = (tag.get("href") or "").strip()
+        match = USER_HREF_RE.match(href)
+        return bool(match and (viewer_id is None or match.group(1) == viewer_id))
+
+    candidate_link = soup.find("a", href=lambda h: bool(h and USER_HREF_RE.match(h)))
+    if candidate_link and not is_user_link(candidate_link):
+        candidate_link = None
+
+    if candidate_link:
+        parent = candidate_link.find_parent()
+        if parent:
+            avatar = parent.find("img", src=True)
+            if avatar and avatar.get("src"):
+                return avatar.get("src")
+
+    for img in soup.select("img[src]"):
+        src = (img.get("src") or "").strip()
+        if "musicleague-user-assets" in src and (not viewer_id or viewer_id in src):
+            return src
+
+    return None

package/vendor/python/requirements.txt

@@ -0,0 +1,3 @@
+playwright==1.44.0
+beautifulsoup4==4.12.3
+