From 93848af285154ba6fe28bc1f431d98cfe5a73b9f Mon Sep 17 00:00:00 2001
From: frxz751113 <156018267+frxz751113@users.noreply.github.com>
Date: Tue, 3 Sep 2024 11:30:33 +0800
Subject: [PATCH] =?UTF-8?q?Update=20=E9=85=92=E5=BA=97=E6=BA=90.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 py/酒店源.py | 33 ++++++++++++++++-----------------
 1 file changed, 16 insertions(+), 17 deletions(-)

diff --git a/py/酒店源.py b/py/酒店源.py
index a879c3b..4da3342 100644
--- a/py/酒店源.py
+++ b/py/酒店源.py
@@ -41,24 +41,23 @@ urls = [
 
 def modify_urls(url):
     modified_urls = []
-    # 找到协议结束的位置，即 "://" 后的部分
-    protocol_end_index = url.find("://") + 3
-    # 找到端口号后面第一个 "/" 的位置
-    slash_after_port_index = url.find("/", protocol_end_index)
-    if slash_after_port_index == -1:
-        # 如果没有找到 "/"，说明 URL 可能不包含路径，我们使用 URL 的结尾
-        slash_after_port_index = len(url)
-    # 截取从协议部分到端口后面的第一个 "/" 之前的部分
-    base_url = url[:slash_after_port_index]
-    # 固定字符
-    fixed_string = "/ZHGXTV/Public/json/live_interface.txt"
-    # 组合成新的 URL
-    modified_url = f"{base_url}{fixed_string}"
-    # 将新的 URL 添加到列表中
-    modified_urls.append(modified_url)
-    # 返回修改后的 URL 列表
+    # 正则表达式匹配协议、IP 地址或域名以及端口，但不包括端口后的斜杠
+    pattern = r'^(https?://[^/:]+(?::\d+)?)[/]?'
+    match = re.match(pattern, url)
+    
+    if match:
+        base_url = match.group(1)
+        # 固定字符
+        fixed_string = "/ZHGXTV/Public/json/live_interface.txt"
+        # 组合成新的 URL
+        modified_url = f"{base_url}{fixed_string}"
+        modified_urls.append(modified_url)
+    else:
+        print("URL format is not recognized.")
+    
     return modified_urls
-
+    for url in modified_urls
+        print(url)