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

Added command-line argument instead of user prompt.

Browse Source
This commit is contained in:
Mikolai Fajer 2018-03-23 12:39:51 -04:00
parent 6805f153d2
commit 192757b5be
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ip-api.py
View File

@ -1,11 +1,12 @@
import argparse
import json import json
import requests import requests
import os import os
def search(): def search(host):
main_api = 'http://ip-api.com/json/' main_api = 'http://ip-api.com/json/'
ip = findMX() ip = findMX(host)
for host in ip: for host in ip:
json_data = requests.get(main_api + host).json() json_data = requests.get(main_api + host).json()
@ -17,8 +18,7 @@ def search():
json_data['query'], json_data['query'],
host)) host))
def findMX(): def findMX(host):
host = input("Who do you want to look up?: ")
p = os.popen('host -t MX ' + host) p = os.popen('host -t MX ' + host)
#initialize dicts #initialize dicts
@ -39,4 +39,10 @@ def findMX():
MXServer.append(split[6]) MXServer.append(split[6])
i = i + 1 i = i + 1
return MXServer return MXServer
search()
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("host", help="hostname to lookip")
args = parser.parse_args()
search(args.host)