COLORFUL UPDATE

Make output more and more and more colorful!

Change file structure and README.md

modules/functions/mainly/get_song.py

@@ -0,0 +1,132 @@
+"""集合 下载歌词 以及 获取歌曲信息 的功能"""
+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
+
+
+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, allinfo: bool = False, bar: CompactBar = None):
+    """获取歌词
+    
+    ``identify`` 提供一个歌曲id
+    ``path`` 提供歌曲下载的路径
+    ``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"
+    replaces = {  # 处理非法字符所用的替换字典(根据网易云下载的文件分析得到)
+        "|": "｜",
+        ":": "：",
+        "<": "＜",
+        ">": "＞",
+        "?": "？",
+        "/": "／",
+        "\\": "＼",
+        "*": "＊",
+        '"': "＂"
+    }
+    for k, v in replaces.items():
+        name = name.replace(k, v)
+        artists = artists.replace(k, v)
+
+    bprint(Fore.YELLOW + "\t-> 歌曲:" + Style.RESET_ALL + f"{name} - {artists}", bar)
+    filename = f"{name} - {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