1 files changed, 12 insertions, 10 deletions

diff --git a/wsgi.py b/wsgi.py
index 49224b9..cfa1326 100644
--- a/wsgi.py
+++ b/wsgi.py
@@ -1,19 +1,21 @@
 #!/usr/bin/env python3.6
-
 import time
 import re
 import subprocess
 from cgi import parse_qs
 
-def application(environ, start_response):
-    d = parse_qs(environ['QUERY_STRING'])
+zone = "wire-me.de"
 
+def application(environ, start_response):
     start_response('200 OK', [('Content-Type', 'text/plain')])
-    if re.match('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}', d.get('ip', ['0'])[0]):
-        ip = d.get('ip', ['0'])[0].encode()
-        if re.match('[0-9a-z]+', d.get('host', [''])[0]):
-            host = d.get('host', ['0'])[0].encode()
-            yield(subprocess.run(["/usr/local/bin/pdnsutil", "delete-rrset", "wire-me.de", host, "A"], stdout=subprocess.PIPE).stdout)
-            yield(subprocess.run(["/usr/local/bin/pdnsutil", "add-record", "wire-me.de", host, "A", ip], stdout=subprocess.PIPE))
-            yield(subprocess.run(["/usr/local/bin/pdnsutil", "list-zone", "wire-me.de"], stdout=subprocess.PIPE).stdout)
+    args = parse_qs(environ['QUERY_STRING'])
+    host = args.get('host', ['0'])[0]
+
+    if re.match('[0-9a-z-_]+', host):
+        yield(subprocess.run(["/usr/local/bin/pdnsutil", "delete-rrset", zone, host, "A"], stdout=subprocess.PIPE).stdout)
+
+        for ip in args.get('ip', ['0']):
+            if re.match('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}', ip):
+                yield(subprocess.run(["/usr/local/bin/pdnsutil", "add-record", zone, host, "A", ip], stdout=subprocess.PIPE))
 
+    yield(subprocess.run(["/usr/local/bin/pdnsutil", "list-zone", zone], stdout=subprocess.PIPE).stdout)