From 192757b5be73897c97c7e3c2d7223c8dfcebed6e Mon Sep 17 00:00:00 2001
From: Mikolai Fajer <mfajer@gmail.com>
Date: Fri, 23 Mar 2018 12:39:51 -0400
Subject: [PATCH] Added command-line argument instead of user prompt.

---
 ip-api.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/ip-api.py b/ip-api.py
index 048b16d..6080f40 100644
--- a/ip-api.py
+++ b/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)