nem大型更新

neteasemusic.py

@@ -1,18 +1,28 @@
 import json
 import requests
 import re
+import base64
+import codecs
+import random
+import math
 from time import sleep
 from pagermaid.listener import listener
 from pagermaid import bot
 from pagermaid.utils import obtain_message
 from os import remove, path
+from os.path import exists
 from collections import defaultdict
 
+songid=''
+name=''
+
 
 @listener(is_plugin=True, outgoing=True, command="nem",
           description="网易云搜/点歌。",
           parameters="<指令> <关键词>")
 async def nem(context):
+    global name
+    global songid
     headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/52.0.2743.116 Safari/537.36 Edge/15.15063',
                "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9"}
     if len(context.parameter) < 2:
@@ -120,12 +130,18 @@ async def nem(context):
                 req = json.loads(req.content)
                 if req['result']:
                     info = {'id': '', 'title': '', 'alias': '',
-                            'album': '', 'albumpic': '', 'artist': ''}
+                            'album': '', 'albumpic': '', 'artist': '','br':''}
                     info['id'] = req['result']['songs'][0]['id']
                     info['title'] = req['result']['songs'][0]['name']
                     info['alias'] = req['result']['songs'][0]['alias']
                     info['album'] = req['result']['songs'][0]['album']['name']
                     info['albumpic'] = req['result']['songs'][0]['album']['picUrl']
+                    if req['result']['songs'][0]['hMusic']:
+                        info['br'] = req['result']['songs'][0]['hMusic']['bitrate']
+                    elif req['result']['songs'][0]['mMusic']:
+                        info['br'] = req['result']['songs'][0]['mMusic']['bitrate']
+                    elif req['result']['songs'][0]['lMusic']:
+                        info['br'] = req['result']['songs'][0]['lMusic']['bitrate']
                     for j in range(len(req['result']['songs'][0]['artists'])):
                         info['artist'] += req['result']['songs'][0]['artists'][j]['name'] + ";"
                     info['artist'] = info['artist'][:-1]
@@ -134,13 +150,184 @@ async def nem(context):
                     else:
                         title = f"【{info['title']}】"
                     await context.edit(f"{title}下载中 . . .")
-                    # 下载
+                    
-                    music = requests.request(
+                    try:
-                        "GET", "http://music.163.com/song/media/outer/url?id=" + str(info['id']) + ".mp3", headers=headers)
+                        from Crypto.Cipher import AES
+                        AES.new("0CoJUm6Qyw8W8jud".encode('utf-8'), AES.MODE_CBC, "0102030405060708".encode('utf-8'))
+                        ccimported=True
+                    except ImportError:
+                        ccimported=False
+                        await bot.send_message(context.chat_id, '(`PyCryptodome`支持库未安装，音乐曲库/音质受限\n请使用 `-sh` `pip3` `install` `pycryptodome` 安装，或自行ssh安装)')
                     name = info['title'].replace('/', " ") + ".mp3"
+                    if ccimported: #尝试使用高清音质下载
+                        songid = str(info['id'])
+                        class WangyiyunDownload(object):
+                            def __init__(self):
+                                # "爱心", "女孩", "惊恐", "大笑"的值
+                                # 对应的js --> bqL0x(["爱心", "女孩", "惊恐", "大笑"])
+                                self.key = '0CoJUm6Qyw8W8jud'
+                                # "流泪", "强"的值
+                                # 对应的js --> bqL0x(["流泪", "强"])
+                                self.public_key = "010001"
+                                # 一串表情的值(省略,对应的js --> Yb5g.md)
+                                # 对应的js --> bqL0x(Yb5g.md)
+                                self.modulus = '00e0b509f6259df8642dbc35662901477df22677ec152b5ff68ace615bb7b725152b3ab17a876aea8a5aa76d2e417629ec4ee341f56135fccf695280104e0312ecbda92557c93870114af6c9d05c4f7f0c3685b7a46bee255932575cce10b424d813cfe4875d3e82047b97ddef52741d546b8e289dc6935b3ece0462db0a22b8e7'
+                                # 偏移量
+                                self.iv = "0102030405060708"
+                                # 请求头
+                                self.headers = {
+                                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.97 Safari/537.36',
+                                    # 这里需传入登录cookie,并且必须是会员账户才能访问会员曲目,否则只能访问免费曲目
+                                    'Cookie': 'MUSIC_U=f52f220df171da480dbf33ce89947961585a7fdf08c89a2a4bdd6efebd86544233a649814e309366;',
+                                }
+                                # 请求url
+                                self.url = 'https://music.163.com/weapi/song/enhance/player/url/v1?csrf_token='
+
+                            # 生成16位随机数字符串
+                            # 对应的js --> a函数
+                            def set_random_num(self):
+                                random_num = ''
+                                # 从此字符串随机取出16个字符
+                                string = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+                                for i in range(16):
+                                    # random.uniform(0, 1) * len(string): 生成一个实数,范围在0 <= n < len(string)
+                                    # math.floor(n): 将n向下取整
+                                    n = math.floor(random.uniform(0, 1) * len(string))
+                                    # 从string中取出下标为n的字符拼接到random_num中
+                                    random_num += string[n]
+                                # 返回16位随机数字符串
+                                return random_num
+
+                            # 生成encSecKey
+                            # 对应的js --> c函数
+                            # 通过public_key和modulus对random_num进行RSA加密
+                            # random_num: 生成的16位随机数字符串
+                            def RSA_encrypt(self, random_num):
+                                # 先将16位随机数字符串倒序并以utf-8编码
+                                random_num = random_num[::-1].encode('utf-8')
+                                # 然后再将其以hex(16进制)编码
+                                random_num = codecs.encode(random_num, 'hex_codec')
+                                # 加密(三者均要从16进制转换为10进制)
+                                # int(n, 16) --> 将16进制字符串n转换为10进制
+                                encryption = int(random_num, 16) ** int(self.public_key, 16) % int(self.modulus, 16)
+                                # 将加密后的数据转换为16进制字符串
+                                encryption = format(encryption, 'x')
+                                # 返回加密后的字符串
+                                return encryption
+
+                            # 生成params
+                            # 对应的js --> b函数
+                            # 根据key和iv对msg进行AES加密,需调用两次
+                            # key:
+                            #   第一次: key
+                            #   第二次: random_num
+                            # iv: 偏移量iv
+                            def AES_encrypt(self, msg, key, iv):
+                                # 先将msg按需补全至16的倍数
+                                # 需补全的位数
+                                pad = (16 - len(msg) % 16)
+                                # 补全
+                                msg = msg + pad * chr(pad)
+                                # 这里需要将key,iv和msg均以utf-8编码
+                                key = key.encode('utf-8')
+                                iv = iv.encode('utf-8')
+                                msg = msg.encode('utf-8')
+                                # 根据key和iv生成密钥,模式为CBC模式
+                                encryptor = AES.new(key, AES.MODE_CBC, iv)
+                                # 加密
+                                encrypt_aes = encryptor.encrypt(msg)
+                                # 先将加密后的值进行base64编码
+                                encrypt_text = base64.encodebytes(encrypt_aes)
+                                # 再将其转换为utf-8字符串
+                                encrypt_text = str(encrypt_text, 'utf-8')
+                                # 返回加密后的字符串
+                                return encrypt_text
+
+                            # 根据歌曲song_id,生成需要传输的data
+                            # 其中包括params和encSecKey
+                            def construct_data(self, song_id):
+                                # 先生成16位随机数字符串
+                                random_num = self.set_random_num()
+                                # 生成encSecKey
+                                encSecKey = self.RSA_encrypt(random_num=random_num)
+                                # 调用两次AES加密生成params
+                                # 先初始化歌曲song_info
+                                song_info = '{"ids":"[%s]","level":"exhigh","encodeType":"mp3","csrf_token":"477c1bd99fddedb3adc074f47fee2d35"}' % song_id
+                                # 第一次加密,传入encText, key和iv
+                                first_encryption = self.AES_encrypt(msg=song_info, key=self.key, iv=self.iv)
+                                # 第二次加密, 传入first_encryption, random_num和iv
+                                encText = self.AES_encrypt(msg=first_encryption, key=random_num, iv=self.iv)
+                                # 生成data
+                                data = {
+                                    'params': encText,
+                                    'encSecKey': encSecKey
+                                }
+                                # 返回data
+                                return data
+
+                            # 发送请求,获取下载链接
+                            def get_real_url(self):
+                                global songid
+                                # 输入歌曲song_id
+                                self.song_id = songid
+                                # 获取data
+                                data = self.construct_data(song_id=self.song_id)
+                                # 发送请求
+                                request = requests.post(url=self.url, headers=self.headers, data=data)
+                                # 初始化real_url
+                                real_url = ''
+                                # 处理返回信息
+                                try:
+                                    js_text = json.loads(request.text)
+                                    data = js_text['data']
+                                    if len(data) != 0:
+                                        code = data[0]['code']
+                                        # 获取成功
+                                        if code == 200:
+                                            # 歌曲真实地址
+                                            real_url = data[0]['url']
+                                except:
+                                    print('生成的params和encSecKey有误!可重试!')
+                                # 返回real_url
+                                return real_url
+
+                            def download(self):
+                                global name
+                                # 获取下载链接
+                                real_url = self.get_real_url()
+                                if real_url == '':
+                                    print('链接获取失败!')
+                                else:
+                                    file = name
+                                    # 开始下载
+                                    content = requests.get(url=real_url, headers=self.headers).content
+                                    with open(file, 'wb') as fp:
+                                        fp.write(content)
+                        try:
+                            WangyiyunDownload().download()
+                        except:
+                            pass
+                        if not exists(name):
+                            ccimported = False
+                    
+                    if ccimported is False: # 下载(普通音质)
+                        music = requests.request(
+                            "GET", "http://music.163.com/api/song/enhance/download/url?&br=" + str(info['br'])+ "&id=" + str(info['id']) , headers=headers)
+                        if music.status_code == 200:
+                            music = json.loads(music.content)
+                            if not music['data']['url']:
+                                music = requests.request("GET", "https://music.163.com/song/media/outer/url?id=" + str(info['id']) + ".mp3", headers= headers)
+                                if music.status_code != 200:
+                                    continue
+                            else:
+                                music = requests.request("GET", music['data']['url'] , headers=headers)
+                        else:
+                            continue
+
                     cap = info['artist'].replace(
                         ';', ', ') + " - " + "**" + info['title'] + "**"
-                    pic = requests.get(info['albumpic'])
+                    
+                    if ccimported is False:
                         with open(name, 'wb') as f:
                             f.write(music.content)
                     if (path.getsize(name) / 1024) < 100:
@@ -157,11 +344,13 @@ async def nem(context):
                         await bot.send_message(context.chat_id, f"<strong>【{info['title']}】</strong>\n" + "歌曲获取失败，可能歌曲为VIP专属，或受到地区版权限制。\n" + res, parse_mode='html', link_preview=True)
                         return
                     if imported is True:
+                        imagedata = requests.get(info['albumpic'], headers=headers).content
                         tag = eyed3.load(name).tag
                         tag.artist = info['artist']
                         tag.title = info['title']
                         tag.album = info['album']
-                            tag.images.set(3, pic.content, "image/jpeg", u'')
+                        tag.images.remove('')
+                        tag.images.set(3,imagedata,"image/jpeg",u"you can put a description here")
                         tag.save(
                             version=eyed3.id3.ID3_DEFAULT_VERSION, encoding='utf-8')
                     await context.edit(f"{title}上传中 . . .")