yandex-music-api/yandex_music/utils/sign_request.py

import base64
import datetime
import hashlib
import hmac
from dataclasses import dataclass
from typing import Union

from yandex_music.utils.convert_track_id import convert_track_id_to_number

DEFAULT_SIGN_KEY = 'p93jhgh689SBReK6ghtw62'
""":obj:`str`: Ключ для подписи из Android приложения."""


@dataclass
class Sign:
    """Подпись запроса.

    Attributes:
        timestamp (:obj:`int`): Время создания подписи.
        value (:obj:`str`): Подпись.
    """

    timestamp: int
    value: str


def get_sign_request(track_id: Union[int, str], key: str = DEFAULT_SIGN_KEY) -> Sign:
    """Создает подпись для запроса.

    Args:
        track_id (:obj:`str` | :obj:`int`): Уникальный идентификатора трека.
        key (:obj:`str`, optional): Ключ для подписи.

    Returns:
        :obj:`Sign`: Подпись.
    """
    track_id = convert_track_id_to_number(track_id)

    timestamp = int(datetime.datetime.now().timestamp())
    message = f'{track_id}{timestamp}'

    hmac_sign = hmac.new(key.encode('UTF-8'), message.encode('UTF-8'), hashlib.sha256).digest()
    sign = base64.b64encode(hmac_sign).decode('UTF-8')

    return Sign(timestamp, sign)