From 7c5cc0d1fe9cb8ac44848827fdcf78691642c98f Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Fri, 12 Oct 2018 00:08:27 +0200
Subject: Basic whois subprocess with parsing

---
 whois.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)
 create mode 100644 whois.py

(limited to 'whois.py')

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
-- 
cgit v1.3.1