Move call to get token to main class.

[ddns.git] / src / ddns / providers.py

diff --git a/src/ddns/providers.py b/src/ddns/providers.py
index 03201baadee0b3352ac34edc1df6689b8f7b084d..cbbeeea2c74f9f8ce1783a14f0b989f4e62ab900 100644 (file)
--- a/src/ddns/providers.py
+++ b/src/ddns/providers.py
@@ -115,6 +115,13 @@ class DDNSProvider(object):
        def protocols(self):
                return self.INFO.get("protocols")
 
+       @property
+       def token(self):
+               """
+                       Fast access to the token.
+               """
+               return self.get("token")
+
        def __call__(self, force=False):
                if force:
                        logger.info(_("Updating %s forced") % self.hostname)
@@ -195,7 +202,7 @@ class DDNSProviderDHS(DDNSProvider):
                        return
 
                # Handle error codes.
-               elif response.code == "401":
+               elif response.code == 401:
                        raise DDNSAuthenticationError
 
                # If we got here, some other update error happened.
@@ -255,7 +262,109 @@ class DDNSProviderDNSpark(DDNSProvider):
                # If we got here, some other update error happened.
                raise DDNSUpdateError
 
-               
+
+class DDNSProviderDtDNS(DDNSProvider):
+       INFO = {
+               "handle"    : "dtdns.com",
+               "name"      : "DtDNS",
+               "website"   : "http://dtdns.com/",
+               "protocols" : ["ipv4",]
+               }
+
+       # Information about the format of the HTTPS request is to be found
+       # http://www.dtdns.com/dtsite/updatespec
+       url = "https://www.dtdns.com/api/autodns.cfm"
+
+
+       def update(self):
+               data = {
+                       "ip" : self.get_address("ipv4"),
+                       "id" : self.hostname,
+                       "pw" : self.password
+               }
+
+               # Send update to the server.
+               response = self.send_request(self.url, data=data)
+
+               # Get the full response message.
+               output = response.read()
+
+               # Remove all leading and trailing whitespace.
+               output = output.strip()
+
+               # Handle success messages.
+               if "now points to" in output:
+                       return
+
+               # Handle error codes.
+               if output == "No hostname to update was supplied.":
+                       raise DDNSRequestError(_("No hostname specified."))
+
+               elif output == "The hostname you supplied is not valid.":
+                       raise DDNSRequestError(_("Invalid hostname specified."))
+
+               elif output == "The password you supplied is not valid.":
+                       raise DDNSAuthenticationError
+
+               elif output == "Administration has disabled this account.":
+                       raise DDNSRequestError(_("Account has been disabled."))
+
+               elif output == "Illegal character in IP.":
+                       raise DDNSRequestError(_("Invalid IP address has been sent."))
+
+               elif output == "Too many failed requests.":
+                       raise DDNSRequestError(_("Too many failed requests."))
+
+               # If we got here, some other update error happened.
+               raise DDNSUpdateError
+
+
+class DDNSProviderFreeDNSAfraidOrg(DDNSProvider):
+       INFO = {
+               "handle"    : "freedns.afraid.org",
+               "name"      : "freedns.afraid.org",
+               "website"   : "http://freedns.afraid.org/",
+               "protocols" : ["ipv6", "ipv4",]
+               }
+
+       # No information about the request or response could be found on the vendor
+       # page. All used values have been collected by testing.
+       url = "https://freedns.afraid.org/dynamic/update.php"
+
+       @property
+       def proto(self):
+               return self.get("proto")
+
+       def update(self):
+               address = self.get_address(self.proto)
+
+               data = {
+                       "address" : address,
+               }
+
+               # Add auth token to the update url.
+               url = "%s?%s" % (self.url, self.token)
+
+               # Send update to the server.
+               response = self.send_request(url, data=data)
+
+               # Get the full response message.
+               output = response.read()
+
+               # Handle success messages.
+               if output.startswith("Updated") or "has not changed" in output:
+                       return
+
+               # Handle error codes.
+               if output == "ERROR: Unable to locate this record":
+                       raise DDNSAuthenticationError
+               elif "is an invalid IP address" in output:
+                       raise DDNSRequestError(_("Invalid IP address has been sent."))
+
+               # If we got here, some other update error happened.
+               raise DDNSUpdateError
+
+
 class DDNSProviderLightningWireLabs(DDNSProvider):
        INFO = {
                "handle"    : "dns.lightningwirelabs.com",
@@ -268,13 +377,6 @@ class DDNSProviderLightningWireLabs(DDNSProvider):
        # https://dns.lightningwirelabs.com/knowledge-base/api/ddns
        url = "https://dns.lightningwirelabs.com/update"
 
-       @property
-       def token(self):
-               """
-                       Fast access to the token.
-               """
-               return self.get("token")
-
        def update(self):
                data =  {
                        "hostname" : self.hostname,