Cleanup whois parsing

1 files changed, 2 insertions, 2 deletions

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