Basic whois subprocess with parsing

1 files changed, 13 insertions, 0 deletions

diff --git a/whois.py b/whois.py
new file mode 100644
index 0000000..57993f4
--- /dev/null
+++ b/whois.py
@@ -0,0 +1,13 @@
+from subprocess import run
+
+def extract_field(field, response):
+    for line in response:
+        if line.startswith(field):
+            return line.split(' ')[-1]
+    
+def whois(ip):
+    cp = run(['whois', ip], capture_output=True)
+    lines = cp.stdout.decode().split('\n')
+    fields = ['route', 'origin']
+    res = {f: extract_field(f, lines) for f in fields}
+    return res