python爬取搜索引擎输入信息_[Python爬虫] 之二十：Selenium +phantomjs 利用 pyquery通过搜狗搜索引擎数据...

#coding=utf-8

importosimportrefrom selenium importwebdriverimportselenium.webdriver.support.ui as uiimporttimefrom datetime importdatetimeimportIniFile#from threading import Thread

from pyquery importPyQuery as pqimportLogFileimportmongoDBimporturllibclasssogouSpider(object):def __init__(self):

logfile= os.path.join(os.path.dirname(os.getcwd()), time.strftime('%Y-%m-%d') + '.txt')

self.log=LogFile.LogFile(logfile)

configfile= os.path.join(os.path.dirname(os.getcwd()), 'setting.conf')

cf=IniFile.ConfigFile(configfile)

self.webSearchUrl_list= cf.GetValue("sogou", "webSearchUrl").split(';')

self.keyword_list= cf.GetValue("section", "information_keywords").split(';')

self.db=mongoDB.mongoDbBase()

self.start_urls=[]for word inself.keyword_list:

keyword=urllib.quote(word)for url inself.webSearchUrl_list:

self.start_urls.append(url+keyword)

self.driver=webdriver.PhantomJS()

self.wait= ui.WebDriverWait(self.driver, 2)

self.driver.maximize_window()defComapre_to_days(self,leftdate, rightdate):'''比较连个字符串日期，左边日期大于右边日期多少天

:param leftdate: 格式：2017-04-15

:param rightdate: 格式：2017-04-15

:return: 天数'''l_time= time.mktime(time.strptime(leftdate, '%Y-%m-%d'))

r_time= time.mktime(time.strptime(rightdate, '%Y-%m-%d'))

result= int(l_time - r_time) / 86400

returnresultdefdate_isValid(self, strDateText):'''判断日期时间字符串是否合法：如果给定时间大于当前时间是合法，或者说当前时间给定的范围内

:param strDateText: 四种格式 '慧聪网 7小时前'; '新浪游戏 29分钟前' ; '中国行业研究网 2017-6-13'

:return: True:合法；False:不合法'''currentDate= time.strftime('%Y-%m-%d')

source= strDateText.split('')[0]if strDateText.find('分钟前') >0:returnTrue, source, currentDateelif strDateText.find('小时前') >0:

datePattern= re.compile(r'\d{1,2}')

ch= int(time.strftime('%H')) #当前小时数

strDate =re.findall(datePattern, strDateText)if len(strDate) == 1:if int(strDate[0]) <= ch: #只有小于当前小时数，才认为是今天

returnTrue, source, currentDateelse:

datePattern= re.compile(r'\d{4}-\d{1,2}-\d{1,2}')

strDate=re.findall(datePattern, strDateText)if len(strDate) == 1:if self.Comapre_to_days(currentDate, strDate[0]) ==0:returnTrue, source, currentDatereturn False, '', ''

deflog_print(self, msg):'''# 日志函数

# :param msg: 日志信息

# :return:

#'''

print '%s: %s' % (time.strftime('%Y-%m-%d %H-%M-%S'), msg)defscrapy_date(self):

strsplit= '------------------------------------------------------------------------------------'

for link inself.start_urls:

self.driver.get(link)

selenium_html= self.driver.execute_script("return document.documentElement.outerHTML")

doc=pq(selenium_html)

infoList=[]

self.log.WriteLog(strsplit)

self.log_print(strsplit)

Elements= doc('div[class="news151102"]')for element inElements.items():

strdate= element('p[class="news-from"]').text().encode('utf8').strip()

flag, source, date=self.date_isValid(strdate)ifflag:

title= element('h3[class="vrTitle"]').find('a').text().encode('utf8').strip()for keyword inself.keyword_list:if title.find(keyword) > -1:

url= element('h3[class="vrTitle"]').find('a').attr('href')

dictM= {'title': title, 'date': date,'url': url, 'keyword': keyword, 'introduction': title, 'source': source}

infoList.append(dictM)

self.log.WriteLog('title:%s' %title)

self.log.WriteLog('url:%s' %url)

self.log.WriteLog('source:%s' %source)

self.log.WriteLog('kword:%s' %keyword)

self.log.WriteLog(strsplit)

self.log_print('title:%s' % dictM['title'])

self.log_print('url:%s' % dictM['url'])

self.log_print('date:%s' % dictM['date'])

self.log_print('source:%s' % dictM['source'])

self.log_print('kword:%s' % dictM['keyword'])

self.log_print(strsplit)break

if len(infoList)>0:

self.db.SaveInformations(infoList)

self.driver.close()

self.driver.quit()

obj=sogouSpider()

obj.scrapy_date()