python - 如何查找用户是否输入了主机名或 IP 地址？

Question

用户将输入主机名或 IP 地址。如果用户输入 IP 地址，我想保持原样，但如果用户输入主机名，我想使用以下方法将其转换为 IP 地址:

def convert(hostname):
    command = subprocess.Popen(['host', hostname],
                           stdout=subprocess.PIPE).communicate()[0]

    progress1 = re.findall(r'\d+.', command)
    progress1 = ''.join(progress1)
    return progress1 

我该怎么做？

Answer 1

无论输入的是 ip 还是主机名，获取 ip:

ip4 = socket.gethostbyname(ip4_or_hostname)