添加了自动替换久游链接的方法

src/change_url/change_url.py

@@ -1,5 +1,9 @@
 import re
 
+hth_keywords = ["华体会", "hth", "HTH"]
+ky_keywords = ["开云"]
+jy_keywords = ["九游"]
+
 
 def read_file(filename):
     try:
@@ -26,11 +30,9 @@ def replace_urls(js_content, url_mapping):
 
 
 def get_url_mapping(urls_with_context, is_seo, template_type):
-    categories = {}
-
     # 初始化categories字典，区分SEO和非SEO模式
-    if is_seo or template_type == 'ky':
-        categories = {"WEB": [], "H5": [], "全站": [], "体育": []}
+    if is_seo or template_type == 'ky' or template_type == 'jy':
+        categories = {"WEB": [], "H5": [], "全站": [], "体育": [], "棋牌": [], "电子": []}
     else:  # 非SEO模式下的hth模板
         categories = {"web": [], "h5": [], "全站APP": [], "体育APP": []}
 
@@ -48,46 +50,65 @@ def get_url_mapping(urls_with_context, is_seo, template_type):
             "{{hthPc}}": categories.get("WEB", [""])[0] if is_seo else categories.get("web", [""])[0],
             "{{hthH5}}": categories.get("H5", [""])[0] if is_seo else categories.get("h5", [""])[0]
         }
-    else:  # ky模板
+    elif template_type == 'ky':  # ky模板
         url_mapping = {
             "{{kyPc1}}": categories.get("WEB", [""])[0],
             "{{kyPc2}}": categories.get("WEB", ["", ""])[1],
             "{{kyH51}}": categories.get("H5", [""])[0],
             "{{kyH52}}": categories.get("H5", ["", ""])[1],
             "{{kyApp1}}": categories.get("全站", [""])[0],
-            "{{kyApp2}}": categories.get("全站", ["", ""])[1],
+            "{{kyApp2}}": categories.get("体育", [""])[0],
+            # "{{kyApp2}}": categories.get("全站", ["", ""])[1],
             "{{kyTy1}}": categories.get("体育", [""])[0],
-            "{{kyTy2}}": categories.get("体育", ["", ""])[1]
+            "{{kyTy2}}": categories.get("全站", [""])[0],
+            # "{{kyTy2}}": categories.get("体育", ["", ""])[1]
         }
+    elif template_type == 'jy':  # jy模板
+        url_mapping = {
+            "{{jyPc}}": categories.get("WEB", [""])[0],
+            "{{jyH5}}": categories.get("H5", [""])[0],
+            "{{jyApp}}": categories.get("全站", [""])[0],
+            "{{jyQp}}": categories.get("棋牌", [""])[0],
+            "{{jyDz}}": categories.get("电子", [""])[0],
+            "{{jyFljPc}}": categories.get("WEB", [""])[1],
+            "{{jyFljH5}}": categories.get("H5", [""])[1],
+        }
+    else:
+        print('未知模板')
+        return
 
     return url_mapping
 
 
 def extract_urls(text):
-    seo_part_index = text.find("SEO 防拦截域名")
-    relevant_text = text[:seo_part_index] if seo_part_index != -1 else text
+    # seo_part_index = text.find("SEO 防拦截域名")
+    # relevant_text = text[:seo_part_index] if seo_part_index != -1 else text
     pattern = r'(?:\n|^)\s*([^\n]*?)\s*(https?://[^\s]+)'
-    matches = re.findall(pattern, relevant_text)
+    matches = re.findall(pattern, text)
     return [(context.strip(), url) for context, url in matches]
 
 
 def change_url(text, hth_template_path='/www/wwwroot/change_url/hth.js.template',
-               ky_template_path='/www/wwwroot/change_url/ky.js.template'):
+               ky_template_path='/www/wwwroot/change_url/ky.js.template',
+               jy_template_path='/www/wwwroot/change_url/jy.js.template'):
     try:
         urls_with_context = extract_urls(text)
         expected_filename = None
-
-        if '代理' in text:
+        # 如果字符中有代理，则带后台链接
+        if '代理WEB' in text or '代理web' in text:
             is_seo = False
         else:
             is_seo = True
 
-        if len(urls_with_context) in [4, 6]:
+        if any(keyword in text for keyword in hth_keywords) or not is_seo:
             template_path = hth_template_path
             template_type = 'hth'
-        elif len(urls_with_context) == 8:
+        elif any(keyword in text for keyword in ky_keywords):
             template_path = ky_template_path
             template_type = 'ky'
+        elif any(keyword in text for keyword in jy_keywords):
+            template_path = jy_template_path
+            template_type = 'jy'
         else:
             return False, 'unknown.js'
 
@@ -101,3 +122,17 @@ def change_url(text, hth_template_path='/www/wwwroot/change_url/hth.js.template'
     except Exception as e:
         print(f"An error occurred: {e}")
         return False, expected_filename if expected_filename else 'unknown.js'
+
+if __name__ == '__main__':
+    text = '''SEO— 代理 专用域名
+
+WEB     https://www.04qm83.vip:6004  
+H5       https://www.17wq4x.vip:9520   【新】 
+全站    https://www.abucqv.com:9519 【新】 
+棋牌    https://www.2ghc85.vip:8004   【新】
+电子    https://www.3mczux.vip:6001  【新】
+
+九游娱乐【SEO防拦截域名】
+WEB     https://www.9ji5rw.com:9519  
+H5        https://www.9myqr9.com:9502'''
+    change_url(text,hth_template_path='template/hth.js.template',ky_template_path='template/ky.js.template',jy_template_path='template/jy.js.template')

src/change_url/template/hth.js.template

@@ -0,0 +1,12 @@
+const hthCode = '3016341'
+
+var hth_link = {
+        hthApp: '{{hthApp}}/?i_code='+hthCode,
+        hthtyApp: '{{hthtyApp}}/?i_code='+hthCode,
+        hthPc: '{{hthPc}}/register/?i_code='+hthCode,
+        hthH5: '{{hthH5}}/entry/register?i_code='+hthCode,
+}
+
+function visit_hth(key) {
+    window['open'](hth_link[key]);
+}

src/change_url/template/jy.js.template

@@ -0,0 +1,16 @@
+const jyCode = '90413847'
+
+var jy_link = {
+    jyPc: '{{jyPc}}/register/?i_code='+jyCode,
+    jyH5: '{{jyH5}}/entry/register?i_code='+jyCode,
+    jyApp: '{{jyApp}}/?i_code='+jyCode,
+    jyQp:'{{jyQp}}/?i_code='+jyCode,
+    jyDz:'{{jyDz}}/?i_code='+jyCode,
+    jyFljPc:'{{jyFljPc}}/?i_code='+jyCode,
+    jyFljH5:'{{jyFljH5}}/?i_code='+jyCode,
+}
+
+
+function visit_jy(key) {
+    window['open'](jy_link[key]);
+}

src/change_url/template/ky.js.template

@@ -0,0 +1,15 @@
+const kyCode = '97238304'
+
+var ky_link = {
+    kyApp1: '{{kyApp1}}/?i_code='+kyCode,
+    kyApp2: '{{kyApp2}}/?i_code='+kyCode,
+    kyPc1: '{{kyPc1}}/register/?i_code='+kyCode,
+    kyPc2: '{{kyPc2}}/register?i_code='+kyCode,
+    kyH51: '{{kyH51}}/entry/register?i_code='+kyCode,
+    kyH52: '{{kyH52}}/entry/register?i_code='+kyCode
+}
+
+
+function visit_ky(key) {
+    window['open'](ky_link[key]);
+}