Update AliCloudIPUpdater.py

- 实现样例文件逻辑
- 实现循环功能
- 实现UDP管理

AliCloudIPUpdater.py

@@ -1,13 +1,21 @@
 import os
 import json
 import requests
+import time
+import shutil
 from requests.exceptions import RequestException
 from aliyunsdkcore.client import AcsClient
 from aliyunsdkcore.acs_exception.exceptions import ClientException, ServerException
 from aliyunsdkecs.request.v20140526 import DescribeSecurityGroupAttributeRequest, AuthorizeSecurityGroupRequest, RevokeSecurityGroupRequest
 
 # 从配置文件加载配置信息
-def load_config(config_file):
+def load_config(config_file, sample_config_file):
+    if not os.path.exists(config_file):
+        print(f"配置文件 {config_file} 不存在。")
+        print(f"将从样例配置文件 {sample_config_file} 复制一份。")
+        shutil.copy(sample_config_file, config_file)
+        print(f"请在 {config_file} 中配置您的设置。")
+        raise FileNotFoundError(f"配置文件 {config_file} 不存在。")
     try:
         with open(config_file, 'r') as file:
             config = json.load(file)
@@ -65,14 +73,16 @@ def delete_old_rules(client, security_group_id, tag):
 # 更新安全组白名单
 def update_security_group_white_list(client, security_group_id, ip, ports, tag, priority):
     try:
-        for port in ports:
+        for port_info in ports:
+            port = port_info["port"]
+            protocol = port_info["protocol"]
             # 将单个端口转换为端口范围格式
             if '-' not in port:
                 port = f"{port}/{port}"
             request = AuthorizeSecurityGroupRequest.AuthorizeSecurityGroupRequest()
             request.set_accept_format('json')
             request.set_SecurityGroupId(security_group_id)
-            request.set_IpProtocol('tcp')
+            request.set_IpProtocol(protocol)
             request.set_PortRange(port)
             request.set_SourceCidrIp(ip + "/32")
             request.set_Policy('accept')
@@ -111,11 +121,12 @@ def load_ip_records(ip_record_file):
         print(f"IP记录文件 {ip_record_file} 解析错误: {e}")
         raise
 
-if __name__ == "__main__":
+def main():
     try:
         # 加载配置文件
         config_path = os.environ.get('CONFIG_PATH', 'config.json')
-        config = load_config(config_path)
+        sample_config_path = os.environ.get('SAMPLE_CONFIG_PATH', 'config.sample.json')
+        config = load_config(config_path, sample_config_path)
         
         ACCESS_KEY_ID = config['ACCESS_KEY_ID']
         ACCESS_KEY_SECRET = config['ACCESS_KEY_SECRET']
@@ -126,9 +137,11 @@ if __name__ == "__main__":
         PORTS = config['PORTS']
         PRIORITY = config.get('PRIORITY', 1)
         IP_RECORD_FILE = config.get('IP_RECORD_FILE', 'ip_records.json')
+        INTERVAL_SECONDS = config.get('INTERVAL_SECONDS', 3600)
         
         client = AcsClient(ACCESS_KEY_ID, ACCESS_KEY_SECRET, REGION_ID)
 
+        while True:
             # 读取本地记录的IP地址
             ip_records = load_ip_records(IP_RECORD_FILE)
             new_ip_records = {}
@@ -151,5 +164,11 @@ if __name__ == "__main__":
             # 记录新的IP地址到本地文件
             record_ip(IP_RECORD_FILE, new_ip_records)
 
+            # 等待指定的时间间隔
+            time.sleep(INTERVAL_SECONDS)
+
     except Exception as e:
         print(f"程序运行过程中出现错误: {e}")
+
+if __name__ == "__main__":
+    main()