Moved data classes to there own files

This commit is contained in:
Benjamyn Love 2020-05-12 13:27:40 -04:00
parent 98442178ac
commit 191ab4ed66
5 changed files with 33 additions and 31 deletions

3
.gitignore vendored
View File

@ -1,2 +1,3 @@
env/ env/
.vscode/ .vscode/
*.pyc

View File

@ -1,43 +1,16 @@
from flask import Flask, request from flask import Flask, request
from flask_restful import Resource, Api from flask_restful import Resource, Api
import utils
import json import json
app = Flask(__name__) app = Flask(__name__)
api = Api(app) 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']
class DNS():
def __init__(self, domain):
self.domain = domain
self.getRecords()
def getRecords(self):
self.a = '1.2.3.4'
self.aaaa = '2001:0db8:85a3:0000:0000:8a2e:0370:7334'
self.mx = {'mail.domain.tld': 10}
self.txt = 'v=spf1 +a +mx +include:spf.hostingplatform.net.au'
self.ns = ['ns1.domain.tld', 'ns2.domain.tld']
self.soa = 'ns1.domain.tld. webmaster.domain.tld. 310875860 900 900 1800 60'
class Domain(): class Domain():
def __init__(self, domain): def __init__(self, domain):
self.whois = WhoIS(domain).__dict__ self.whois = utils.whois.WhoIS(domain).__dict__
self.dns = DNS(domain).__dict__ self.dns = utils.dns.DNS(domain).__dict__
class DNSLookup(Resource): class DNSLookup(Resource):
@ -55,6 +28,8 @@ class DNSLookup(Resource):
api.add_resource(DNSLookup, '/lookup', '/lookup/<string:domain>') api.add_resource(DNSLookup, '/lookup', '/lookup/<string:domain>')
print(dir())
@app.route('/') @app.route('/')
def index(): def index():

2
utils/__init__.py Normal file
View File

@ -0,0 +1,2 @@
from . import dns
from . import whois

12
utils/dns.py Normal file
View File

@ -0,0 +1,12 @@
class DNS():
def __init__(self, domain):
self.domain = domain
self.getRecords()
def getRecords(self):
self.a = '1.2.3.4'
self.aaaa = '2001:0db8:85a3:0000:0000:8a2e:0370:7334'
self.mx = {'mail.domain.tld': 10}
self.txt = 'v=spf1 +a +mx +include:spf.hostingplatform.net.au'
self.ns = ['ns1.domain.tld', 'ns2.domain.tld']
self.soa = 'ns1.domain.tld. webmaster.domain.tld. 310875860 900 900 1800 60'

12
utils/whois.py Normal file
View File

@ -0,0 +1,12 @@
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']