first_commit .gitignore

plugins/songpicker2/__init__.py

@@ -0,0 +1,81 @@
+from .data_source import dataGet, dataProcess
+from nonebot.adapters import Bot, Event
+from nonebot.typing import T_State
+from nonebot import on_command
+
+
+dataget = dataGet()
+
+songpicker = on_command("点歌", priority=5, block=True)
+
+
+@songpicker.handle()
+async def handle_first_receive(bot: Bot, event: Event, state: T_State):
+    args = str(event.get_message()).strip()
+    # if args.isdigit():
+    #     if "songName" in state:
+    #         state["songNum"] = int(args)
+    if args:
+        state["songName"] = args
+
+
+@songpicker.got("songName", prompt="歌名是？")
+async def handle_songName(bot: Bot, event: Event, state: T_State):
+    songName = state["songName"]
+    songIdList = await dataget.songIds(songName=songName)
+    if not songIdList:
+        await songpicker.reject("没有找到这首歌，请发送其它歌名！")
+    songInfoList = list()
+    for songId in songIdList:
+        songInfoDict = await dataget.songInfo(songId)
+        songInfoList.append(songInfoDict)
+    # songInfoMessage = await dataProcess.mergeSongInfo(songInfoList)
+    # await songpicker.send(songInfoMessage)
+    state["songIdList"] = songIdList
+
+
+@songpicker.got("songName")
+async def handle_songNum(bot: Bot, event: Event, state: T_State):
+    songIdList = state["songIdList"]
+    # songNum = state["songNum"]
+    songNum = 0
+    # 处理重选
+    # if not songNum.isdigit():
+    #     await songpicker.finish()
+    # else:
+    #     songNum = int(songNum)
+    #
+    # if songNum >= len(songIdList) or songNum < 0:
+    #     await songpicker.reject("数字序号错误")
+
+    selectedSongId = songIdList[int(songNum)]
+
+    songContent = [
+        {
+            "type": "music",
+            "data": {
+                "type": 163,
+                "id": selectedSongId
+            }
+        }
+    ]
+    await songpicker.send(songContent)
+
+    # songCommentsDict = await dataget.songComments(songId=selectedSongId)
+    # songCommentsMessage = await dataProcess.mergeSongComments(songCommentsDict)
+    # commentContent = [
+    #     {
+    #         "type": "text",
+    #         "data": {
+    #             "text": "下面为您播送热评：\n"
+    #         }
+    #     },
+    #     {
+    #         "type": "text",
+    #         "data": {
+    #             "text": songCommentsMessage
+    #         }
+    #     }
+    # ]
+    #
+    # await songpicker.send(commentContent)

plugins/songpicker2/data_source.py

@@ -0,0 +1,156 @@
+import aiohttp
+import json
+
+
+class dataApi():
+    '''
+    从网易云音乐接口直接获取数据（实验性）
+    '''
+    headers = {"referer": "http://music.163.com"}
+    cookies = {"appver": "2.0.2"}
+
+    async def search(self, songName: str):
+        '''
+        搜索接口，用于由歌曲名查找id
+        '''
+        async with aiohttp.ClientSession(headers=self.headers, cookies=self.cookies) as session:
+            async with session.post(f"http://music.163.com/api/search/get/", data={"s": songName, "limit": 5, "type": 1, "offset": 0},) as r:
+                if r.status != 200:
+                    return None
+                r = await r.text()
+                return json.loads(r)
+
+    # async def getHotComments(self, songId: int):
+    #     '''
+    #     获取热评
+    #     '''
+    #     async with httpx.AsyncClient() as client:
+    #         async with aiohttp.ClientSession(headers=self.headers, cookies=self.cookies) as session:
+    #             async with session.post(
+    #                     f"https://music.163.com/weapi/v1/resource/hotcomments/R_SO_4_{songId}?csrf_token=",
+    #                     data={
+    #                         "params": 'D33zyir4L/58v1qGPcIPjSee79KCzxBIBy507IYDB8EL7jEnp41aDIqpHBhowfQ6iT1Xoka8jD+0p
+    #                         44nRKNKUA0dv+n5RWPOO57dZLVrd+T1J/sNrTdzUhdHhoKRIgegVcXYjYu+CshdtCBe6WEJozBRlaHyLeJtGrA
+    #                         BfMOEb4PqgI3h/uELC82S05NtewlbLZ3TOR/TIIhNV6hVTtqHDVHjkekrvEmJzT5pk1UY6r0=',
+    #                         "encSecKey": '45c8bcb07e69c6b545d3045559bd300db897509b8720ee2b45a72bf2d3b216ddc77fb10dae
+    #                         c4ca54b466f2da1ffac1e67e245fea9d842589dc402b92b262d3495b12165a721aed880bf09a0a99ff94c959d
+    #                         04e49085dc21c78bbbe8e3331827c0ef0035519e89f097511065643120cbc478f9c0af96400ba4649265781fc9079'
+    #                     },
+    #                                     ) as r:
+    #                 if r.status != 200:
+    #                     return None
+    #                 r = await r.json()
+    #                 return r
+
+    async def getSongInfo(self, songId: int):
+        '''
+        获取歌曲信息
+        '''
+        async with aiohttp.ClientSession(headers=self.headers, cookies=self.cookies) as session:
+            async with session.post(f"http://music.163.com/api/song/detail/?id={songId}&ids=%5B{songId}%5D",) as r:
+                if r.status != 200:
+                    return None
+                r = await r.text()
+                return json.loads(r)
+
+
+class dataGet(dataApi):
+    '''
+    从dataApi获取数据，并做简单处理
+    '''
+
+    api = dataApi()
+
+    async def songIds(self, songName: str, amount=5) -> list:
+        '''
+        根据用户输入的songName 获取候选songId列表 [默认songId数量：5]
+        '''
+        songIds = list()
+        r = await self.api.search(songName=songName)
+        if r is None:
+            raise WrongDataError
+        idRange = amount if amount < len(
+            r["result"]["songs"]) else len(r["result"]["songs"])
+        for i in range(idRange):
+            songIds.append(r["result"]["songs"][i]["id"])
+        return songIds
+
+    # async def songComments(self, songId: int, amount=3) -> dict:
+    #     '''
+    #     根据传递的单一songId，获取songComments dict [默认评论数量上限：3]
+    #     '''
+    #     songComments = dict()
+    #     r = await self.api.getHotComments(songId)
+    #     if r is None:
+    #         raise WrongDataError
+    #     commentsRange = amount if amount < len(
+    #         r['hotComments']) else len(r['hotComments'])
+    #     for i in range(commentsRange):
+    #         songComments[r['hotComments'][i]['user']
+    #         ['nickname']] = r['hotComments'][i]['content']
+    #     return songComments
+
+    async def songInfo(self, songId: int) -> dict:
+        '''
+        根据传递的songId，获取歌曲名、歌手、专辑等信息，作为dict返回
+        '''
+        songInfo = dict()
+        r = await self.api.getSongInfo(songId)
+        if r is None:
+            raise WrongDataError
+        songInfo["songName"] = r["songs"][0]["name"]
+
+        songArtists = list()
+        for ars in r["songs"][0]["artists"]:
+            songArtists.append(ars["name"])
+        songArtistsStr = "、".join(songArtists)
+        songInfo["songArtists"] = songArtistsStr
+
+        songInfo["songAlbum"] = r["songs"][0]["album"]["name"]
+
+        return songInfo
+
+
+class dataProcess():
+    '''
+    将获取的数据处理为用户能看懂的形式
+    '''
+
+    @staticmethod
+    async def mergeSongInfo(songInfos: list) -> str:
+        '''
+        将歌曲信息list处理为字符串，供用户点歌
+        传递进的歌曲信息list含有多个歌曲信息dict
+        '''
+        songInfoMessage = "请输入欲点播歌曲的序号：\n"
+        numId = 0
+        for songInfo in songInfos:
+            songInfoMessage += f"{numId}："
+            songInfoMessage += songInfo["songName"]
+            songInfoMessage += "-"
+            songInfoMessage += songInfo["songArtists"]
+            songInfoMessage += " 专辑："
+            songInfoMessage += songInfo["songAlbum"]
+            songInfoMessage += "\n"
+            numId += 1
+        return songInfoMessage
+
+    @staticmethod
+    async def mergeSongComments(songComments: dict) -> str:
+        songCommentsMessage = '\n'.join(
+            ['%s： %s' % (key, value) for (key, value) in songComments.items()])
+        return songCommentsMessage
+
+
+class Error(Exception):
+    '''
+    谁知道网易的接口会出什么幺蛾子
+    '''
+    pass
+
+
+class WrongDataError(Error):
+    def __init__(self, expression, message):
+        self.expression = expression
+        self.message = message
+        self.message += "\n未从网易接口获取到有效的数据！"