youtube link to scratch forum link

Author: FAReTek1

scratchattach/__init__.py

@@ -15,7 +15,7 @@
 from .site.backpack_asset import BackpackAsset
 from .site.comment import Comment
 from .site.cloud_activity import CloudActivity
-from .site.forum import ForumPost, ForumTopic, get_topic, get_topic_list
+from .site.forum import ForumPost, ForumTopic, get_topic, get_topic_list, youtube_link_to_scratch
 from .site.project import Project, get_project, search_projects, explore_projects
 from .site.session import Session, login, login_by_id, login_by_session_string
 from .site.studio import Studio, get_studio, search_studios, explore_studios

scratchattach/site/forum.py

@@ -6,9 +6,11 @@
 from ._base import BaseSiteComponent
 import xml.etree.ElementTree as ET
 from bs4 import BeautifulSoup
+from urllib.parse import urlparse, parse_qs
 
 from ..utils.requests import Requests as requests
 
+
 class ForumTopic(BaseSiteComponent):
     '''
     Represents a Scratch forum topic.
@@ -383,3 +385,18 @@ def get_topic_list(category_id, *, page=1):
     except Exception as e:
         raise exceptions.ScrapeError(str(e))
 
+
+def youtube_link_to_scratch(link: str):
+    """
+    Converts a YouTube url (in multiple formats) like https://youtu.be/1JTgg4WVAX8?si=fIEskaEaOIRZyTAz
+    to a link like https://scratch.mit.edu/discuss/youtube/1JTgg4WVAX8
+    """
+    url_parse = urlparse(link)
+    query_parse = parse_qs(url_parse.query)
+
+    if 'v' in query_parse:
+        video_id = query_parse['v'][0]
+    else:
+        video_id = url_parse.path.split('/')[-1]
+
+    return f"https://scratch.mit.edu/discuss/youtube/{video_id}"

scratchattach/utils/requests.py

@@ -9,7 +9,7 @@ class Requests:
     """
 
     @staticmethod
-    def check_response(r : requests.Response):
+    def check_response(r: requests.Response):
         if r.status_code == 403 or r.status_code == 401:
             raise exceptions.Unauthorized
         if r.status_code == 500: