122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
import requests
|
|
import os
|
|
import json
|
|
|
|
# Function to get the public IP address
|
|
def get_public_ip():
|
|
response = requests.get('https://httpbin.org/ip')
|
|
if response.status_code == 200:
|
|
return response.json().get('origin')
|
|
else:
|
|
print('Failed to fetch IP address')
|
|
return None
|
|
|
|
# Function to read the last recorded IP from a file
|
|
def read_last_ip(file_path):
|
|
if os.path.exists(file_path):
|
|
with open(file_path, 'r') as file:
|
|
return file.read().strip()
|
|
return None
|
|
|
|
# Function to save the current IP to a file
|
|
def save_current_ip(file_path, current_ip):
|
|
with open(file_path, 'w') as file:
|
|
file.write(current_ip)
|
|
|
|
# Function to update all 'A' type DNS records to current IP if it has changed
|
|
def update_dns_if_ip_changed(ip_file_path, zone_id, api_key, email):
|
|
current_ip = get_public_ip()
|
|
if not current_ip:
|
|
print("Could not retrieve current IP. Aborting DNS update.")
|
|
return
|
|
|
|
last_ip = read_last_ip(ip_file_path)
|
|
if current_ip != last_ip:
|
|
save_current_ip(ip_file_path, current_ip)
|
|
update_all_a_records(zone_id, api_key, email, current_ip)
|
|
else:
|
|
print("IP has not changed. Skipping DNS update.")
|
|
|
|
# Function to update all 'A' type DNS records to current IP
|
|
def update_all_a_records(zone_id, api_key, email, current_ip):
|
|
if not current_ip:
|
|
print("Could not retrieve current IP. Aborting DNS update.")
|
|
return
|
|
|
|
# Fetch all 'A' type DNS records for the specified zone
|
|
url = f"https://api.cloudflare.com/client/v4/zones/{zone_id}/dns_records?type=A"
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
"X-Auth-Email": email,
|
|
"X-Auth-Key": api_key
|
|
}
|
|
|
|
response = requests.get(url, headers=headers)
|
|
|
|
if response.status_code == 200:
|
|
dns_records = response.json()["result"]
|
|
for record in dns_records:
|
|
record_id = record["id"]
|
|
payload = {
|
|
"content": current_ip,
|
|
"name": record["name"],
|
|
"proxied": record["proxied"],
|
|
"type": record["type"],
|
|
"ttl": record["ttl"]
|
|
}
|
|
update_dns_record(zone_id, api_key, email, record_id, payload)
|
|
else:
|
|
print("Failed to fetch DNS records")
|
|
print(f"Status Code: {response.status_code}")
|
|
print(f"Response: {response.text}")
|
|
|
|
# Function to update DNS record
|
|
def update_dns_record(zone_id, api_key, email, record_id, payload):
|
|
url = f"https://api.cloudflare.com/client/v4/zones/{zone_id}/dns_records/{record_id}"
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
"X-Auth-Email": email,
|
|
"X-Auth-Key": api_key
|
|
}
|
|
|
|
response = requests.put(url, headers=headers, json=payload)
|
|
|
|
if response.status_code == 200:
|
|
print(f"DNS record for ID {record_id} updated successfully!")
|
|
print(response.json())
|
|
else:
|
|
print(f"Failed to update DNS record for ID {record_id}")
|
|
print(f"Status Code: {response.status_code}")
|
|
print(f"Response: {response.text}")
|
|
|
|
# Check if the IP file exists and read the last recorded IP
|
|
ip_file_path = 'last_ip.txt'
|
|
|
|
# Check if the credentials file exists and read the credentials
|
|
credentials_file_path = 'credentials.json'
|
|
|
|
# If credentials file doesn't exist, prompt user for information and save it to the file
|
|
if not os.path.exists(credentials_file_path):
|
|
zone_id = input("Enter your Zone ID: ")
|
|
auth_email = input("Enter your Cloudflare email: ")
|
|
auth_key = input("Enter your Cloudflare API key: ")
|
|
|
|
credentials = {
|
|
"zone_id": zone_id,
|
|
"auth_email": auth_email,
|
|
"auth_key": auth_key
|
|
}
|
|
|
|
with open(credentials_file_path, 'w') as file:
|
|
json.dump(credentials, file)
|
|
else:
|
|
# Read credentials from the file
|
|
with open(credentials_file_path, 'r') as file:
|
|
credentials = json.load(file)
|
|
zone_id = credentials.get("zone_id")
|
|
auth_email = credentials.get("auth_email")
|
|
auth_key = credentials.get("auth_key")
|
|
|
|
# Call the update_dns_if_ip_changed function
|
|
update_dns_if_ip_changed(ip_file_path, zone_id, auth_key, auth_email)
|