From: Đặng Ngọc Quảng <quanghd96@gmail.com>
Date: Mon, 15 Aug 2022 23:33:36 +0000 (+0700)
Subject: Fix Youtube Music support for some countries (#454)
X-Git-Url: https://git.kitaultman.com/?a=commitdiff_plain;h=4cdcc64ed9cbf05032380919cf3a9987b301b962;p=music-assistant-server.git

Fix Youtube Music support for some countries (#454)

* support some country that not supported youtube music
---

diff --git a/music_assistant/music_providers/ytmusic/ytmusic.py b/music_assistant/music_providers/ytmusic/ytmusic.py
index 960cbd09..9e2b1c74 100644
--- a/music_assistant/music_providers/ytmusic/ytmusic.py
+++ b/music_assistant/music_providers/ytmusic/ytmusic.py
@@ -50,6 +50,7 @@ from music_assistant.music_providers.ytmusic.helpers import (
     search,
 )
 
+YT_DOMAIN = "https://www.youtube.com"
 YTM_DOMAIN = "https://music.youtube.com"
 YTM_BASE_URL = f"{YTM_DOMAIN}/youtubei/v1/"
 
@@ -612,6 +613,10 @@ class YoutubeMusicProvider(MusicProvider):
         """Get a signature timestamp required to generate valid stream URLs."""
         response = await self._get_data(url=YTM_DOMAIN)
         match = re.search(r'jsUrl"\s*:\s*"([^"]+)"', response)
+        if match is None:
+            # retry with youtube domain
+            response = await self._get_data(url=YT_DOMAIN)
+            match = re.search(r'jsUrl"\s*:\s*"([^"]+)"', response)
         if match is None:
             raise Exception("Could not identify the URL for base.js player.")
         url = YTM_DOMAIN + match.group(1)