Add Update Local Hosts

.gitignore

@@ -0,0 +1 @@
+.idea

dns.py

@@ -87,4 +87,27 @@ def UpdateDNS(access_key_id, access_key_secret, host, old_host_ip, new_load_bala
     subdomain_record_id = DescribeDomainRecords(access_key_id, access_key_secret, domain_name, subdomain_name, old_host_ip)
     # print (subdomain_record_id)
     UpdateDomainRecord(access_key_id, access_key_secret, subdomain_record_id, subdomain_name, new_load_balancer_ip)
-    print("DNS update record successful domain:%s, old ip:%s, new ip:%s " % (host, old_host_ip, new_load_balancer_ip))
+    print("DNS update record successful domain:%s, old ip:%s, new ip:%s " % (host, old_host_ip, new_load_balancer_ip))
+
+def UpdateHosts(new_load_balancer_ip, host):
+    """
+    更新本地Hosts, 防止DNS更新不及时导致一直检测旧IP
+    :param new_load_balancer_ip:
+    :param host:
+    :return:
+    """
+
+    with open("/etc/hosts", "r", encoding="utf-8") as f:
+        lines = f.readlines()
+        f.close()
+        # print(lines)
+
+    with open("/etc/hosts", "w", encoding="utf-8") as f_w:
+        for line in lines:
+            if host in line:
+                f_w.write("%s %s\n" % (new_load_balancer_ip, host))
+                print("Hosts update record successful domain:%s, new ip:%s " % (host, new_load_balancer_ip))
+                continue
+
+            f_w.write(line)
+        f_w.close()

main.py

@@ -73,6 +73,9 @@
     # 去DNS里更新A记录
     dns.UpdateDNS(access_key_id, access_key_secret, host, host_ip, new_load_balancer_ip)
 
+    # 更新本地Hosts, 防止DNS更新不及时导致一直检测旧IP
+    dns.UpdateHosts(new_load_balancer_ip, host)
+
 else:
     print("New LoadBalancer:%s Port:%s is not open，return code：%s" % (new_load_balancer_ip, port, result))