128 lines
4.7 KiB
Python
128 lines
4.7 KiB
Python
"""集合 下载歌词 以及 获取歌曲信息 的功能"""
|
||
import os
|
||
from requests import post
|
||
from requests.exceptions import ConnectionError
|
||
from time import sleep
|
||
from colorama import Fore, Style
|
||
|
||
from modules.utils.bar import CompactBar, bprint
|
||
from modules.utils.dump import regular_filename
|
||
|
||
|
||
def wait_retry(kind, identify, bar=None):
|
||
bprint("api提示操作频繁，等待恢复...", bar)
|
||
if kind == "information":
|
||
url = f"https://music.163.com/api/song/detail/?&ids=[{identify}]"
|
||
elif kind == "lyric":
|
||
url = f"https://music.163.com/api/song/media?id={identify}"
|
||
else:
|
||
return "unknown_kind"
|
||
while True:
|
||
try:
|
||
tmp = post(url).json()
|
||
except ConnectionError:
|
||
return "dl_err_connection"
|
||
else:
|
||
if tmp["code"] == 200:
|
||
return tmp
|
||
sleep(1)
|
||
|
||
|
||
def get_song_info_raw(types: list, identify: str, bar: CompactBar = None):
|
||
"""获取歌曲信息
|
||
|
||
``types`` 提供一个list,将会返回内部所有符合要求的信息类型\n
|
||
``identify`` 提供一个歌曲id(str),将会把歌曲的`types`信息返回"""
|
||
bprint(Fore.CYAN + "ID:%s" % identify, bar)
|
||
|
||
try:
|
||
info = post(f"https://music.163.com/api/song/detail/?&ids=[{identify}]").json()
|
||
except ConnectionError:
|
||
return "dl_err_connection"
|
||
else:
|
||
if info["code"] == 406: # 判断当操作频繁时，继续获取，直到可以返回值为200为止
|
||
result = wait_retry("information", identify, bar=bar)
|
||
if type(result) == dict:
|
||
info = result
|
||
elif result == "dl_err_connection":
|
||
return "dl_err_connection"
|
||
else:
|
||
raise Exception("Unknown exception...")
|
||
|
||
if not info.get("songs"): # 判断是否存在该歌曲
|
||
bprint(Fore.LIGHTBLACK_EX + "\t-> 这首歌没有找到，跳过...", bar)
|
||
return "song_nf"
|
||
else:
|
||
need = {}
|
||
for i in types: # 通过传入的变量 types 获取信息(根据返回的信息分析得到的下面这句语句)
|
||
need.setdefault(i, info["songs"][0][i])
|
||
return need
|
||
|
||
|
||
def get_song_lyric(identify: str | int | dict,
|
||
path: str,
|
||
lyric_format="%(name)s - %(artists)s",
|
||
allinfo: bool = False,
|
||
bar: CompactBar = None):
|
||
"""获取歌词
|
||
|
||
``identify`` 提供一个歌曲id
|
||
``path`` 提供歌曲下载的路径
|
||
``lyric_format`` 提供歌词保存的格式
|
||
``allinfo`` 若此项为 True ,则提供的identify格式必须为存储在网易云下载文件中meta_data的格式
|
||
``bar`` 若获取歌词时下方有进度条, 则应当传入此参数"""
|
||
if allinfo:
|
||
sinfo = identify
|
||
identify = identify["id"]
|
||
bprint(Fore.CYAN + f"ID: {identify}", bar)
|
||
else:
|
||
sinfo = get_song_info_raw(["name", "artists"], identify, bar)
|
||
if sinfo == "dl_err_connection": # 处理各式各样的事件
|
||
return "dl_err_connection"
|
||
elif sinfo == "song_nf":
|
||
return "song_nf"
|
||
|
||
# 整理歌曲数据，获取歌词
|
||
artists = ""
|
||
if allinfo:
|
||
for i in sinfo["artists"]:
|
||
artists += f"{i[0]},"
|
||
else:
|
||
for i in sinfo["artists"]:
|
||
artists += f"{i['name']},"
|
||
artists = artists[:-1]
|
||
|
||
name = sinfo["name"]
|
||
if not name:
|
||
bprint(Fore.RED + "歌曲错误!这是网易云的问题,请不要找作者", bar)
|
||
return "song_err"
|
||
|
||
name = regular_filename(name)
|
||
artists = regular_filename(artists)
|
||
|
||
bprint(Fore.YELLOW + "\t-> 歌曲:" + Style.RESET_ALL + f"{name} - {artists}", bar)
|
||
filename = f"{lyric_format % {'name': name, 'artists': artists}}.lrc"
|
||
|
||
try:
|
||
info = post(f"https://music.163.com/api/song/media?id={identify}").json()
|
||
except ConnectionError:
|
||
return "dl_err_connection"
|
||
else:
|
||
if info["code"] == 406: # 此处与上方一样，防止因为请求限制而跳过下载
|
||
result = wait_retry("lyric", identify, bar=bar)
|
||
if type(result) == dict:
|
||
info = result
|
||
elif result == "dl_err_connection":
|
||
return "dl_err_connection"
|
||
else:
|
||
raise Exception("Unknown exception...")
|
||
|
||
if info.get("nolyric") or not info.get('lyric'):
|
||
bprint(Fore.LIGHTBLACK_EX + "\t--> 这首歌没有歌词，跳过...\n", bar)
|
||
return
|
||
else:
|
||
with open(os.path.join(path, filename), "w", encoding="utf-8") as f:
|
||
f.write(info["lyric"])
|
||
bprint(Fore.GREEN + "\t--> 歌词下载完成!被保存在" + Style.RESET_ALL + f"{os.path.join(path, filename)}\n", bar)
|
||
return
|