Browse Source

Merge pull request #1 from mfajer/master

Added command-line argument instead of user prompt.
master
Nu2This 7 years ago
committed by GitHub
parent
commit
f1a4d29a8b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
  1. 16
      ip-api.py

16
ip-api.py

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