from flask import Flask, render_template, request 
import main as dnspy
app = Flask(__name__)
app.config['DEBUG'] = True

@app.route('/', methods=['POST', 'GET'])
def index():
    if request.method == 'POST':
        data = request.form['domain']
	print(data)
        dnsinfo = dnspy.domainWhois(str(data))
        #dnsinfo = {"." : "."}
	return render_template('dns.html', dnsinfo=dnsinfo)
    else:
        dnsinfo = {"Status" : "Please enter a domain"}
        return render_template('dns.html', dnsinfo=dnsinfo)

@app.route('/domain/', methods=['POST', 'GET'])
def getDomain():
    data = request.form['domain']
    return data

if __name__ == '__main__':
    app.run()