dnspy3-server/utils/whoisData.py
2020-05-19 09:46:32 -04:00

43 lines
1.4 KiB
Python

import whois
class WhoIS():
def __init__(self, domain):
self.domain = domain
if domain != 'blank':
self.getWhois(domain)
def getWhois(self, domain):
whoisData = whois.whois.Whois(domain).query()
whoisData = whois.parser.Parser(domain, whoisData[1]).parse()
self.registrar = whoisData.get('Registrar')
self.status = whoisData.get('Status')
self.registrant = self.buildRegistrantInfo(whoisData)
self.nameservers = self.splitNameservers(
whoisData.get('NameServer'))
self.createDate = whoisData.get('CreationDate')
self.expireDate = whoisData.get('ExpirationDate')
print(whoisData)
# if self.status == None:
# self.status = "We blacklisted bois"
def buildRegistrantInfo(self, whoisData):
ret = {}
ret['registrantName'] = whoisData.get('RegistrantName')
ret['eligibilityType'] = whoisData.get('EligibilityType')
ret['registrantID'] = whoisData.get('RegistrantID')
return ret
# I hate this, kill me now
def splitNameservers(self, nameservers):
if nameservers == None:
return
ret = []
if '\n' in nameservers[0]:
nameservers = nameservers[0]
nameservers = nameservers.split('\n')
for x in nameservers:
ret.append(x.strip())
return ret
return nameservers