Skip to content

Commit

Permalink
feat: 支持kg源歌词获取
Browse files Browse the repository at this point in the history
  • Loading branch information
helloplhm-qwq committed Dec 31, 2023
1 parent 18fd6a4 commit 0d21d71
Show file tree
Hide file tree
Showing 5 changed files with 197 additions and 6 deletions.
29 changes: 28 additions & 1 deletion common/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
import time
import re
import xmltodict
from urllib.parse import quote
from urllib.parse import quote, unquote, urlparse
from hashlib import md5 as handleCreateMD5

def createBase64Encode(data_bytes):
Expand Down Expand Up @@ -88,8 +88,35 @@ def unique_list(list_in):
return unique_list

def encodeURIComponent(component):
if (isinstance(component, str)):
component = component.encode('utf-8')
elif (not isinstance(component, bytes)):
raise TypeError('component must be str or bytes')
return quote(component)

def decodeURIComponent(component):
return unquote(component)

def encodeURI(uri):
parse_result = urlparse(uri)
params = {}
for q in parse_result.query.split('&'):
k, v = q.split('=')
v = encodeURIComponent(v)
params[k] = v
query = '&'.join([f'{k}={v}' for k, v in params.items()])
return parse_result._replace(query=query).geturl()

def decodeURI(uri):
parse_result = urlparse(uri)
params = {}
for q in parse_result.query.split('&'):
k, v = q.split('=')
v = decodeURIComponent(v)
params[k] = v
query = '&'.join([f'{k}={v}' for k, v in params.items()])
return parse_result._replace(query=query).geturl()

def sortDict(dictionary):
sorted_items = sorted(dictionary.items())
sorted_dict = {k: v for k, v in sorted_items}
Expand Down
39 changes: 38 additions & 1 deletion modules/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ async def url(source, songId, quality):
}
try:
result = await func(songId, quality)
logger.debug(f'获取{source}_{songId}_{quality}成功,URL:{result["url"]}')
logger.info(f'获取{source}_{songId}_{quality}成功,URL:{result["url"]}')

canExpire = sourceExpirationTime[source]['expire']
expireTime = sourceExpirationTime[source]['time'] + int(time.time())
Expand Down Expand Up @@ -116,6 +116,43 @@ async def url(source, songId, quality):
},
},
}
except FailedException as e:
logger.info(f'获取{source}_{songId}_{quality}失败,原因:' + e.args[0])
return {
'code': 2,
'msg': e.args[0],
'data': None,
}

async def lyric(source, songId, _):
cache = config.getCache('lyric', f'{source}_{songId}')
if cache:
return {
'code': 0,
'msg': 'success',
'data': cache['data']
}
try:
func = require('modules.' + source + '.lyric')
except:
return {
'code': 1,
'msg': '未知的源或不支持的方法',
'data': None,
}
try:
result = await func(songId)
config.updateCache('lyric', f'{source}_{songId}', {
"data": result,
"time": int(time.time() + (86400 * 3)), # 歌词缓存3天
"expire": True,
})
logger.debug(f'缓存已更新:{source}_{songId}, lyric: {result}')
return {
'code': 0,
'msg': 'success',
'data': result
}
except FailedException as e:
return {
'code': 2,
Expand Down
9 changes: 8 additions & 1 deletion modules/kg/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@
from .musicInfo import getMusicInfo as _getInfo
from .utils import tools
from .player import url
from .lyric import getLyric as _getLyric
from .lyric import lyricSearchByHash as _lyricSearch
from .mv import getMvInfo as _getMvInfo
from .mv import getMvPlayURL as _getMvUrl
from common.exceptions import FailedException
Expand Down Expand Up @@ -68,4 +70,9 @@ async def mv(hash_):
res1 = res[0]
res2 = res[1]
res1['play_info'] = res2
return res1
return res1

async def lyric(hash_):
lyric_search_result = await _lyricSearch(hash_)
choosed_lyric = lyric_search_result[0]
return await _getLyric(choosed_lyric['id'], choosed_lyric['accesskey'])
123 changes: 123 additions & 0 deletions modules/kg/lyric.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,123 @@
# ----------------------------------------
# - mode: python -
# - author: helloplhm-qwq -
# - name: lyric.py -
# - project: lx-music-api-server -
# - license: MIT -
# ----------------------------------------
# This file is part of the "lx-music-api-server" project.

from common.exceptions import FailedException
from common.utils import encodeURI, createBase64Decode
from .musicInfo import getMusicInfo
from common import Httpx
import ujson as json
import zlib
import re

class ParseTools:
def __init__(self):
self.head_exp = r'^.*\[id:\$\w+\]\n'

def parse(self, string):
string = string.replace('\r', '')
if re.match(self.head_exp, string):
string = re.sub(self.head_exp, '', string)
trans = re.search(r'\[language:([\w=\\/+]+)\]', string)
lyric = None
rlyric = None
tlyric = None
if trans:
string = re.sub(r'\[language:[\w=\\/+]+\]\n', '', string)
decoded_trans = createBase64Decode(trans.group(1)).decode('utf-8')
trans_json = json.loads(decoded_trans)
for item in trans_json['content']:
if item['type'] == 0:
rlyric = item['lyricContent']
elif item['type'] == 1:
tlyric = item['lyricContent']
self.i = 0
lxlyric = re.sub(r'\[((\d+),\d+)\].*', lambda x: self.process_lyric_match(x, rlyric, tlyric, self.i), string)
rlyric = '\n'.join(rlyric) if rlyric else ''
tlyric = '\n'.join(tlyric) if tlyric else ''
lxlyric = re.sub(r'<(\d+,\d+),\d+>', r'<\1>', lxlyric)
lyric = re.sub(r'<\d+,\d+>', '', lxlyric)
return {
'lyric': lyric,
'tlyric': tlyric,
'rlyric': rlyric,
'lxlyric': lxlyric
}

def process_lyric_match(self, match, rlyric, tlyric, i):
result = re.match(r'\[((\d+),\d+)\].*', match.group(0))
time = int(result.group(2))
ms = time % 1000
time /= 1000
m = str(int(time / 60)).zfill(2)
time %= 60
s = str(int(time)).zfill(2)
time_string = f'{m}:{s}.{ms}'
transformed_t = ''
if (tlyric):
for t in tlyric[i]:
transformed_t += t
tlyric[i] = transformed_t
if (rlyric):
nr = []
for r in rlyric[i]:
nr.append(r)
_tnr = ''.join(nr)
if (' ' in _tnr):
rlyric[i] = _tnr
else:
nr = []
for r in rlyric[i]:
nr.append(r.strip())
rlyric[i] = ' '.join(nr)
if rlyric:
rlyric[i] = f'[{time_string}]{rlyric[i] if rlyric[i] else ""}'.replace(' ', ' ')
if tlyric:
tlyric[i] = f'[{time_string}]{tlyric[i] if tlyric[i] else ""}'
self.i += 1
return re.sub(result.group(1), time_string, match.group(0))

global_parser = ParseTools()

def krcDecode(a:bytes):
encrypt_key = (64, 71, 97, 119, 94, 50, 116, 71, 81, 54, 49, 45, 206, 210, 110, 105)
content = a[4:] # krc1
compress_content = bytes(content[i] ^ encrypt_key[i % len(encrypt_key)] for i in range(len(content)))
text_bytes = zlib.decompress(bytes(compress_content))
text = text_bytes.decode("utf-8")
return text

async def lyricSearchByHash(hash_):
musicInfo = await getMusicInfo(hash_)
if (not musicInfo):
raise FailedException('歌曲信息获取失败')
hash_new = musicInfo['audio_info']['hash']
name = musicInfo['songname']
timelength = int(musicInfo['audio_info']['timelength']) // 1000
req = await Httpx.AsyncRequest(encodeURI(f'https://lyrics.kugou.com/search?ver=1&man=yes&client=pc&keyword=' +
name + '&hash=' + hash_new + '&timelength=' + str(timelength)), {
'method': 'GET',
})
body = req.json()
if (body['status'] != 200):
raise FailedException('歌词获取失败')
if (not body['candidates']):
raise FailedException('歌词获取失败: 当前歌曲无歌词')
return body['candidates']

async def getLyric(lyric_id, accesskey):
req = await Httpx.AsyncRequest(f'https://lyrics.kugou.com/download?ver=1&client=pc&id={lyric_id}&accesskey={accesskey}', {
'method': 'GET',
})
body = req.json()
if (body['status'] != 200 or body['error_code'] != 0 or (not body['content'])):
raise FailedException('歌词获取失败')
content = createBase64Decode(body['content'])
content = krcDecode(content)

return global_parser.parse(content)
3 changes: 0 additions & 3 deletions modules/tx/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,10 @@
# ----------------------------------------
# This file is part of the "lx-music-api-server" project.

from .player import url
from .musicInfo import getMusicInfo as _getInfo
from .utils import formatSinger
from .lyric import getLyric as _getLyric
from common import utils
from . import refresh_login


async def info(songid):
req = await _getInfo(songid)
Expand Down

0 comments on commit 0d21d71

Please sign in to comment.