From: Stefan Schantl <stefan.schantl@ipfire.org>
Date: Mon, 9 Jun 2014 19:35:04 +0000 (+0200)
Subject: Add DHS as new provider.
X-Git-Tag: 001~74
X-Git-Url: http://git.ipfire.org/?p=ddns.git;a=commitdiff_plain;h=f3cf1f7090102ea4a4520542d470e18a4791b8d3;ds=sidebyside

Add DHS as new provider.
---

diff --git a/ddns.conf b/ddns.conf
index b54f8ef..83ac1dc 100644
--- a/ddns.conf
+++ b/ddns.conf
@@ -15,6 +15,11 @@
 # Accounts are its own sections.
 # These are some examples.
 
+# [test.dhs.org]
+# provider = dhs.org
+# username = user
+# password = pass
+
 # [test.no-ip.org]
 # provider = no-ip.com
 # username = user
diff --git a/src/ddns/__init__.py b/src/ddns/__init__.py
index 3b465f0..74989f3 100644
--- a/src/ddns/__init__.py
+++ b/src/ddns/__init__.py
@@ -86,6 +86,7 @@ class DDNSCore(object):
 			Simply registers all providers.
 		"""
 		for provider in (
+			DDNSProviderDHS,
 			DDNSProviderNOIP,
 			DDNSProviderLightningWireLabs,
 			DDNSProviderSelfhost,
diff --git a/src/ddns/providers.py b/src/ddns/providers.py
index 0a20e3a..9ffd67d 100644
--- a/src/ddns/providers.py
+++ b/src/ddns/providers.py
@@ -121,6 +121,48 @@ 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",