qqmusic-api-python 0.1.0__py3-none-any.whl

qqmusic_api/__init__.py

@@ -0,0 +1,4 @@
+from .api import album, mv, login, search, song, songlist, top
+from .utils.credential import Credential
+
+__all__ = ["album", "Credential", "login", "mv", "search", "song", "songlist", "top"]

qqmusic_api/api/album.py

@@ -0,0 +1,40 @@
+from qqmusic_api.api.song import Song
+
+from ..utils.common import get_api
+from ..utils.network import Api
+
+API = get_api("album")
+
+
+class Album:
+    """
+    专辑类
+    """
+
+    def __init__(self, mid: str):
+        """
+        Args:
+            mid: 专辑 mid
+        """
+        self.mid = mid
+
+    async def get_detail(self) -> dict:
+        """
+        Returns:
+            dict: 专辑详细信息
+        """
+        return await Api(**API["detail"]).update_params(albumMid=self.mid).result
+
+    async def get_song(self) -> list[Song]:
+        """
+        获取专辑歌曲
+
+        Returns:
+            list: 歌曲列表
+        """
+        result = (
+            await Api(**API["song"])
+            .update_params(albumMid=self.mid, begin=0, num=0)
+            .result
+        )
+        return [Song.from_dict(song["songInfo"]) for song in result["songList"]]

qqmusic_api/api/login.py

@@ -0,0 +1,437 @@
+import random
+import re
+import time
+from abc import ABC, abstractmethod
+from enum import Enum
+from typing import Optional
+
+from ..exceptions import AuthcodeExpiredException, ResponseCodeException
+from ..utils.common import get_api, hash33, random_uuid
+from ..utils.credential import Credential
+from ..utils.network import Api, get_aiohttp_session
+
+API = get_api("login")
+
+
+class QrCodeLoginEvents(Enum):
+    """
+    二维码登录状态枚举
+
+    + SCAN:    未扫描二维码
+    + CONF:    未确认登录
+    + TIMEOUT: 二维码过期
+    + DONE:    成功
+    + REFUSE:  拒绝登录
+    + OTHER:   未知情况
+    """
+
+    DONE = 0
+    SCAN = 1
+    CONF = 2
+    TIMEOUT = 3
+    REFUSE = 4
+    OTHER = 5
+
+
+class PhoneLoginEvents(Enum):
+    """
+    手机登录状态枚举
+
+    + SEND:    发送成功
+    + CAPTCHA: 需要滑块验证
+    + OTHER:   未知情况
+    """
+
+    SEND = 0
+    CAPTCHA = 1
+    OTHER = 5
+
+
+class Login(ABC):
+    """登录抽象类"""
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.musicid = ""
+        self.auth_url = ""
+        self.state: Optional[QrCodeLoginEvents] = None
+        self.qrcode_data: Optional[bytes] = None
+        self.credential: Optional[Credential] = None
+
+    async def __aenter__(self):
+        self.session = get_aiohttp_session()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.close()
+
+    async def close(self):
+        if not self.session.closed:
+            await self.session.close()
+
+    def initialized(self) -> bool:
+        return bool(
+            self.qrcode_data
+            and self.state not in [QrCodeLoginEvents.TIMEOUT, QrCodeLoginEvents.REFUSE]
+        )
+
+    @abstractmethod
+    async def get_qrcode(self) -> bytes:
+        """
+        获取二维码
+
+        Returns:
+            bytes: 二维码图像数据
+        """
+
+    @abstractmethod
+    async def get_qrcode_state(self) -> QrCodeLoginEvents:
+        """
+        获取二维码状态
+
+        Returns:
+            QrCodeLoginEvents: 二维码状态
+        """
+
+    @abstractmethod
+    async def authorize(self, authcode: Optional[int] = None) -> Credential:
+        """
+        登录鉴权
+
+        Args:
+            code: 验证码. Defaluts to None
+
+        Returns:
+            Credential: 用户凭证
+        """
+
+    @abstractmethod
+    async def send_authcode(self) -> PhoneLoginEvents:
+        """
+        发送验证码
+
+        Returns:
+            PhoneLoginEvents: 操作状态
+        """
+
+
+class QQLogin(Login):
+    """QQ登录"""
+
+    async def send_authcode(self):
+        raise NotImplementedError("不支持")
+
+    async def get_qrcode(self):
+        if self.initialized():
+            return self.qrcode_data
+        async with self.session.get(
+            "https://xui.ptlogin2.qq.com/cgi-bin/xlogin",
+            params={
+                "appid": "716027609",
+                "daid": "383",
+                "style": "33",
+                "login_text": "登录",
+                "hide_title_bar": "1",
+                "hide_border": "1",
+                "target": "self",
+                "s_url": "https://graph.qq.com/oauth2.0/login_jump",
+                "pt_3rd_aid": "100497308",
+                "pt_feedback_link": "https://support.qq.com/products/77942?customInfo=.appid100497308",
+                "theme": "2",
+                "verify_theme": "",
+            },
+        ) as res:
+            self.sig = res.cookies.get("pt_login_sig").value
+        async with self.session.get(
+            "https://ssl.ptlogin2.qq.com/ptqrshow",
+            params={
+                "appid": "716027609",
+                "e": "2",
+                "l": "M",
+                "s": "3",
+                "d": "72",
+                "v": "4",
+                "t": str(random.random()),
+                "daid": "383",
+                "pt_3rd_aid": "100497308",
+            },
+        ) as res:
+            self.ptqrtoken = hash33(res.cookies.get("qrsig").value)
+            self.qrcode_data = await res.read()
+            return self.qrcode_data
+
+    async def get_qrcode_state(self):
+        async with self.session.get(
+            "https://ssl.ptlogin2.qq.com/ptqrlogin",
+            params={
+                "u1": "https://graph.qq.com/oauth2.0/login_jump",
+                "ptqrtoken": self.ptqrtoken,
+                "ptredirect": "0",
+                "h": "1",
+                "t": "1",
+                "g": "1",
+                "from_ui": "1",
+                "ptlang": "2052",
+                "action": "0-0-%s" % int(time.time() * 1000),
+                "js_ver": "20102616",
+                "js_type": "1",
+                "login_sig": self.sig,
+                "pt_uistyle": "40",
+                "aid": "716027609",
+                "daid": "383",
+                "pt_3rd_aid": "100497308",
+                "has_onekey": "1",
+            },
+        ) as res:
+            data = await res.text()
+        text_to_state = {
+            "二维码未失效": QrCodeLoginEvents.SCAN,
+            "二维码认证中": QrCodeLoginEvents.CONF,
+            "二维码已失效": QrCodeLoginEvents.TIMEOUT,
+            "本次登录已被拒绝": QrCodeLoginEvents.REFUSE,
+            "登录成功": QrCodeLoginEvents.DONE,
+        }
+        state = QrCodeLoginEvents.OTHER
+        for text, value in text_to_state.items():
+            if text in data:
+                state = value
+                break
+        self.state = state
+        if state == QrCodeLoginEvents.DONE:
+            self.musicid = re.findall(r"&uin=(.+?)&service", data)[0]
+            self.auth_url = re.findall(r"'(https:.*?)'", data)[0]
+        return state
+
+    async def authorize(self, authcode: Optional[int] = None):
+        if self.credential:
+            return self.credential
+        async with self.session.get(self.auth_url, allow_redirects=False) as res:
+            from http import cookies
+
+            set_cookie_header = res.headers.getall("Set-Cookie", [])
+            for header in set_cookie_header:
+                cookie: cookies.SimpleCookie = cookies.SimpleCookie()
+                cookie.load(header)
+                for key, morsel in cookie.items():
+                    if morsel.value:
+                        self.session.cookie_jar.update_cookies(cookie)
+
+        skey = self.session.cookie_jar.filter_cookies(self.auth_url).get("p_skey").value
+        async with self.session.post(
+            "https://graph.qq.com/oauth2.0/authorize",
+            params={
+                "Referer": "https://graph.qq.com/oauth2.0/show?which=Login&display=pc&response_type=code&client_id"
+                "=100497308&redirect_uri=https://y.qq.com/portal/wx_redirect.html?login_type=1&surl=https"
+                "://y.qq.com/portal/profile.html#stat=y_new.top.user_pic&stat=y_new.top.pop.logout"
+                "&use_customer_cb=0&state=state&display=pc",
+                "Content-Type": "application/x-www-form-urlencoded",
+            },
+            data={
+                "response_type": "code",
+                "client_id": "100497308",
+                "redirect_uri": "https://y.qq.com/portal/wx_redirect.html?login_type=1&surl=https://y.qq.com"
+                "/#&use_customer_cb=0",
+                "scope": "",
+                "state": "state",
+                "switch": "",
+                "from_ptlogin": "1",
+                "src": "1",
+                "update_auth": "1",
+                "openapi": "80901010",
+                "g_tk": hash33(skey, 5381),
+                "auth_time": str(int(time.time())),
+                "ui": random_uuid(),
+            },
+            allow_redirects=False,
+        ) as res:
+            location = res.headers.get("Location", "")
+            code = re.findall(r"(?<=code=)(.+?)(?=&)", location)[0]
+        res = (
+            await Api(**API["QQ_login"])
+            .update_params(code=code)
+            .update_extra_common(tmeLoginType="2")
+            .result
+        )
+        self.credential = Credential.from_cookies(res)
+        return self.credential
+
+
+class WXLogin(Login):
+    """微信登录"""
+
+    def __init__(self):
+        super().__init__()
+        self.uuid = ""
+
+    async def send_authcode(self):
+        raise NotImplementedError("不支持")
+
+    async def get_qrcode(self):
+        if self.initialized():
+            return self.qrcode_data
+        async with self.session.get(
+            "https://open.weixin.qq.com/connect/qrconnect",
+            params={
+                "appid": "wx48db31d50e334801",
+                "redirect_uri": "https://y.qq.com/portal/wx_redirect.html?login_type=2&surl=https://y.qq.com/",
+                "response_type": "code",
+                "scope": "snsapi_login",
+                "state": "STATE",
+                "href": "https://y.qq.com/mediastyle/music_v17/src/css/popup_wechat.css#wechat_redirect",
+            },
+        ) as res:
+            self.uuid = re.findall(r"uuid=(.+?)\"", await res.text())[0]
+        async with self.session.get(
+            f"https://open.weixin.qq.com/connect/qrcode/{self.uuid}",
+            headers={
+                "referer": "https://open.weixin.qq.com/connect/qrconnect?appid=wx48db31d50e334801"
+                "&redirect_uri="
+                "https%3A%2F%2Fy.qq.com%2Fportal%2Fwx_redirect.html%3Flogin_type%3D2%26surl%3Dhttps"
+                "%3A%2F%2Fy.qq.com%2F"
+                "&response_type=code&scope=snsapi_login&state=STATE"
+                "&href=https%3A%2F%2Fy.qq.com%2Fmediastyle%2Fmusic_v17%2Fsrc%2Fcss%2Fpopup_wechat.css"
+                "%23wechat_redirect"
+            },
+        ) as res:
+            self.qrcode_data = await res.read()
+            return self.qrcode_data
+
+    async def get_qrcode_state(self):
+        async with self.session.get(
+            "https://lp.open.weixin.qq.com/connect/l/qrconnect",
+            headers={
+                "referer": "https://open.weixin.qq.com/",
+            },
+            params={
+                "uuid": self.uuid,
+                "_": str(int(round(time.time() * 1000))),
+            },
+        ) as res:
+            data = await res.text()
+
+        text_to_state = {
+            "408": QrCodeLoginEvents.SCAN,
+            "404": QrCodeLoginEvents.CONF,
+            "403": QrCodeLoginEvents.REFUSE,
+            "405": QrCodeLoginEvents.DONE,
+        }
+        state = QrCodeLoginEvents.OTHER
+        for text, value in text_to_state.items():
+            if text in await res.text():
+                state = value
+                break
+        self.state = state
+        if state == QrCodeLoginEvents.DONE:
+            self.musicid = re.findall(r"wx_code='(.+?)';", data)[0]
+            self.auth_url = (
+                "https://y.qq.com/portal/wx_redirect.html?login_type=2"
+                f"&surl=https://y.qq.com/&code={self.musicid}&state=STATE"
+            )
+            await self.session.get(
+                "https://lp.open.weixin.qq.com/connect/l/qrconnect",
+                params={
+                    "uuid": self.uuid,
+                    "_": str(int(round(time.time() * 1000))),
+                    "last": "404",
+                },
+            )
+        return state
+
+    async def authorize(self, authcode: Optional[int] = None):
+        if self.credential:
+            return self.credential
+        await self.session.get(self.auth_url, allow_redirects=False)
+        res = (
+            await Api(**API["WX_login"])
+            .update_params(strAppid="wx48db31d50e334801", code=self.musicid)
+            .update_extra_common(tmeLoginType="1")
+            .result
+        )
+        self.credential = Credential.from_cookies(res)
+        return self.credential
+
+
+class PhoneLogin(Login):
+    """手机号登录"""
+
+    def __init__(self, phone: int):
+        """
+        Args:
+            phone: 手机号码
+        """
+        pattern = re.compile(r"^1[3-9]\d{9}$")
+        if not pattern.match(str(phone)):
+            raise ValueError("非法手机号")
+        self.phone = phone
+
+    async def __aenter__(self):
+        pass
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        pass
+
+    async def close(self):
+        pass
+
+    async def get_qrcode(self):
+        raise NotImplementedError("不支持")
+
+    async def get_qrcode_state(self):
+        raise NotImplementedError("不支持")
+
+    async def send_authcode(self):
+        params = {
+            "tmeAppid": "qqmusic",
+            "phoneNo": str(self.phone),
+            "areaCode": "86",
+        }
+        msg = ""
+        res = await Api(**API["send_authcode"]).update_params(**params).result
+        msg = res["errMsg"]
+        if msg == "OK":
+            return PhoneLoginEvents.SEND
+        elif msg == "robot defense":
+            self.auth_url = res["securityURL"]
+            return PhoneLoginEvents.CAPTCHA
+        else:
+            return PhoneLoginEvents.OTHER
+
+    async def authorize(self, authcode: Optional[int] = None):
+        if not authcode:
+            raise ValueError("authcode 为空")
+        params = {"code": str(authcode), "phoneNo": str(self.phone), "loginMode": 1}
+        try:
+            res = (
+                await Api(**API["phone_login"])
+                .update_params(**params)
+                .update_extra_common(tmeLoginMethod="3")
+                .result
+            )
+            return Credential.from_cookies(res)
+        except ResponseCodeException as e:
+            if e.code == 20271:
+                raise AuthcodeExpiredException()
+
+
+async def refresh_cookies(credential: Credential) -> Credential:
+    """
+    刷新 Cookies
+
+    Args:
+        credential (Credential): 用户凭证
+
+    Return:
+        Credential: 新的用户凭证
+    """
+    params = {
+        "refresh_key": credential.refresh_key,
+        "musicid": credential.musicid,
+        "loginMode": 2,
+    }
+    res = (
+        await Api(**API["refresh"])
+        .update_params(**params)
+        .update_extra_common(tmeLoginType=str(credential.login_type))
+        .result
+    )
+    return Credential.from_cookies(res)

qqmusic_api/api/mv.py

@@ -0,0 +1,115 @@
+import random
+
+from ..utils.common import get_api, random_string
+from ..utils.network import Api
+from .song import query_by_id
+
+API = get_api("mv")
+
+
+class MV:
+    """
+    MV 类
+    """
+
+    def __init__(self, vid: str):
+        self.vid = vid
+
+    async def get_detail(self) -> dict:
+        """
+        获取 MV 详细信息
+
+        Return:
+            dict: 视频信息
+        """
+        param = {
+            "vidlist": [self.vid],
+            "required": [
+                "vid",
+                "type",
+                "sid",
+                "cover_pic",
+                "duration",
+                "singers",
+                "video_switch",
+                "msg",
+                "name",
+                "desc",
+                "playcnt",
+                "pubdate",
+                "isfav",
+                "gmid",
+                "uploader_headurl",
+                "uploader_nick",
+                "uploader_encuin",
+                "uploader_uin",
+                "uploader_hasfollow",
+                "uploader_follower_num",
+                "uploader_hasfollow",
+            ],
+        }
+        return (await Api(**API["detail"]).update_params(**param).result)[self.vid]
+
+    async def get_related_song(self) -> list[dict]:
+        """
+        获取 MV 相关歌曲
+
+        Return:
+            list: 歌曲基本信息
+        """
+        param = {
+            "vidlist": [self.vid],
+            "required": ["related_songs"],
+        }
+        song_id = (await Api(**API["detail"]).update_params(**param).result)[self.vid][
+            "related_songs"
+        ]
+        return await query_by_id(song_id)
+
+    async def get_url(self) -> dict:
+        """
+        获取 MV 播放链接
+
+        Return:
+            dict: 视频播放链接
+        """
+        return (await get_mv_urls([self.vid]))[self.vid]
+
+
+async def get_mv_urls(vid: list[str]) -> dict:
+    """
+    获取 MV 播放链接
+
+    Args:
+        vid: 视频 vid 列表
+
+    Return:
+        dict: 视频播放链接
+    """
+    param = {
+        "vids": vid,
+        "request_type": 10003,
+        "guid": random_string(32, "abcdef1234567890"),
+        "videoformat": 1,
+        "format": 265,
+        "dolby": 1,
+        "use_new_domain": 1,
+        "use_ipv6": 1,
+    }
+    result = await Api(**API["url"]).update_params(**param).result
+    urls: dict[str, dict] = {}
+
+    def get_play_urls(data):
+        play_urls = {}
+        for url_info in data:
+            if url_info["freeflow_url"]:
+                play_url = random.choice(url_info["freeflow_url"])
+                play_urls[url_info["filetype"]] = play_url
+        return play_urls
+
+    for vid, data in result.items():
+        urls[vid] = {}
+        urls[vid]["mp4"] = get_play_urls(data["mp4"])
+        urls[vid]["hls"] = get_play_urls(data["hls"])
+
+    return urls