summaryrefslogtreecommitdiff
path: root/whois.py
diff options
context:
space:
mode:
Diffstat (limited to 'whois.py')
-rw-r--r--whois.py4
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