mirror of
https://github.com/Alexander-D-Karpov/scripts.git
synced 2025-06-01 11:43:08 +03:00
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
import requests
|
|
import json
|
|
import os
|
|
|
|
from urllib.parse import unquote
|
|
from pydub import AudioSegment
|
|
from mutagen.easyid3 import EasyID3
|
|
from mutagen.mp3 import MP3
|
|
from mutagen.id3 import APIC, ID3
|
|
|
|
url = input("https://castbox.fm/channel/: ")
|
|
|
|
if not url.startswith("https://castbox.fm/channel/"):
|
|
url = "https://castbox.fm/channel/" + url
|
|
|
|
|
|
def download_file(file_url):
|
|
local_filename = file_url.split("/")[-1]
|
|
with requests.get(file_url, stream=True) as r:
|
|
r.raise_for_status()
|
|
with open(local_filename, "wb") as f:
|
|
for chunk in r.iter_content(chunk_size=8192):
|
|
f.write(chunk)
|
|
return local_filename
|
|
|
|
|
|
def get_data(url):
|
|
r = requests.get(url)
|
|
if r.status_code != 200:
|
|
raise LookupError("Site not found")
|
|
inner_data = r.text.splitlines()
|
|
data = []
|
|
for line in inner_data:
|
|
if "window.__INITIAL_STATE__" in line:
|
|
data.append(line)
|
|
|
|
if len(data) != 1:
|
|
raise ValueError("Payload not found")
|
|
|
|
d = json.loads(unquote(data[0].split('"')[1::2][0])) # type: dict
|
|
return d
|
|
|
|
|
|
d = get_data(url)
|
|
while (
|
|
not d
|
|
or "ch" not in d
|
|
or "chInfo" not in d["ch"]
|
|
or "title" not in d["ch"]["chInfo"]
|
|
):
|
|
d = get_data(url)
|
|
print("Data not loaded, retrying...")
|
|
title = d["ch"]["chInfo"]["title"]
|
|
main_image = d["ch"]["chInfo"]["cover_web"]
|
|
author = d["ch"]["chInfo"]["author"]
|
|
episode_count = d["ch"]["chInfo"]["episode_count"]
|
|
print("Downloading podcast " + title)
|
|
episodes = d["ch"]["eps"]
|
|
if not os.path.isdir(title):
|
|
os.mkdir(title)
|
|
for i, episode in enumerate(episodes):
|
|
n_path = title + "/" + f"{title}.mp3" # имя из ep
|
|
if not os.path.exists(n_path): # на 1 выключается
|
|
print(f"Downloading: {episode['title']}", end="\r")
|
|
if "url" in episode and episode["url"]:
|
|
ep_url = episode["url"]
|
|
else:
|
|
ep_url = episode["urls"][0]
|
|
orig_path = download_file(ep_url)
|
|
AudioSegment.from_file(orig_path).export(n_path)
|
|
os.remove(orig_path)
|
|
if "cover_url" not in episode or not episode["cover_url"]:
|
|
img_path = download_file(main_image)
|
|
else:
|
|
img_path = download_file(episode["cover_url"])
|
|
if "author" in episode and episode["author"]:
|
|
ep_author = episode["author"]
|
|
else:
|
|
ep_author = author
|
|
|
|
print(f"Processing: {episode['title']}", end="\r")
|
|
tag = MP3(n_path, ID3=ID3)
|
|
tag.tags.add(
|
|
APIC(
|
|
encoding=3,
|
|
mime="image/png",
|
|
type=3,
|
|
desc="Cover",
|
|
data=open(img_path, "rb").read(),
|
|
)
|
|
)
|
|
tag.save()
|
|
tag = EasyID3(n_path)
|
|
|
|
tag["title"] = episode["title"]
|
|
tag["album"] = title
|
|
tag["tracknumber"] = f"{episode_count - i}/{episode_count}"
|
|
tag["artist"] = ep_author
|
|
|
|
tag.save()
|
|
os.remove(img_path)
|