summaryrefslogtreecommitdiff
path: root/whois.py
blob: 57993f40abca9cf1a679120e65712b37b572cb6e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
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