| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- # -*- coding: utf-8 -*-
- import os
- import yaml
- import shutil
- # 分组关键词
- proxy_split_group = {
- "US": ['us', '美国', 'la-', '加拿大'],
- 'UE': ['uk', '英国'],
- 'ASIA': ['hk', '香港', 'tw', '台湾', 'sg', '新加坡', '马来西亚', '澳门', '泰国', '日本', 'jp', '韩国']
- }
- # 新文件夹名称
- new_folder_name = "config"
- downloads_path = os.path.join(os.path.expanduser('~'), 'Downloads')
- # 新文件夹的完整路径
- new_folder_path = os.path.join(downloads_path, new_folder_name)
- # 如果文件夹存在,删除整个文件夹
- if os.path.exists(new_folder_path):
- shutil.rmtree(new_folder_path)
- # 创建新的文件夹
- os.makedirs(new_folder_path)
- config_path = os.path.join(os.getcwd(), "merge.yaml")
- with open(config_path, 'r', encoding='utf-8') as file:
- data = yaml.safe_load(file)
- if not data:
- print(f"Error reading {config_path}")
- exit(1)
- config_name_serial_start = 1
- port_start = 7890
- out_data = {}
- for i in data.setdefault('proxies'):
- for group_name, group_keywords in proxy_split_group.items():
- if any(keyword in i['name'].lower() for keyword in group_keywords):
- if group_name not in out_data:
- out_data[group_name] = []
- if i['name'] not in [x['name'] for x in out_data[group_name]]:
- out_data[group_name].append(i)
- for g_name, g_proxy in out_data.items():
- config_template = {
- "mixed-port": port_start,
- "allow-lan": True,
- "bind-address": "*",
- "mode": "rule",
- "log-level": "info",
- "external-controller": "0.0.0.0:9090",
- "secret": "",
- "dns": {
- "enable": False,
- "ipv6": False,
- "nameserver": [],
- "fallback": []
- },
- "proxies": [],
- "proxy-groups": [{'name': '所有节点', 'type': 'select', 'proxies': []}],
- }
- for proxy in g_proxy:
- config_template['proxies'].append(proxy)
- config_template['proxy-groups'][0]['proxies'].append(proxy['name'])
- config_path = os.path.join(new_folder_path, f'{g_name.lower()}.yaml')
- with open(config_path, 'w', encoding='utf-8') as file:
- yaml.dump(config_template, file, allow_unicode=True, sort_keys=False)
- print('OK')
|