From cdaa29a56cd387234a633f1e1739e4d514b4ca1b Mon Sep 17 00:00:00 2001
From: Gleb Liutsko <gleb@liutsko.ru>
Date: Sun, 13 Nov 2022 21:19:31 +0400
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=B0=20=D0=B2=D0=BE=D0=B7=D0=BC=D0=BE=D0=B6=D0=BD=D0=BE?=
 =?UTF-8?q?=D1=81=D1=82=D1=8C=20=D0=BF=D0=BE=D0=B4=D0=BF=D0=B8=D1=81=D0=B8?=
 =?UTF-8?q?=20=D0=B7=D0=B0=D0=BF=D1=80=D0=BE=D1=81=D0=BE=D0=B2=20=D0=B8=20?=
 =?UTF-8?q?=D0=BA=D0=BE=D0=BD=D0=B5=D0=B2=D0=B5=D1=80=D1=82=D0=B0=D1=86?=
 =?UTF-8?q?=D0=B8=D0=B8=20=D1=84=D0=BE=D1=80=D0=BC=D0=B0=D1=82=D0=BE=D0=B2?=
 =?UTF-8?q?=20track=5Fid?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Добавлены функции get_sign_request, convert_track_id_to_number
---
 yandex_music/utils/convert_track_id.py |  9 ++++++++
 yandex_music/utils/sign_request.py     | 32 ++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 yandex_music/utils/convert_track_id.py
 create mode 100644 yandex_music/utils/sign_request.py

diff --git a/yandex_music/utils/convert_track_id.py b/yandex_music/utils/convert_track_id.py
new file mode 100644
index 0000000..113e8b4
--- /dev/null
+++ b/yandex_music/utils/convert_track_id.py
@@ -0,0 +1,9 @@
+from typing import Union
+
+
+def convert_track_id_to_number(track_id: Union[str, int]) -> int:
+    """TODO"""
+    if isinstance(track_id, str):
+        track_id = int(track_id.split(':')[0])
+
+    return track_id
diff --git a/yandex_music/utils/sign_request.py b/yandex_music/utils/sign_request.py
new file mode 100644
index 0000000..42ab75f
--- /dev/null
+++ b/yandex_music/utils/sign_request.py
@@ -0,0 +1,32 @@
+from typing import Union
+from dataclasses import dataclass
+import datetime
+import hmac
+import hashlib
+import base64
+
+from yandex_music.utils.convert_track_id import convert_track_id_to_number
+
+
+DEFAULT_SIGN_KEY = 'p93jhgh689SBReK6ghtw62'
+
+
+@dataclass
+class Sign:
+    """TODO"""
+
+    timestamp: int
+    value: str
+
+
+def get_sign_request(track_id: Union[int, str], key: str = DEFAULT_SIGN_KEY) -> Sign:
+    """TODO"""
+    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)