generate_node.py 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. # -*- coding: utf-8 -*-
  2. # 生成代理节点, 并过滤, 传入 merge.yaml 生成 config.yaml
  3. import yaml
  4. file_name = 'merge.yaml'
  5. need_nodes = ['日本', 'JP', '新加坡', 'SG', '台湾', 'TW', '韩国', '澳门', '马来西亚', '澳大利亚', '菲律宾']
  6. config_file = {
  7. "mixed-port": 7890,
  8. "allow-lan": True,
  9. "bind-address": "*",
  10. "mode": "rule",
  11. "log-level": "info",
  12. "external-controller": "0.0.0.0:9090",
  13. "secret": "",
  14. "dns": {
  15. "enable": False,
  16. "ipv6": False,
  17. "nameserver": [],
  18. "fallback": []
  19. },
  20. "proxies": [],
  21. "proxy-groups": [{"name": "GLOBAL", "type": "select", "proxies": []}],
  22. }
  23. def process_yaml_file():
  24. proxy_list = []
  25. with open(file_name, 'r', encoding='utf-8') as file:
  26. data = yaml.safe_load(file)
  27. if not data:
  28. print(f"Error reading {file_name}")
  29. return
  30. proxies = data['proxies']
  31. for proxy in proxies:
  32. if any(x in proxy['name'] for x in ['剩余流量', '套餐到期', '有效期']):
  33. continue
  34. if any(node in proxy['name'] for node in need_nodes):
  35. proxy_list.append(proxy)
  36. proxy_list_sorted = sorted(proxy_list, key=lambda x: x['name'] if 'name' in x else '')
  37. return proxy_list_sorted
  38. def create_config_file(proxy_list_sorted):
  39. config_file['proxies'] = proxy_list_sorted
  40. for proxy_list in proxy_list_sorted:
  41. config_file['proxy-groups'][0]['proxies'].append(proxy_list['name'])
  42. if __name__ == '__main__':
  43. proxy_list_sorted = process_yaml_file()
  44. create_config_file(proxy_list_sorted)
  45. with open(f'config.yaml', 'w', encoding='utf-8') as file:
  46. yaml.dump(config_file, file, allow_unicode=True, default_flow_style=False)
  47. print('done!')