LunaChocken
128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
import re
|
|
|
|
from loguru import logger
|
|
from nicegui import ui
|
|
|
|
from libs.scraper import (
|
|
fetch_bandcamp_url,
|
|
find_bandcamp_artists,
|
|
find_bandcamp_name,
|
|
request_site,
|
|
)
|
|
|
|
|
|
class DataCard(ui.card):
|
|
def __init__(self, image_url: str, name: str, type: str, artist: str, url: str):
|
|
super().__init__()
|
|
self.name = name
|
|
self.url = url
|
|
self.tmp_dir = f"/tmp/bandcamp_dl/{self.name}"
|
|
with self:
|
|
ui.image(image_url).style("width: 100%")
|
|
ui.label(f"{artist} - {name} ({type})")
|
|
ui.separator()
|
|
with ui.row():
|
|
ui.button(text="Close", on_click=self.delete)
|
|
self.fetch_button = ui.button(text="Fetch", on_click=self.fetch)
|
|
self.download_button = ui.button(
|
|
text="Download", on_click=self.download
|
|
)
|
|
self.download_button.disable()
|
|
|
|
def delete(self):
|
|
self.remove(self)
|
|
|
|
# Downloads track/album to client
|
|
def download(self):
|
|
# first zip the folder
|
|
import os
|
|
import zipfile
|
|
|
|
zip_path = f"{self.tmp_dir}.zip"
|
|
with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zipf:
|
|
for root, dirs, files in os.walk(self.tmp_dir):
|
|
for file in files:
|
|
zipf.write(
|
|
os.path.join(root, file),
|
|
os.path.relpath(os.path.join(root, file), self.tmp_dir),
|
|
)
|
|
ui.download(zip_path, filename=f"{self.name}.zip")
|
|
# os.remove(zip_path)
|
|
|
|
def execute(self, base):
|
|
import subprocess as sp
|
|
|
|
sp.call(["bandcamp-dl", "--base-dir", base, self.url])
|
|
|
|
# Downloads track/album to folder
|
|
def fetch(self):
|
|
# Use bandcamp_dl to fetch the track/album
|
|
# mktmp
|
|
import os
|
|
|
|
try:
|
|
os.makedirs(self.tmp_dir)
|
|
self.execute(self.tmp_dir)
|
|
except FileExistsError:
|
|
# If empty, remove the tmp_dir and fetch again
|
|
if not os.listdir(self.tmp_dir):
|
|
os.rmdir(self.tmp_dir)
|
|
self.fetch()
|
|
else:
|
|
ui.notify("Folder not empty, cannot fetch")
|
|
|
|
if os.path.exists(self.tmp_dir):
|
|
self.download_button.enable()
|
|
self.fetch_button.remove(self.fetch_button)
|
|
|
|
|
|
def sanitise_url(url: str) -> str:
|
|
return re.match(
|
|
r"^https?://[a-zA-Z0-9.-]+\.bandcamp\.com/(album|track)/[a-zA-Z0-9_-]+/?$",
|
|
url,
|
|
).group(0)
|
|
|
|
|
|
class Main:
|
|
def __init__(self):
|
|
self.user_input = ui.input(label="Download bandcamp url")
|
|
self.button = ui.button(text="Fetch", on_click=self.fetch)
|
|
|
|
def fetch(self):
|
|
url = self.user_input.value
|
|
if not url:
|
|
return
|
|
elif not sanitise_url(url):
|
|
return
|
|
|
|
result = fetch_bandcamp_url(url)
|
|
if result:
|
|
# ui.context.client.elements is a dict {id: element_object}
|
|
all_elements = ui.context.client.elements.values()
|
|
|
|
# Check if the track is already present
|
|
already_exists = any(
|
|
isinstance(c, DataCard) and c.name == result["name"]
|
|
for c in all_elements
|
|
)
|
|
|
|
if already_exists:
|
|
ui.notify("This track is already present.", type="warning")
|
|
return
|
|
|
|
# Create the new card if it doesn't exist
|
|
DataCard(
|
|
result["image_url"],
|
|
result["name"],
|
|
result["type"],
|
|
result["artist"],
|
|
url=url,
|
|
)
|
|
|
|
|
|
def root():
|
|
main = Main()
|
|
|
|
|
|
ui.run(root)
|