import re from typing import Any from httpx import AsyncClient from ..post import Post from .platform import NewMessage from ..utils import SchedulerConfig from ..types import Target, RawPost, ApiError class NcmSchedConf(SchedulerConfig): name = "music.163.com" schedule_type = "interval" schedule_setting = {"minutes": 1} class NcmArtist(NewMessage): categories = {} platform_name = "ncm-artist" enable_tag = False enabled = True is_common = True scheduler = NcmSchedConf name = "网易云-歌手" has_target = True parse_target_promot = "请输入歌手主页(包含数字ID)的链接" @classmethod async def get_target_name(cls, client: AsyncClient, target: Target) -> str | None: res = await client.get( f"https://music.163.com/api/artist/albums/{target}", headers={"Referer": "https://music.163.com/"}, ) res_data = res.json() if res_data["code"] != 200: raise ApiError(res.request.url) return res_data["artist"]["name"] @classmethod async def parse_target(cls, target_text: str) -> Target: if re.match(r"^\d+$", target_text): return Target(target_text) elif match := re.match(r"(?:https?://)?music\.163\.com/#/artist\?id=(\d+)", target_text): return Target(match.group(1)) else: raise cls.ParseTargetException() async def get_sub_list(self, target: Target) -> list[RawPost]: res = await self.client.get( f"https://music.163.com/api/artist/albums/{target}", headers={"Referer": "https://music.163.com/"}, ) res_data = res.json() if res_data["code"] != 200: return [] else: return res_data["hotAlbums"] def get_id(self, post: RawPost) -> Any: return post["id"] def get_date(self, post: RawPost) -> int: return post["publishTime"] // 1000 async def parse(self, raw_post: RawPost) -> Post: text = "新专辑发布:{}".format(raw_post["name"]) target_name = raw_post["artist"]["name"] pics = [raw_post["picUrl"]] url = "https://music.163.com/#/album?id={}".format(raw_post["id"]) return Post(self, text, url=url, images=pics, nickname=target_name) class NcmRadio(NewMessage): categories = {} platform_name = "ncm-radio" enable_tag = False enabled = True is_common = False scheduler = NcmSchedConf name = "网易云-电台" has_target = True parse_target_promot = "请输入主播电台主页(包含数字ID)的链接" @classmethod async def get_target_name(cls, client: AsyncClient, target: Target) -> str | None: res = await client.post( "http://music.163.com/api/dj/program/byradio", headers={"Referer": "https://music.163.com/"}, data={"radioId": target, "limit": 1000, "offset": 0}, ) res_data = res.json() if res_data["code"] != 200 or res_data["programs"] == 0: return return res_data["programs"][0]["radio"]["name"] @classmethod async def parse_target(cls, target_text: str) -> Target: if re.match(r"^\d+$", target_text): return Target(target_text) elif match := re.match(r"(?:https?://)?music\.163\.com/#/djradio\?id=(\d+)", target_text): return Target(match.group(1)) else: raise cls.ParseTargetException() async def get_sub_list(self, target: Target) -> list[RawPost]: res = await self.client.post( "http://music.163.com/api/dj/program/byradio", headers={"Referer": "https://music.163.com/"}, data={"radioId": target, "limit": 1000, "offset": 0}, ) res_data = res.json() if res_data["code"] != 200: return [] else: return res_data["programs"] def get_id(self, post: RawPost) -> Any: return post["id"] def get_date(self, post: RawPost) -> int: return post["createTime"] // 1000 async def parse(self, raw_post: RawPost) -> Post: text = "网易云电台更新:{}".format(raw_post["name"]) target_name = raw_post["radio"]["name"] pics = [raw_post["coverUrl"]] url = "https://music.163.com/#/program/{}".format(raw_post["id"]) return Post(self, text, url=url, images=pics, nickname=target_name)