From 1a16c62638033db24d0446a9e2f88293192c9563 Mon Sep 17 00:00:00 2001
From: ChocoLZS <chocoielzs@gmail.com>
Date: Thu, 22 Aug 2024 19:51:05 +0800
Subject: [PATCH] chore: rename regex and add referer

---
 yt_dlp/extractor/pialive.py        | 13 ++++---------
 yt_dlp/extractor/piaulizaportal.py |  4 ++--
 2 files changed, 6 insertions(+), 11 deletions(-)

diff --git a/yt_dlp/extractor/pialive.py b/yt_dlp/extractor/pialive.py
index 9ea1b22432..a6fc8ffffc 100644
--- a/yt_dlp/extractor/pialive.py
+++ b/yt_dlp/extractor/pialive.py
@@ -36,10 +36,10 @@ class PiaLiveIE(InfoExtractor):
     ]
 
     def handle_embed_player(self, player_tag, video_id, info_dict={}):
-        player_data_url = self._search_regex([PIAULIZAPortalAPIIE.TAG_REGEX_PATTERN],
+        player_data_url = self._search_regex([PIAULIZAPortalAPIIE.TAG_REGEX],
                                              player_tag, 'player data url', fatal=False)
 
-        if PIAULIZAPortalAPIIE.BASE_URL in player_data_url:
+        if player_data_url.startswith(PIAULIZAPortalAPIIE.BASE_URL):
             return self.url_result(
                 smuggle_url(
                     player_data_url,
@@ -57,8 +57,7 @@ class PiaLiveIE(InfoExtractor):
 
         prod_configure = self._download_webpage(
             self.PLAYER_ROOT_URL + self._search_regex(r'<script [^>]*\bsrc="(/statics/js/s_prod[^"]+)"', webpage, 'prod configure page'),
-            program_code,
-            headers={'Referer': self.PLAYER_ROOT_URL},
+            program_code, headers={'Referer': self.PLAYER_ROOT_URL},
             note='Fetching prod configure page', errnote='Unable to fetch prod configure page',
         )
 
@@ -69,11 +68,7 @@ class PiaLiveIE(InfoExtractor):
         })
         player_tag_list = self._download_json(
             f'{self.PIA_LIVE_API_URL}/perf/player-tag-list/{program_code}',
-            program_code,
-            data=payload,
-            headers={
-                'Content-Type': content_type,
-            },
+            program_code, data=payload, headers={'Content-Type': content_type, 'Referer': self.PLAYER_ROOT_URL},
         )
 
         return self.handle_embed_player(
diff --git a/yt_dlp/extractor/piaulizaportal.py b/yt_dlp/extractor/piaulizaportal.py
index 615392347a..0be13bc70c 100644
--- a/yt_dlp/extractor/piaulizaportal.py
+++ b/yt_dlp/extractor/piaulizaportal.py
@@ -5,7 +5,7 @@ from ..utils import ExtractorError, int_or_none, parse_qs, smuggle_url, time_sec
 class PIAULIZAPortalAPIIE(InfoExtractor):
     IE_DESC = 'https://player-api.p.uliza.jp - PIA ULIZA m3u8'
     BASE_URL = 'https://player-api.p.uliza.jp'
-    TAG_REGEX_PATTERN = r'<script [^>]*\bsrc="(https://player-api\.p\.uliza\.jp/v1/players/[^"]+)"'
+    TAG_REGEX = r'<script [^>]*\bsrc="(https://player-api\.p\.uliza\.jp/v1/players/[^"]+)"'
 
     _VALID_URL = r'https://player-api\.p\.uliza\.jp/v1/players/(?P<id>.*)'
     _TESTS = [
@@ -117,7 +117,7 @@ class PIAULIZAPortalIE(InfoExtractor):
         webpage = self._download_webpage(url, video_id)
 
         player_data_url = self._search_regex(
-            PIAULIZAPortalAPIIE.TAG_REGEX_PATTERN,
+            PIAULIZAPortalAPIIE.TAG_REGEX,
             webpage, 'player data url')
         return self.url_result(
             smuggle_url(