Artist Ripper working

happy_ripper.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from http.client import NETWORK_AUTHENTICATION_REQUIRED
 import pathlib
 import os
 import re
@@ -16,6 +17,7 @@ from mutagen.easyid3 import EasyID3
 from mutagen.mp3 import MP3
 from mutagen.id3 import ID3, APIC
 from flask import Flask, render_template, request, jsonify
+from ytmusicapi import YTMusic
 
 MUSIC_FOLDER = os.environ.get("HAPPY_RIPPER_MUSIC_DIR", "Music")
 ACCESS_URL = os.environ.get("HAPPY_RIPPER_ACCESS_URL", "https://test.test.test")
@@ -284,11 +286,32 @@ def rip_worker(rip_queue: list[str], status_callback: Callable):
             status_callback("Finished sucessfully", True)
 
 
+def get_all_playlists_from_channel(browseId: str) -> list[str]:
+    urls: list[str] = []
+    yt = YTMusic()
+    artist = yt.get_artist(browseId)
+    albums_info = artist["albums"]
+    albums = yt.get_artist_albums(
+        albums_info["browseId"],
+        albums_info["params"],
+        limit=None,
+    )
+
+    for album in albums:
+        playlist_id = album.get("playlistId")
+        if playlist_id:
+            urls.append(f"https://music.youtube.com/playlist?list={playlist_id}")
+
+    return urls
+
+
 if __name__ == "__main__":
     app = Flask(__name__)
     rip_queue: list[str] = []
     finished_queue: list[ReleaseStatus] = []
 
+    yt_music = YTMusic()
+
     current_release_info: ReleaseInfo | None = None
     current_release_status_message: str = ""
     current_release_destinations: set[str] = set()
@@ -328,13 +351,33 @@ if __name__ == "__main__":
     def index():
         return render_template("index.html")
 
+    @app.route("/browse")
+    def browse():
+        return render_template("browse.html")
+
     @app.route("/rip", methods=["POST"])
     def rip():
+        print("RIP")
+
         url = request.form.get("url")
         if url is None:
             return jsonify({"success": False, "message": "Url is None"})
 
+        is_channel = request.form.get("is-channel-check")
+        is_channel = is_channel is not None
+
+        if is_channel:
+            channel_playlists = get_all_playlists_from_channel(url)
+
+            if len(channel_playlists) == 0:
+                print("ERROR: Found no playlists in channel")
+                return jsonify({"success": False, "message": "channel has no playlists"})
+
+            for playlist_url in channel_playlists:
+                rip_queue.append(playlist_url)
+        else:
             rip_queue.append(url)
+
         return jsonify({"success": True, "message": f"Added, {url}!"})
 
     @app.route("/status", methods=["GET"])
@@ -364,4 +407,68 @@ if __name__ == "__main__":
 
         return jsonify(data)
 
+    @app.route("/search", methods=["POST"])
+    def search():
+        global yt_music
+
+        query = request.form.get("search")
+        if query is None:
+            return jsonify({"success": False, "message": "Search query is None"})
+
+        responce = yt_music.search(query)
+
+        data = []
+        for release in responce:
+            release_data = {}
+
+            # Category ===============================================================
+            category = release.get("resultType")
+            if category is None:
+                return jsonify({"success": False, "message": "Category is None"})
+
+            release_data["category"] = category
+
+            # Artist ===============================================================
+            if "artist" in release:
+                release_data["artist"] = release.get("artist", "Unknown")
+
+            elif "artists" in release:
+                artists = release.get("artists")
+                if artists is None:
+                    return jsonify({"success": False, "message": "Artists exists but is None"})
+
+                release_data["artist"] = artists[0].get("name", "Unknown")
+                print(artists)
+
+            elif "author" in release:
+                release_data["artist"] = release.get("author", "Unknown")
+
+            else:
+                release_data["artist"] = "Unknown"
+
+            # Title ===============================================================
+            if category == "artist":
+                release_data["title"] = release_data["artist"]
+            else:
+                release_data["title"] = release.get("title", "Unknown")
+
+            # Thumbnails ===============================================================
+            if "thumbnails" in release:
+                release_data["thumbnails"] = release.get("thumbnails")
+            else:
+                return jsonify({"success": False, "message": "Thumbnails is None"})
+
+            if "playlistId" in release:
+                release_data["url"] = f"https://music.youtube.com/playlist?list={release['playlistId']}"
+            elif "videoId" in release:
+                release_data["url"] = f"https://music.youtube.com/watch?v={release['videoId']}"
+            elif "browseId" in release and category == "artist":
+                release_data["url"] = release.get("browseId")
+            elif category == "artist" and "artists" in release:
+                release_data["url"] = release.get("artists")[0].get("id")
+
+            data.append(release_data)
+
+        return jsonify(data)
+
     app.run(host="0.0.0.0", port=PORT)

templates/base.html

@@ -0,0 +1,47 @@
+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1" />
+    <title>Happy Ripper</title>
+    <link
+      href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.8/dist/css/bootstrap.min.css"
+      rel="stylesheet"
+      integrity="sha384-sRIl4kxILFvY47J16cr9ZwB07vP4J8+LH7qKQnuqkuIAvNWLzeN8tE5YBujZqJLB"
+      crossorigin="anonymous"
+    />
+  </head>
+
+  <body>
+    <nav class="navbar navbar-expand-lg bg-body-tertiary">
+      <div class="container-fluid">
+        <a class="navbar-brand" href="#">Happy Ripper</a>
+        <button
+          class="navbar-toggler"
+          type="button"
+          data-bs-toggle="collapse"
+          data-bs-target="#navbarNavAltMarkup"
+          aria-controls="navbarNavAltMarkup"
+          aria-expanded="false"
+          aria-label="Toggle navigation"
+        >
+          <span class="navbar-toggler-icon"></span>
+        </button>
+        <div class="collapse navbar-collapse" id="navbarNavAltMarkup">
+          <div class="navbar-nav">
+            <a class="nav-link active" aria-current="page" href="/">Queue</a>
+            <a class="nav-link" href="/browse">Browse</a>
+          </div>
+        </div>
+      </div>
+    </nav>
+
+    <section class="content">{% block content %}{% endblock %}</section>
+
+    <script
+      src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.8/dist/js/bootstrap.bundle.min.js"
+      integrity="sha384-FKyoEForCGlyvwx9Hj09JcYn3nv7wiPVlz7YYwJrWVcXK/BmnVDxM+D2scQbITxI"
+      crossorigin="anonymous"
+    ></script>
+  </body>
+</html>

templates/browse.html

@@ -0,0 +1,95 @@
+{% extends 'base.html' %} {% block content %}
+  <div class="container">
+    <form id="browse-form">
+      <div class="mb-3">
+        <input
+          type="text"
+          name="search"
+          id="searchInput"
+          placeholder="Album, Arist, Playlist..."
+          class="form-control"
+          required
+        />
+      </div>
+  
+  
+      <button type="submit" class="btn btn-primary">Search</button>
+    </form>
+  </div>
+  
+  </br>
+
+  <div class="container" id="results"></div>
+  
+  <script>
+    async function enqueue(url, is_artist) {
+      const formData = new FormData();
+      formData.append("url", url);
+
+      if (is_artist) {
+        formData.append("is-channel-check", "");
+      }
+
+      const response = await fetch("/rip", {
+        method: "POST",
+        body: formData,
+      });
+  
+      const data = await response.json();
+    }
+
+    const searchInput = document.getElementById("searchInput");
+    const results = document.getElementById("results");
+  
+    document
+      .getElementById("browse-form")
+      .addEventListener("submit", async function (e) {
+        e.preventDefault();
+  
+        const formData = new FormData(this);
+  
+        const response = await fetch("/search", {
+          method: "POST",
+          body: formData,
+        });
+  
+        const data = await response.json();
+        console.log(data)
+
+        results.innerHTML = "";
+        for (release of data) {
+          var button = "url" in release ? `
+              <div class="col">
+                <button class="btn btn-primary" onclick="enqueue('${release.url}', ${release.category == 'artist'})">
+                  Enqueue
+                </button>
+              </div>` : "";
+
+          results.innerHTML += `
+          <div class="card" style="max-width: 720px;">
+            <div class="row">
+              <div class="col">
+                <img 
+                  src="${release.thumbnails[0].url}" 
+                  class="card-img-top"
+                  style="max-width: 120px;"
+                  >
+              </div>
+
+              <div class="col">
+                <div class="card-body">
+                  <h5 class="card-title">${release.title}</h5>
+                  <p class="card-text">${release.category} • ${release.artist}</p>
+                </div>
+              </div>
+
+              ${button}
+
+            </div>
+          </div>
+          <br>
+          `;
+        }
+      });
+  </script>
+{% endblock %}

templates/index.html

@@ -1,18 +1,4 @@
-<!doctype html>
+{% extends 'base.html' %} {% block content %}
-<html lang="en">
-  <head>
-    <meta charset="UTF-8" />
-    <meta name="viewport" content="width=device-width, initial-scale=1" />
-    <title>Happy Ripper</title>
-    <link
-      href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.8/dist/css/bootstrap.min.css"
-      rel="stylesheet"
-      integrity="sha384-sRIl4kxILFvY47J16cr9ZwB07vP4J8+LH7qKQnuqkuIAvNWLzeN8tE5YBujZqJLB"
-      crossorigin="anonymous"
-    />
-  </head>
-
-  <body>
   <div class="container">
     <h2>Request</h2>
     <form id="rip-form">
@@ -26,6 +12,12 @@
           required
         />
       </div>
+  
+      <div class="mb-3 form-check">
+        <input type="checkbox" class="form-check-input" id="is-channel-check" name="is-channel-check">
+        <label class="form-check-label" for="is-channel-check">Is Channel</label>
+      </div>
+  
       <button type="submit" class="btn btn-primary">Queue</button>
     </form>
   </div>
@@ -164,11 +156,4 @@
   
     setInterval(status, 100);
   </script>
-
+{% endblock %}
-    <script
-      src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.8/dist/js/bootstrap.bundle.min.js"
-      integrity="sha384-FKyoEForCGlyvwx9Hj09JcYn3nv7wiPVlz7YYwJrWVcXK/BmnVDxM+D2scQbITxI"
-      crossorigin="anonymous"
-    ></script>
-  </body>
-</html>