From fc197ea2a2ac4ead704ca71d606f468085bac40f Mon Sep 17 00:00:00 2001 From: Polarolouis Date: Fri, 29 Jul 2022 11:18:15 +0200 Subject: [PATCH] Changed the script not to use OOP and to have arguments --- gandyndns.py | 130 +++++++++++++++++++++++++++++---------------------- 1 file changed, 75 insertions(+), 55 deletions(-) diff --git a/gandyndns.py b/gandyndns.py index 90c4b48..9de52b6 100644 --- a/gandyndns.py +++ b/gandyndns.py @@ -3,65 +3,85 @@ # @author: lordof20th import requests +import argparse -class GanDynDns: - def __init__(self, fqdn, rrset_name, rrset_type, apikey) -> None: - self.fqdn = fqdn - self.rrset_name = rrset_name - self.rrset_type = rrset_type - self.domain_record_string = f"DNS {self.rrset_type} record for {self.rrset_name}.{self.fqdn}" - self.apiUrl = f"https://api.gandi.net/v5/livedns/domains/{fqdn}/records/{rrset_name}/{rrset_type}" - self.headers = {"Authorization": f"Apikey {apikey}", 'User-Agent': 'Mozilla/5.0', "Content-Type": "application/json"} - self.update_dns_IP() +parser = argparse.ArgumentParser( + description="A script which connect to Gandi.net API to change IP associated to a DNS record") +parser.add_argument("-v", "--verbose", + help="Enable verbose mode", action="store_true") +parser.add_argument("domain", metavar="DOMAIN", + help="The domain for which you want to write a DNS record. Example: example.com") +parser.add_argument("subdomain", metavar="SUBDOMAIN", + help="The subdomain to point to. Examples: 'sub' or '@'") +parser.add_argument("apikey", metavar="APIKEY", help="Your Gandi.net API key") +parser.add_argument("--type", metavar="TYPE", default='A', + help="The type of DNS record to create. Default: A") - def retrieve_dns_IP(self): - """Retrieves the IP in the DNS record using the fqdn, rrset_name (for instance subdomain like www etc) and rrset_type (the DNS record type A, CNAME ...). - - The function uses requests library and connect to Gandi.net API - - Returns: - retrievedDnsIp (str): string of the IP address in the DNS record""" - - response = requests.get(self.apiUrl, headers=self.headers) +args = parser.parse_args() +print(args) - responseJson = response.json() # IPs are stored in a list as string - try: - retrievedDnsIp = responseJson['rrset_values'][0] - except KeyError as key_error: - print(f"{key_error} means the record doesn't exist, we'll return an empty string instead") - retrievedDnsIp ="" - return retrievedDnsIp +verbose = args.verbose +domain = args.domain +apikey = args.apikey +subdomain = args.subdomain +type = args.type - def retrieve_public_IP(self): - """Retrieves the public IP by connecting to ipinfo.io API - - Returns: - data['ip'] (str) : string of the public IP address""" - endpoint = "https://ipinfo.io/json" - response = requests.get(endpoint, verify = True) - - data = response.json() - return data['ip'] +domain_record_string = f"DNS {type} record for {subdomain}.{domain}" - def ips_are_equals(self): - """The method compares both IPs and returns a boolean for the equality test - - Returns: - (bool) : result of the IP equality test""" - self.currentPublicIP = self.retrieve_public_IP() - self.dnsIP = self.retrieve_dns_IP() - return self.currentPublicIP == self.dnsIP +domain_record_string_lenght = len(domain_record_string) - def update_dns_IP(self): - """Updates the IP in the DNS record using the IP provided (acquired by retrieve_public_IP) if it is different from the DNS IP""" - - if not self.ips_are_equals(): - data = { - "rrset_values": [self.currentPublicIP] - } - print(f"{self.domain_record_string : <40} | Old IP : {self.dnsIP} replaced -> by New IP : {self.currentPublicIP}") - requests.put(url=self.apiUrl, headers=self.headers, json=data) - else: - print(f"{self.domain_record_string : <40} | {'IPs are the same.':<17}") +apiUrl = f"https://api.gandi.net/v5/livedns/domains/{domain}/records/{subdomain}/{type}" -main = GanDynDns("example.org", "@", "A", "your-api-key") \ No newline at end of file +headers = {"Authorization": f"Apikey {apikey}", + 'User-Agent': 'Mozilla/5.0', "Content-Type": "application/json"} + + +def retrieve_public_IP(): + """Retrieves the public IP by connecting to ipinfo.io API + + Returns: + data['ip'] (str) : string of the public IP address""" + endpoint = "https://ipinfo.io/json" + response = requests.get(endpoint, verify=True) + + data = response.json() + return data['ip'] + + +def retrieve_dns_IP(apiUrl, headers): + """Retrieves the IP in the DNS record using the domain, rrset_name (for instance subdomain like www etc) and rrset_type (the DNS record type A, CNAME ...). + + The function uses requests library and connect to Gandi.net API + + Returns: + retrievedDnsIp (str): string of the IP address in the DNS record""" + + response = requests.get(apiUrl, headers=headers) + + responseJson = response.json() # IPs are stored in a list as string + try: + retrievedDnsIp = responseJson['rrset_values'][0] + except KeyError as key_error: + print( + f"{key_error} means the record doesn't exist, we'll return an empty string instead") + retrievedDnsIp = "" + return retrievedDnsIp + + +def update_dns_IP(apiUrl, headers): + """Updates the IP in the DNS record using the IP provided (acquired by retrieve_public_IP) if it is different from the DNS IP""" + publicIP = retrieve_public_IP() + dnsIP = retrieve_dns_IP(apiUrl, headers) + + if not publicIP == dnsIP: + data = { + "rrset_values": [publicIP] + } + print(f"{domain_record_string : <{domain_record_string_lenght}} | Old IP : {dnsIP} replaced -> by New IP : {publicIP}") + requests.put(url=apiUrl, headers=headers, json=data) + else: + print( + f"{domain_record_string : <{domain_record_string_lenght}} | {'IPs are the same.':<17}") + +if __name__ == '__main__': + update_dns_IP(apiUrl, headers)