python批量查询网站名称

前言:最近一段时间老大给我一大批网站域名,让我帮忙查询这些网站域名对应的网站名称。作为一个IT屌丝,这种事情不可能手动一个一个去查询,于是决定写个自动化脚本去查询。

思路:网上有很多域名查询的网站,但是大部分都是要验证码的,这样的话对于编写脚本的难度就增加了很多。千辛万苦找到了一个http://icp.chinaz.com/。于是就开始编写脚本,由于近段时间在学习selenium所以就用它的框架去写。

功能:读取txt文件中的域名,然后查询域名对应的网站名称,查询结束后把结果写入excel表格里。

废话不多说,直接上代码:

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
import xlwt

class Myurltest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://icp.chinaz.com/"
        self.verificationErrors = []
        self.accept_next_alert = True
    
    def test_myurl(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.maximize_window()
        f=open("newurl.txt",'rb+')
        line=f.readlines()
        book = xlwt.Workbook(encoding='utf-8',style_compression=0)
        sheet=book.add_sheet(u'data',cell_overwrite_ok=True)
        for i in range(len(line)):
            driver.find_element_by_id("s").clear()
            driver.find_element_by_id("s").send_keys(line[i])
            driver.find_element_by_id("search").click()
            try:
                driver.implicitly_wait(0.5)
                data=driver.find_element_by_xpath("//div/ul[@id='first']/li[4]/p").text
            except NoSuchElementException,e:
                sheet.write(i,0,line[i])
                sheet.write(i,1,u"没查到")
                print u"没查到"
                continue
            print line[i],
            print data
            sheet.write(i,0,line[i])
            sheet.write(i,1,data)
        book.save('myurl.xls')
    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException, e: return False
        return True
    
    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException, e: return False
        return True
    
    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True
    
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main()


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq1124794084/article/details/51535793
文章标签: 域名查询 python
个人分类: Python selenium
上一篇selenium设置元素发现超时等待时间-python
下一篇python正则表达式split()方法详解
想对作者说点什么? 我来说一句

天天SEO网站批量查询

2010年01月14日 2.46MB 下载

Python实现批量输入ID查询成绩

2017年10月22日 1KB 下载

IP归属地批量查询

2016年09月12日 1KB 下载

没有更多推荐了,返回首页

关闭
关闭