41 lines
958 B
Python
41 lines
958 B
Python
from flask import Flask, request
|
|
from flask_cors import CORS
|
|
from flask_restful import Resource, Api
|
|
import json
|
|
import utils
|
|
|
|
app = Flask(__name__)
|
|
CORS(app)
|
|
api = Api(app)
|
|
|
|
|
|
class Domain():
|
|
def __init__(self, domain):
|
|
self.whois = utils.whoisData.WhoIS(domain).__dict__
|
|
self.dns = utils.dns.DNS(domain).__dict__
|
|
|
|
|
|
class DNSLookup(Resource):
|
|
def get(self, domain=None):
|
|
auth = request.headers.get('X-AUTH') # This is inverted while in dev
|
|
if auth != None:
|
|
return [{'message': 'Please auth'}, 'error']
|
|
if domain == None:
|
|
return [{'message': 'Please specify a domain name'}, 'error']
|
|
else:
|
|
ret = Domain(domain)
|
|
status = 'success'
|
|
return [ret.__dict__, status]
|
|
|
|
|
|
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')
|