Nu2This/MXLookup
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Changed the logic to look for MX records of IPs and if there is none just return location data

Browse Source
This commit is contained in:
Michael Heidelberger 2018-04-19 11:49:08 -05:00
parent 89fde516ca
commit 3904363c71
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ip-api.py
View File

@ -33,6 +33,9 @@ def findMX(host):
# append terminal output to variable std_out # append terminal output to variable std_out
for line in p: for line in p:
if re.search('not found', line):
query_api([host])
break
std_out.append(line) std_out.append(line)
p.close p.close
@ -50,8 +53,4 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("host", help="hostname to lookip") parser.add_argument("host", help="hostname to lookip")
args = parser.parse_args() args = parser.parse_args()
# If the arg is an IP address go straight to API
if re.search('[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}', args.host):
query_api([args.host])
else:
findMX(args.host) findMX(args.host)