Moved api request to /lookup

README.md

@@ -1,2 +1,3 @@
 # dnspy3-server
 
+## Server for DNSPy3

dnspy3-server.py

@@ -1,5 +1,6 @@
 from flask import Flask, request
 from flask_restful import Resource, Api
+import json
 
 app = Flask(__name__)
 api = Api(app)
@@ -8,6 +9,12 @@ api = Api(app)
 class WhoIS():
     def __init__(self, domain):
         self.domain = domain
+        self.getWhois()
+
+    def getWhois(self):
+        self.registrar = 'Synergy Wholesale'
+        self.status = ['clientDeleteProhibited https://icann.org/epp#clientDeleteProhibited', 'clientTransferProhibited https://icann.org/epp#clientTransferProhibited', 'clientUpdateProhibited https://icann.org/epp#clientUpdateProhibited',
+                       'serverDeleteProhibited https://icann.org/epp#serverDeleteProhibited', 'serverTransferProhibited https://icann.org/epp#serverTransferProhibited', 'serverUpdateProhibited https://icann.org/epp#serverUpdateProhibited']
         self.registrant = {'name': 'Test Case', 'email': 'tcase@test.com',
                            'eligibilitytype': 'Company', 'eligibilityid': '123456789'}
         self.nameservers = ['ns1.dommain.tld', 'ns2.domain.tld']
@@ -15,37 +22,29 @@ class WhoIS():
 
 class DNS():
     def __init__(self, domain):
-        self.getARecords()
-        self.getAAAARecords()
-        self.getMXRecords()
-        self.getTXTRecords()
-        self.getNSRecords()
+        self.domain = domain
+        self.getRecords()
 
-    def getARecords(self):
+    def getRecords(self):
         self.a = '1.2.3.4'
-
-    def getAAAARecords(self):
         self.aaaa = '2001:0db8:85a3:0000:0000:8a2e:0370:7334'
-
-    def getMXRecords(self):
         self.mx = {'mail.domain.tld': 10}
-
-    def getTXTRecords(self):
         self.txt = 'v=spf1 +a +mx +include:spf.hostingplatform.net.au'
-
-    def getNSRecords(self):
         self.ns = ['ns1.domain.tld', 'ns2.domain.tld']
+        self.soa = 'ns1.domain.tld. webmaster.domain.tld. 310875860 900 900 1800 60'
 
 
 class Domain():
     def __init__(self, domain):
-        self.domain = domain
         self.whois = WhoIS(domain).__dict__
         self.dns = DNS(domain).__dict__
 
 
 class DNSLookup(Resource):
     def get(self, domain=None):
+        auth = request.headers.get('X-AUTH')
+        if auth != None:
+            return [{'message': 'Please auth'}, 'error']
         if domain == None:
             return [{'message': 'Please specify a domain name'}, 'error']
         else:
@@ -54,7 +53,13 @@ class DNSLookup(Resource):
             return [ret.__dict__, status]
 
 
-api.add_resource(DNSLookup, '/', '/<string:domain>')
+api.add_resource(DNSLookup, '/lookup', '/lookup/<string:domain>')
+
+
+@app.route('/')
+def index():
+    return ''
+
 
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0')