(相关资料图)
前言
由于公司有大量域名信息需要定期查看是否需要续期，前期都是人工操作比较耗时、耗力。所以衍生了这个小工具。
实现了查询域名到期时间、并且将近7天内到期的域名在Excel中标红，当然你也可以添加短信提醒和邮件提醒
代码步骤
1、将域名粘贴到指定txt文件中
比如：domain.txt
2、将指定txt文件中内容读取到list中
# 批量读取文件中的域名def read_file(filePath): with open(filePath, "r") as f: # 打开文件 data = f.readlines() # 读取文件 return data3、通过某网站获取域名到期时间
# 通过某网站获取域名到期时间def get_expiry_date(url_list): url_expiry_date_list = [] for url in url_list: url_expiry_date_dict = {} time.sleep(random.randrange(3)) req_whois = urllib.request.urlopen("http://whois.xxxxxx.com/" + url) result = req_whois.read().decode() html = etree.HTML(result) endTimes = html.xpath("//a[@id="update_a2"]/preceding-sibling::span[1]/text()") if len(endTimes) > 0: endTime = endTimes[0].replace("年", "-").replace("月", "-").replace("日", "") else: errorInfo = html.xpath("//div[@class="IcpMain02"]") endTime = errorInfo[0].xpath("string(.)").strip() url_expiry_date_dict["url"] = url.replace("n", "") url_expiry_date_dict["endTime"] = endTime pprint.pprint(url_expiry_date_dict) url_expiry_date_list.append(url_expiry_date_dict) pprint.pprint(url_expiry_date_list) return url_expiry_date_list4、将结果写入Excel文件
# 写入Excel文件def write_excel(domain_list): # 创建一个新的文件 with xlsxwriter.Workbook("host_ip.xlsx") as workbook: # 添加一个工作表 worksheet = workbook.add_worksheet("域名信息") # 设置一个加粗的格式 bold = workbook.add_format({"bold": True}) # 分别设置一下 A 和 B 列的宽度 worksheet.set_column("A:A", 50) worksheet.set_column("B:B", 15) # 先把表格的抬头写上，并设置字体加粗 worksheet.write("A1", "域名", bold) worksheet.write("B1", "信息", bold) # 设置数据写入文件的初始行和列的索引位置 row = 1 col = 0 for domain_ex_date in domain_list: url = domain_ex_date["url"] endTime = domain_ex_date["endTime"] currDate = datetime.today().date() try: endDate = datetime.strptime(endTime, "%Y-%m-%d").date() diffDate = endDate - currDate if diffDate.days <= 7: style = workbook.add_format({"font_color": "red"}) else: style = workbook.add_format({"font_color": "black"}) except: style = workbook.add_format({"font_color": "red"}) pprint.pprint(url + ": " + endTime) worksheet.write(row, col, url, style) worksheet.write(row, col + 1, endTime, style) row += 15、运行
urls = read_file("domain.txt")urls_list = get_expiry_date(urls)write_excel(urls_list)运行结果：
6、完整代码
#!/usr/bin/env python# -*- coding: utf-8 -*-# Author:高效码农import pprintimport timeimport randomimport xlsxwriterfrom datetime import datetimeimport urllib.requestfrom lxml import etree# 批量读取文件中的域名def read_file(filePath): with open(filePath, "r") as f: # 打开文件 data = f.readlines() # 读取文件 return data# 通过某网站获取域名到期时间def get_expiry_date(url_list): url_expiry_date_list = [] for url in url_list: url_expiry_date_dict = {} time.sleep(random.randrange(3)) req_whois = urllib.request.urlopen("http://whois.xxxxxx.com/" + url) result = req_whois.read().decode() html = etree.HTML(result) endTimes = html.xpath("//a[@id="update_a2"]/preceding-sibling::span[1]/text()") if len(endTimes) > 0: endTime = endTimes[0].replace("年", "-").replace("月", "-").replace("日", "") else: errorInfo = html.xpath("//div[@class="IcpMain02"]") endTime = errorInfo[0].xpath("string(.)").strip() url_expiry_date_dict["url"] = url.replace("n", "") url_expiry_date_dict["endTime"] = endTime pprint.pprint(url_expiry_date_dict) url_expiry_date_list.append(url_expiry_date_dict) pprint.pprint(url_expiry_date_list) return url_expiry_date_list# 写入Excel文件def write_excel(domain_list): # 创建一个新的文件 with xlsxwriter.Workbook("host_ip.xlsx") as workbook: # 添加一个工作表 worksheet = workbook.add_worksheet("域名信息") # 设置一个加粗的格式 bold = workbook.add_format({"bold": True}) # 分别设置一下 A 和 B 列的宽度 worksheet.set_column("A:A", 50) worksheet.set_column("B:B", 15) # 先把表格的抬头写上，并设置字体加粗 worksheet.write("A1", "域名", bold) worksheet.write("B1", "信息", bold) # 设置数据写入文件的初始行和列的索引位置 row = 1 col = 0 for domain_ex_date in domain_list: url = domain_ex_date["url"] endTime = domain_ex_date["endTime"] currDate = datetime.today().date() try: endDate = datetime.strptime(endTime, "%Y-%m-%d").date() diffDate = endDate - currDate if diffDate.days <= 7: style = workbook.add_format({"font_color": "red"}) else: style = workbook.add_format({"font_color": "black"}) except: style = workbook.add_format({"font_color": "red"}) pprint.pprint(url + ": " + endTime) worksheet.write(row, col, url, style) worksheet.write(row, col + 1, endTime, style) row += 1urls = read_file("domain.txt")urls_list = get_expiry_date(urls)write_excel(urls_list)关键词：
责任编辑：Rex_29
