get RECORD_TYPE by sys.argv[2]

hooboxu · Jan 3, 2023 · 61d4e75 · 61d4e75
1 parent 2b3e120
commit 61d4e75
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cf2dns_actions.py b/cf2dns_actions.py
@@ -52,7 +52,10 @@ def log_info(msg: str):
 # 解析生效时间，默认为600秒 如果不是DNS付费版用户 不要修改!!!
 TTL = 600
 # A为筛选出IPv4的IP  AAAA为筛选出IPv6的IP
-RECORD_TYPE = 'A'
+if len(sys.argv) >= 3:
+    RECORD_TYPE = sys.argv[2]
+else:
+    RECORD_TYPE = "A"
 
 
 def get_optimization_ip():