python 获取网站IP以及服务器，,写了一个识别网站的i

python 获取网站IP以及服务器，,写了一个识别网站的i

写了一个识别网站的ip以及服务器的代码，比较烂，希望大佬指教

 1 import requests 2 import sys 3 import socket 4  5 def get_ip(url):            # 获取IP 6     if ‘http‘ in str(url): 7         url = url.split(‘//‘)[1] 8         ip = socket.gethostbyname(url) 9     else:10         ip = socket.gethostbyname(url)11     return ip12 13 def get_server(url):    # 获取服务器14     if ‘http‘ in str(url):15         http_url = url16         https_url = url17     else:18         http_url = ‘http://‘ + str(url)19         https_url = ‘https://‘ + str(url)20     try:21         re_header = requests.get(http_url).headers22     except:23         re_header = requests.get(https_url).headers24     try:25         server = re_header[‘server‘]26     except: server = ‘unknow‘27     return server28 29 30 if __name__ == "__main__":31     if sys.argv[1] == ‘-r‘:32         with open(sys.argv[2]) as f:33             for line in f:34                 line = line.strip()35                 try:36                     target_ip = get_ip(line)37                     server = get_server(line)38                     print("website:", line)39                     print("-------------IP:"+target_ip+"-------------")40                     print(‘Server:‘, server,‘\n‘)41                 except : print("website:", line, "can‘t visit\n")42     else:43         url = sys.argv[1]44         try:45             target_ip = get_ip(url)46             server = get_server(url)47             print("website:", url)48             print("-------------IP:"+target_ip+"-------------")49             print(‘Server:‘, server,‘\n‘)50         except : print("website:", url, "can‘t visit\n")

用了requests，socket，sys 标准库

对单个网站进行操作：

对存放文本文件进行操作：

python 获取网站IP以及服务器