install.py

#@author : Wangs_official
import os.path
import subprocess
import sys
import pip
import pip._internal
import logging
import getpass
import time
import json

print('[~]Author:Wangs_official')
apiurl = 'https://netease-cloud-music-api-lemon.vercel.app/'  # 使用其他的API服务,请修改这里的域名

# Install Library
try:
    import requests

    print("\n[OK]requests 已安装")
except ImportError:
    print("\n[X]requests未安装,正在安装")
    subprocess.check_call(
        [sys.executable, '-m', 'pip', 'install', 'requests', "-i", "https://mirrors.aliyun.com/pypi/simple/"])

try:
    import mutagen

    print("\n[OK]mutagen 已安装")
except ImportError:
    print("\n[X]mutagen未安装,正在安装")
    subprocess.check_call(
        [sys.executable, '-m', 'pip', 'install', 'mutagen', "-i", "https://mirrors.aliyun.com/pypi/simple/"])

try:
    import yaml

    print("\n[OK]yaml 已安装")
except ImportError:
    print("\n[X]yaml未安装,正在安装")
    subprocess.check_call(
        [sys.executable, '-m', 'pip', 'install', 'pyyaml', "-i", "https://mirrors.aliyun.com/pypi/simple/"])

try:
    import wget

    print("\n[OK]wget 已安装")
except ImportError:
    print("\n[X]wget未安装,正在安装")
    subprocess.check_call(
        [sys.executable, '-m', 'pip', 'install', 'wget', "-i", "https://mirrors.aliyun.com/pypi/simple/"])

try:
    import tqdm

    print("\n[OK]tqdm 已安装")
except ImportError:
    print("\n[X]tqdm未安装,正在安装")
    subprocess.check_call(
        [sys.executable, '-m', 'pip', 'install', 'tqdm', "-i", "https://mirrors.aliyun.com/pypi/simple/"])

try:
    import colorlog

    print("\n[OK]colorlog 已安装")
except ImportError:
    print("\n[X]colorlog未安装,正在安装")
    subprocess.check_call(
        [sys.executable, '-m', 'pip', 'install', 'colorlog', "-i", "https://mirrors.aliyun.com/pypi/simple/"])

print("\n[OK]所有库安装完成")

# Write settings.yml
try:
    import yaml
except ImportError:
    print("请重新启动程序,否则刚刚安装的pyyaml无法正常加载")
    exit()
if not os.path.exists('settings.yml'):
    print("\n[...]正在生成设置文件")
    f = open("settings.yml", "w")
    f.close
    data = {'api_url': apiurl, 'use_cookie': False, 'del_tmp_when_complete': False}
    with open('settings.yml', 'w', encoding='utf-8') as f:
        yaml.dump(data=data, stream=f, allow_unicode=True)
    print("\n[OK]生成settings.yml成功!")
    print(f"\n[OK]已使用 {apiurl} 作为网易云API!")
else:
    print("\n[!]settings.yml已存在")

# Create tmp
print("\n[...]正在生成缓存文件夹")
if not os.path.exists('tmp'):
    os.makedirs('tmp')
    print("\n[OK]tmp文件夹创建成功!")
    if not os.path.exists('tmp/pics'):
        os.makedirs('tmp/pics')
        print("\n[OK]tmp/pics文件夹创建成功!")
    else:
        print("\n[!]tmp/pics文件夹已存在!")
    if not os.path.exists('tmp/songs'):
        os.makedirs('tmp/songs')
        print("\n[OK]tmp/songs文件夹创建成功!")
    else:
        print("\n[!]tmp/songs文件夹已存在!")
else:
    print("\n[!]tmp文件夹已存在,正在检查文件完整性!")
    if not os.path.exists('tmp/pics'):
        os.makedirs('tmp/pics')
        print("\n[!]tmp/pics不存在,创建成功")
    else:
        print("\n[OK]tmp/pics文件夹已存在!")
    if not os.path.exists('tmp/songs'):
        os.makedirs('tmp/songs')
        print("\n[!]tmp/songs不存在,创建成功")
    else:
        print("\n[OK]tmp/songs文件夹已存在!")

print("\n[...]正在生成输出文件夹")
if not os.path.exists('output'):
    os.makedirs('output')
    print("\n[OK]输出文件夹创建成功!")
else:
    print("\n[!]输出文件夹已存在!")

# Get Cookie
login_get_cookie = input("\n[?]要登录以获取Cookie吗(可以下载付费歌曲,如果你有会员的话),输入1获取,输入其他跳过:")
if login_get_cookie == "1":
    if not os.path.exists('cookie.txt'):
        print("\n[...]正在生成cookie.txt")
        f = open("cookie.txt", "w")
        f.close
    else:
        print("\n[!]cookie.txt已存在")
    phone_code = input('\n[?]请输入您的账号（手机号码）：')
    countrycode = input('\n[?]输入国家码（中国填86）：')
    user_password = getpass.getpass('\n[?]请输入您的密码（不显示，输入后回车即可）：')
    time.sleep(1)
    print('\n[...]正在登录...')
    login_back = requests.get(
        apiurl + '/login/cellphone?phone=' + phone_code + '&password=' + user_password + '&countrycode=' + countrycode)
    if login_back.status_code == 200:
        origin_login_back_json = json.loads(login_back.text)
        cookie_value = origin_login_back_json.get("cookie")
        user_name = origin_login_back_json.get("profile", {}).get("nickname")
        with open("cookie.txt", "w") as ck_file:
            ck_file.write(cookie_value)
            print(
                f'\n[OK]欢迎您！{user_name}，登录成功,安装完毕,欢迎使用\n如果要使用Cookie访问,请将settings.yml内 use_cookie 的值改为 true')
            exit()
    else:
        print('\n[!]出现错误!状态码：' + login_back.status_code + '\n返回内容：' + login_back.text + '\n 五秒后退出')
        time.sleep(5)
        exit()
else:
    print("\n[!]不选择登录,安装完毕,欢迎使用")
    exit()