halbebruno/cloudflare-ddns
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle sigterm and shutdown immediately

Browse Source
This commit is contained in:
Mark Ormesher 2021-01-20 18:50:36 +00:00
parent 839ffe2551
commit d763be7931
No known key found for this signature in database
GPG Key ID: DE308192D728C145
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
cloudflare-ddns.py
View File

@ -1,4 +1,4 @@
import requests, json, sys, os import requests, json, sys, signal, os
import time import time
PATH = os.getcwd() + "/" PATH = os.getcwd() + "/"
@ -7,6 +7,12 @@ version = float(str(sys.version_info[0]) + "." + str(sys.version_info[1]))
if(version < 3.5): if(version < 3.5):
raise Exception("This script requires Python 3.5+") raise Exception("This script requires Python 3.5+")
def sigtermHandler(sig_no, stack_frame):
print("Caught SIGTERM, shutting down...")
sys.exit(0)
signal.signal(signal.SIGTERM, sigtermHandler)
with open(PATH + "config.json") as config_file: with open(PATH + "config.json") as config_file:
config = json.loads(config_file.read()) config = json.loads(config_file.read())
@ -67,11 +73,11 @@ def commitRecord(ip):
} }
list = cf_api( list = cf_api(
"zones/" + c['zone_id'] + "/dns_records?per_page=100&type=" + ip["type"], "GET", c) "zones/" + c['zone_id'] + "/dns_records?per_page=100&type=" + ip["type"], "GET", c)
full_subdomain = base_domain_name full_subdomain = base_domain_name
if subdomain: if subdomain:
full_subdomain = subdomain + "." + full_subdomain full_subdomain = subdomain + "." + full_subdomain
dns_id = "" dns_id = ""
for r in list["result"]: for r in list["result"]:
if (r["name"] == full_subdomain): if (r["name"] == full_subdomain):
@ -110,7 +116,7 @@ def cf_api(endpoint, method, config, headers={}, data=False):
else: else:
headers = { headers = {
"X-Auth-Email": config['authentication']['api_key']['account_email'], "X-Auth-Email": config['authentication']['api_key']['account_email'],
"X-Auth-Key": config['authentication']['api_key']['api_key'], "X-Auth-Key": config['authentication']['api_key']['api_key'],
} }
if(data == False): if(data == False):
@ -126,17 +132,20 @@ def updateIPs():
for ip in getIPs(): for ip in getIPs():
commitRecord(ip) commitRecord(ip)
if(len(sys.argv) > 1): try:
if(sys.argv[1] == "--repeat"): if(len(sys.argv) > 1):
print("Updating A & AAAA records every 10 minutes") if(sys.argv[1] == "--repeat"):
updateIPs() print("Updating A & AAAA records every 10 minutes")
delay = 10*60 # 10 minutes
next_time = time.time() + delay
while True:
time.sleep(max(0, next_time - time.time()))
updateIPs() updateIPs()
next_time += (time.time() - next_time) // delay * delay + delay delay = 10*60 # 10 minutes
next_time = time.time() + delay
while True:
time.sleep(max(0, next_time - time.time()))
updateIPs()
next_time += (time.time() - next_time) // delay * delay + delay
else:
print("Unrecognized parameter '" + sys.argv[1] + "'. Stopping now.")
else: else:
print("Unrecognized parameter '" + sys.argv[1] + "'. Stopping now.") updateIPs()
else: except SystemExit:
updateIPs() print("Goodbye!")