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)

Answer 2

您可以使用正则表达式来匹配您的输入并测试它是否是 IP 地址

test = re.compile('\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b')
result = test.match(hostname)
if not result:
    # no match -> must be an hostname #
    convert(hostname)

该正则表达式允许无效的 IP 地址（如 999.999.999.999）所以你可能想稍微调整一下，这只是一个简单的例子

Answer 3

stackoverflow 上已经有很多关于验证 IP 地址的问题。

1. python中的IP地址验证
2. 在 python 中验证 IP 地址

当您可以在标准 python 库中执行此操作时，我想问您为什么要与子进程通信。

我建议使用一些 python 内置功能将主机名解析为 IP 地址。

您可以通过导入和使用python 套接字库来做到这一点

例如使用链接 1 中的代码：

import socket
import re
regex = re.compile(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$")
result = regex.match(address)
if not result:
    address = socket.gethostbyname(address)

Answer 4

在我的情况下，主机名只能包含-作为分隔符。因此，您可以取消注释并根据您的要求使用它。

import re

regex = "^((25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])\.){3}(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])$"
# string_check= re.compile('[@_!#$%^&*()<>?/\|}{~:.]')
string_check= re.compile('[-]')

ip_host_detail = {}
    
def is_valid_hostname_ip(IpHost):
    # pass regular expression and ip string into search() method
    
    if (re.search(regex, IpHost)):
        print("Valid Ip address")
        ip_host_detail['is_ip'] = 'True'
        ip_host_detail['is_hostname'] = 'False'
        return True
    elif(string_check.search(IpHost)):
        print("Contain hostname")
        ip_host_detail['is_hostname'] = 'True'
        ip_host_detail['is_ip'] = 'False'
            return True
    else:
        print("Invalid Ip address or hostname:- " + str(IpHost))
        ip_host_detail['is_hostname'] = 'False'
        ip_host_detail['is_ip'] = 'False'
        return False
    
    
IpHost = sys.argv[1]
# IpHost = 'RACDC1-VM123'

is_valid_hostname_ip(IpHost)
print(ip_host_detail)