From 4c81e5f7614946a918d36dbfdccf27e7110ad3ea Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Fri, 12 Oct 2018 00:13:17 +0200
Subject: Cleanup whois parsing

---
 whois.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'whois.py')

diff --git a/whois.py b/whois.py
index 57993f4..2681db0 100644
--- a/whois.py
+++ b/whois.py
@@ -3,11 +3,11 @@ from subprocess import run
 def extract_field(field, response):
     for line in response:
         if line.startswith(field):
-            return line.split(' ')[-1]
+            return line.split('    ')[-1].strip()
     
 def whois(ip):
     cp = run(['whois', ip], capture_output=True)
     lines = cp.stdout.decode().split('\n')
-    fields = ['route', 'origin']
+    fields = ['route', 'origin', 'descr']
     res = {f: extract_field(f, lines) for f in fields}
     return res
-- 
cgit v1.3.1