fork from github/youtube-local

youtube/home.py

@@ -0,0 +1,40 @@
+from youtube import util
+import flask
+import os
+
+def get_youtube_cookies():
+    """Read cookies.txt and return a dict of cookies for YouTube requests."""
+    cookies_path = os.path.join(os.path.dirname(__file__), 'cookies.txt')
+    cookies = {}
+    if os.path.isfile(cookies_path):
+        with open(cookies_path, 'r', encoding='utf-8') as f:
+            for line in f:
+                line = line.strip()
+                if not line or line.startswith('#') or '=' not in line:
+                    continue
+                k, v = line.split('=', 1)
+                cookies[k.strip()] = v.strip()
+    return cookies
+
+def get_recommended_videos():
+    # Use YouTube's browse API to get the home feed (recommended videos)
+    data = {"browseId": "FEwhat_to_watch"}
+    cookies = get_youtube_cookies()
+    response = util.call_youtube_api("web", "browse", data, cookies=cookies)
+    response_json = flask.json.loads(response)
+    # Extract video list from response_json
+    try:
+        contents = response_json["contents"]["twoColumnBrowseResultsRenderer"]["tabs"][0]["tabRenderer"]["content"]["richGridRenderer"]["contents"]
+        videos = []
+        for item in contents:
+            renderer = item.get("richItemRenderer", {}).get("content", {}).get("videoRenderer")
+            if renderer:
+                videos.append(renderer)
+        # If no videos found, check for nudge/empty feed
+        if not videos:
+            print("No recommended videos found. YouTube may require login or watch history.")
+        return videos
+    except Exception as e:
+        print("Error extracting recommended videos:", e)
+        print("Response JSON:", response_json)
+        return []