Wrote new version of script, please test before merging
This commit is contained in:
parent
fd19efd007
commit
26fcc3859f
2 changed files with 99 additions and 1 deletions
4
.gitignore
vendored
4
.gitignore
vendored
|
@ -1,2 +1,4 @@
|
|||
|
||||
# Default ignores
|
||||
venv/
|
||||
.idea/
|
||||
.vscode/
|
||||
|
|
96
cloudflare-dns-new.py
Normal file
96
cloudflare-dns-new.py
Normal file
|
@ -0,0 +1,96 @@
|
|||
import sys
|
||||
import ipaddress
|
||||
from configparser import ConfigParser
|
||||
import json
|
||||
import copy
|
||||
import CloudFlare
|
||||
|
||||
|
||||
class DNSUpdater:
|
||||
def __init__(self, args):
|
||||
# Parse config
|
||||
config = ConfigParser()
|
||||
config.read('config.ini')
|
||||
|
||||
self.HOSTNAME = config['server']['HOSTNAME']
|
||||
__TOKEN = config['cloudflare']['TOKEN']
|
||||
|
||||
# Parse args
|
||||
try:
|
||||
ipversion = args[1].lstrip('-')
|
||||
ipaddress_new = args[2]
|
||||
except IndexError:
|
||||
ipaddress_new = ''
|
||||
ipversion = 'skip'
|
||||
|
||||
match ipversion:
|
||||
case '4': self.IPVER = 'ipv4'
|
||||
case '6': self.IPVER = 'ipv6'
|
||||
case 'skip': self.IPVER = ipversion
|
||||
case other: raise ValueError(f"Wrong IP Address version given: {ipversion}")
|
||||
|
||||
self.IPNEW = ipaddress_new
|
||||
|
||||
# Cloudflare class
|
||||
self._cloudflare = CloudFlare.CloudFlare(token=__TOKEN)
|
||||
|
||||
with open('cloudflare.json', 'r') as file:
|
||||
self.dns_replacements = json.load(file)
|
||||
|
||||
def get_zones(self):
|
||||
# Cloudflare get zones from API token
|
||||
zones = self._cloudflare.zones.get()
|
||||
for zone in zones:
|
||||
self.zone_id = zone['id']
|
||||
self.zone_name = zone['name']
|
||||
print(f"Zone ID: {zone['id']} | Zone Name: {zone['name']}")
|
||||
|
||||
def get_dns_records(self, log=False):
|
||||
dns_records = self._cloudflare.zones.dns_records.get(self.zone_id)
|
||||
|
||||
dns_names = {'A' : [], 'AAAA' : []}
|
||||
for dnsrecord in dns_records:
|
||||
if log:
|
||||
print(f"DNS Name: {dnsrecord['name']} | ID: {dnsrecord['id']} | Type : {dnsrecord['type']} | IP: {dnsrecord['content']}")
|
||||
dns_names[dnsrecord['type']].append(dnsrecord["name"])
|
||||
|
||||
self.DNSNAMES = copy.deepcopy(dns_names)
|
||||
|
||||
def update_records(self):
|
||||
if self.IPVER == 'skip':
|
||||
self.get_dns_records(log=True)
|
||||
return
|
||||
|
||||
self.get_dns_records()
|
||||
|
||||
ipinformation = {
|
||||
'ipv4': [ipaddress.IPv4Address, 'A'],
|
||||
'ipv6': [ipaddress.IPv6Address, 'AAAA']
|
||||
}
|
||||
|
||||
print(f"Updating IP of version {self.IPVER}")
|
||||
assert isinstance(ipaddress.ip_address(self.IPNEW), ipinformation[self.IPVER][0]), f'IP {self.IPNEW} is not a valid IPv6 address'
|
||||
dnstype = ipinformation[self.IPVER][1]
|
||||
|
||||
names_replace = self.dns_replacements[dnstype]
|
||||
print(names_replace)
|
||||
for name in names_replace:
|
||||
fullname = f"{name}.{self.HOSTNAME}"
|
||||
print(fullname)
|
||||
|
||||
for x in self.DNSNAMES:
|
||||
if x['name'] == fullname:
|
||||
dns_record_id = x['id']
|
||||
print(dns_record_id)
|
||||
|
||||
if fullname in self.DNSNAMES[dnstype]:
|
||||
new_dnsrecord = {"name": fullname, "type": "AAAA", "content": self.IPNEW}
|
||||
print(f"Sending request {new_dnsrecord}")
|
||||
r = self._cloudflare.zones.dns_records.patch(self.zone_id, dns_record_id, data=new_dnsrecord)
|
||||
print(r)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
dnsupdate = DNSUpdater(sys.argv)
|
||||
dnsupdate.get_zones()
|
||||
dnsupdate.update_records()
|
Loading…
Reference in a new issue