Artist Ripper working
This commit is contained in:
+108
-1
@@ -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"})
|
||||
|
||||
rip_queue.append(url)
|
||||
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)
|
||||
|
||||
@@ -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>
|
||||
@@ -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 %}
|
||||
+137
-152
@@ -1,174 +1,159 @@
|
||||
<!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>
|
||||
{% extends 'base.html' %} {% block content %}
|
||||
<div class="container">
|
||||
<h2>Request</h2>
|
||||
<form id="rip-form">
|
||||
<div class="mb-3">
|
||||
<input
|
||||
type="text"
|
||||
name="url"
|
||||
id="urlInput"
|
||||
placeholder="https://music.youtube.com/..."
|
||||
class="form-control"
|
||||
required
|
||||
/>
|
||||
</div>
|
||||
|
||||
<body>
|
||||
<div class="container">
|
||||
<h2>Request</h2>
|
||||
<form id="rip-form">
|
||||
<div class="mb-3">
|
||||
<input
|
||||
type="text"
|
||||
name="url"
|
||||
id="urlInput"
|
||||
placeholder="https://music.youtube.com/..."
|
||||
class="form-control"
|
||||
required
|
||||
/>
|
||||
<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>
|
||||
|
||||
</br>
|
||||
|
||||
<div class="container">
|
||||
<h2>Current</h2>
|
||||
<div id="status-message"></div>
|
||||
</div>
|
||||
|
||||
</br>
|
||||
|
||||
<div class="container">
|
||||
<h2>Queue</h2>
|
||||
<table class="table" id="queue-table">
|
||||
<thead>
|
||||
<tr>
|
||||
<th scope="col">#</th>
|
||||
<th scope="col">URL</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody id="queue-table-body"></tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
</br>
|
||||
|
||||
<div class="container">
|
||||
<h2>Finished</h2>
|
||||
<div id=finish-table-body></div>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
function createCard(artist, title, status, art) {
|
||||
return `
|
||||
<div class="card mb-3" style="max-width: 540px;">
|
||||
<div class="row g-0">
|
||||
<div class="col-md-4">
|
||||
<img src="${art}" class="img-fluid rounded-start">
|
||||
</div>
|
||||
<div class="col-md-8">
|
||||
<div class="card-body">
|
||||
<h5 class="card-title">${title}</h5>
|
||||
<p class="card-text">${artist}</p>
|
||||
<p class="card-text"><small class="text-body-secondary">${status}</small></p>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<button type="submit" class="btn btn-primary">Queue</button>
|
||||
</form>
|
||||
</div>
|
||||
`;
|
||||
}
|
||||
|
||||
</br>
|
||||
function createFinishCard(artist, title, status, art, destinations) {
|
||||
let dstList = ""
|
||||
for (destination of destinations) {
|
||||
dstList += `<li>${destination}</li>`;
|
||||
}
|
||||
|
||||
<div class="container">
|
||||
<h2>Current</h2>
|
||||
<div id="status-message"></div>
|
||||
</div>
|
||||
let extraClasses = ""
|
||||
if (status !== "Finished sucessfully") {
|
||||
extraClasses = "border-danger"
|
||||
}
|
||||
|
||||
</br>
|
||||
return `
|
||||
<div class="card mb-3 ${extraClasses}" style="max-width: 540px;">
|
||||
<div class="row g-0">
|
||||
<div class="col-md-4">
|
||||
<img src="${art}" class="img-fluid rounded-start">
|
||||
</div>
|
||||
<div class="col-md-8">
|
||||
<div class="card-body">
|
||||
<h5 class="card-title">${title}</h5>
|
||||
<p class="card-text">${artist}</p>
|
||||
|
||||
<div class="container">
|
||||
<h2>Queue</h2>
|
||||
<table class="table" id="queue-table">
|
||||
<thead>
|
||||
<tr>
|
||||
<th scope="col">#</th>
|
||||
<th scope="col">URL</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody id="queue-table-body"></tbody>
|
||||
</table>
|
||||
</div>
|
||||
<ol>
|
||||
${dstList}
|
||||
</ol>
|
||||
|
||||
</br>
|
||||
|
||||
<div class="container">
|
||||
<h2>Finished</h2>
|
||||
<div id=finish-table-body></div>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
function createCard(artist, title, status, art) {
|
||||
return `
|
||||
<div class="card mb-3" style="max-width: 540px;">
|
||||
<div class="row g-0">
|
||||
<div class="col-md-4">
|
||||
<img src="${art}" class="img-fluid rounded-start">
|
||||
</div>
|
||||
<div class="col-md-8">
|
||||
<div class="card-body">
|
||||
<h5 class="card-title">${title}</h5>
|
||||
<p class="card-text">${artist}</p>
|
||||
<p class="card-text"><small class="text-body-secondary">${status}</small></p>
|
||||
</div>
|
||||
<p class="card-text"><small class="text-body-secondary">${status}</small></p>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
`;
|
||||
}
|
||||
</div>
|
||||
`;
|
||||
}
|
||||
|
||||
function createFinishCard(artist, title, status, art, destinations) {
|
||||
let dstList = ""
|
||||
for (destination of destinations) {
|
||||
dstList += `<li>${destination}</li>`;
|
||||
}
|
||||
const ripFormInput = document.getElementById("urlInput");
|
||||
|
||||
let extraClasses = ""
|
||||
if (status !== "Finished sucessfully") {
|
||||
extraClasses = "border-danger"
|
||||
}
|
||||
document
|
||||
.getElementById("rip-form")
|
||||
.addEventListener("submit", async function (e) {
|
||||
e.preventDefault();
|
||||
|
||||
return `
|
||||
<div class="card mb-3 ${extraClasses}" style="max-width: 540px;">
|
||||
<div class="row g-0">
|
||||
<div class="col-md-4">
|
||||
<img src="${art}" class="img-fluid rounded-start">
|
||||
</div>
|
||||
<div class="col-md-8">
|
||||
<div class="card-body">
|
||||
<h5 class="card-title">${title}</h5>
|
||||
<p class="card-text">${artist}</p>
|
||||
const formData = new FormData(this);
|
||||
|
||||
<ol>
|
||||
${dstList}
|
||||
</ol>
|
||||
|
||||
<p class="card-text"><small class="text-body-secondary">${status}</small></p>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
`;
|
||||
}
|
||||
|
||||
const ripFormInput = document.getElementById("urlInput");
|
||||
|
||||
document
|
||||
.getElementById("rip-form")
|
||||
.addEventListener("submit", async function (e) {
|
||||
e.preventDefault();
|
||||
|
||||
const formData = new FormData(this);
|
||||
|
||||
const response = await fetch("/rip", {
|
||||
method: "POST",
|
||||
body: formData,
|
||||
});
|
||||
|
||||
const data = await response.json();
|
||||
ripFormInput.value = "";
|
||||
const response = await fetch("/rip", {
|
||||
method: "POST",
|
||||
body: formData,
|
||||
});
|
||||
|
||||
const queueTableBody = document.getElementById("queue-table-body");
|
||||
const finishTableBody = document.getElementById("finish-table-body");
|
||||
const statusMessage = document.getElementById("status-message");
|
||||
const statusMessageImage = document.getElementById("status-message-img");
|
||||
const data = await response.json();
|
||||
ripFormInput.value = "";
|
||||
});
|
||||
|
||||
async function status() {
|
||||
const response = await fetch("/status", { method: "GET" });
|
||||
const statusJson = await response.json();
|
||||
const queueTableBody = document.getElementById("queue-table-body");
|
||||
const finishTableBody = document.getElementById("finish-table-body");
|
||||
const statusMessage = document.getElementById("status-message");
|
||||
const statusMessageImage = document.getElementById("status-message-img");
|
||||
|
||||
if ("currentReleaseInfo" in statusJson) {
|
||||
const currentInfo = statusJson.currentReleaseInfo;
|
||||
statusMessage.innerHTML = createCard(currentInfo.artist, currentInfo.title, statusJson.status, currentInfo.art_url);
|
||||
}
|
||||
else {
|
||||
statusMessage.innerHTML = `<p>${statusJson.status}</p>`;
|
||||
}
|
||||
async function status() {
|
||||
const response = await fetch("/status", { method: "GET" });
|
||||
const statusJson = await response.json();
|
||||
|
||||
queueTableBody.innerHTML = "";
|
||||
|
||||
var i = 1;
|
||||
for (const url of statusJson.queue) {
|
||||
queueTableBody.innerHTML += `<tr><td>${i}</td><td>${url}</td></tr>`;
|
||||
i ++;
|
||||
}
|
||||
|
||||
finishTableBody.innerHTML = "";
|
||||
for (const q of statusJson.finishedQueue.toReversed()) {
|
||||
finishTableBody.innerHTML += createFinishCard(q.artist, q.title, q.status, q.art_url, q.destinations);
|
||||
}
|
||||
if ("currentReleaseInfo" in statusJson) {
|
||||
const currentInfo = statusJson.currentReleaseInfo;
|
||||
statusMessage.innerHTML = createCard(currentInfo.artist, currentInfo.title, statusJson.status, currentInfo.art_url);
|
||||
}
|
||||
else {
|
||||
statusMessage.innerHTML = `<p>${statusJson.status}</p>`;
|
||||
}
|
||||
|
||||
setInterval(status, 100);
|
||||
</script>
|
||||
queueTableBody.innerHTML = "";
|
||||
|
||||
<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>
|
||||
var i = 1;
|
||||
for (const url of statusJson.queue) {
|
||||
queueTableBody.innerHTML += `<tr><td>${i}</td><td>${url}</td></tr>`;
|
||||
i ++;
|
||||
}
|
||||
|
||||
finishTableBody.innerHTML = "";
|
||||
for (const q of statusJson.finishedQueue.toReversed()) {
|
||||
finishTableBody.innerHTML += createFinishCard(q.artist, q.title, q.status, q.art_url, q.destinations);
|
||||
}
|
||||
}
|
||||
|
||||
setInterval(status, 100);
|
||||
</script>
|
||||
{% endblock %}
|
||||
|
||||
Reference in New Issue
Block a user