Add DHS as new provider.

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

diff --git a/src/ddns/providers.py b/src/ddns/providers.py
index 124cf2f8280e7618759725d733b0d3573492025a..9ffd67d302597562024ec689bfc600ada7f34d29 100644 (file)
--- a/src/ddns/providers.py
+++ b/src/ddns/providers.py
@@ -121,6 +121,118 @@ class DDNSProvider(object):
                return self.core.system.get_address(proto)
 
 
+class DDNSProviderDHS(DDNSProvider):
+       INFO = {
+               "handle"    : "dhs.org",
+               "name"      : "DHS International",
+               "website"   : "http://dhs.org/",
+               "protocols" : ["ipv4",]
+       }
+
+       # No information about the used update api provided on webpage,
+       # grabed from source code of ez-ipudate.
+       url = "http://members.dhs.org/nic/hosts"
+
+       def __call__(self):
+               url = self.url % {
+                       "username" : self.username,
+                       "password" : self.password,
+               }
+
+               data = {
+                       "domain"       : self.hostname,
+                       "ip"           : self.get_address("ipv4"),
+                       "hostcmd"      : "edit",
+                       "hostcmdstage" : "2",
+                       "type"         : "4",
+               }
+
+               # Send update to the server.
+               response = self.send_request(url, username=self.username, password=self.password,
+                       data=data)
+
+               # Handle success messages.
+               if response.code == 200:
+                       return
+
+               # Handle error codes.
+               elif response.code == "401":
+                       raise DDNSAuthenticationError
+
+               # If we got here, some other update error happened.
+               raise DDNSUpdateError
+
+
+class DDNSProviderLightningWireLabs(DDNSProvider):
+       INFO = {
+               "handle"    : "dns.lightningwirelabs.com",
+               "name"      : "Lightning Wire Labs",
+               "website"   : "http://dns.lightningwirelabs.com/",
+               "protocols" : ["ipv6", "ipv4",]
+       }
+
+       # Information about the format of the HTTPS request is to be found
+       # 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 __call__(self):
+               data =  {
+                       "hostname" : self.hostname,
+               }
+
+               # Check if we update an IPv6 address.
+               address6 = self.get_address("ipv6")
+               if address6:
+                       data["address6"] = address6
+
+               # Check if we update an IPv4 address.
+               address4 = self.get_address("ipv4")
+               if address4:
+                       data["address4"] = address4
+
+               # Raise an error if none address is given.
+               if not data.has_key("address6") and not data.has_key("address4"):
+                       raise DDNSConfigurationError
+
+               # Check if a token has been set.
+               if self.token:
+                       data["token"] = self.token
+
+               # Check for username and password.
+               elif self.username and self.password:
+                       data.update({
+                               "username" : self.username,
+                               "password" : self.password,
+                       })
+
+               # Raise an error if no auth details are given.
+               else:
+                       raise DDNSConfigurationError
+
+               # Send update to the server.
+               response = self.send_request(url, data=data)
+
+               # Handle success messages.
+               if response.code == 200:
+                       return
+
+               # Handle error codes.
+               if response.code == "403":
+                       raise DDNSAuthenticationError
+               elif response.code == "400":
+                       raise DDNSRequestError
+
+               # If we got here, some other update error happened.
+               raise DDNSUpdateError
+
+
 class DDNSProviderNOIP(DDNSProvider):
        INFO = {
                "handle"    : "no-ip.com",