#coding=utf-8
from time import sleep
import unittest
from selenium import webdriver
from selenium.webdriver import Remote
from selenium.webdriver.common.by import By#元素定位
import sys
def browser():#启动浏览器驱动
'''
driver = Remote(command_executor='http://127.0.0.1:4444/wd/hub',
desired_capabilities={'platform':'ANY',
'browserName':'chrome',
'version':'',
})'''
driver = webdriver.Ie()
return driver
class MyTest(unittest.TestCase):#框架
def setUp(self):
print("开始执行用例！")
self.driver = browser()
self.driver.implicitly_wait(10)
self.driver.maximize_window()
def tearDown(self):
self.driver.quit()
print('用例执行结束！')
class Page(object):#基础类
'''
页面基础类，用于所有页面的继承
'''
bbs_url = 'https://book.douban.com/'
def __init__(self,selenium_driver,base_url=bbs_url,parent=None):#初始化
self.base_url = base_url
self.driver = selenium_driver
self.timeout = 30
self.parent = parent
def on_page(self):#断言url是否相等
return self.driver.current_url == (self.base_url + self.url)
def _open(self,url):
url = self.base_url + url
self.driver.get(url)
assert self.on_page(),'获取的url和当前url不匹配.url:%s' % url
def open(self):#打开浏览器
self._open(self.url)
def find_element(self,*loc):#单个元素定位
return self.driver.find_element(*loc)
def find_elements(self,*loc):#多个元素定位
return self.driver.find_elements(*loc)
class Douban_film(Page):#页面对象封装
u'''豆瓣读书排序'''
url = '/'
#元素定位
bbs_booknum_loc = (By.XPATH,'//li/div[2]/p[1]/span[2]')#评分
bbs_bookname_loc = (By.XPATH,'//li/div[2]/h4/a')#书名
bbs_bookauther_loc = (By.XPATH,'//li/div[2]/p[2]')#作者
bbs_booktype_loc = (By.XPATH,'//li/div[2]/p[3]')#类型
bbs_bookcomments_loc = (By.XPATH,'//li/div[2]/p[5]')#评论
#元素封装
def bbs_bookname(self):#豆瓣评分
_input = self.driver.find_elements(*self.bbs_bookname_loc)# 选择页面上所有的tag name 为input 的元素
bookname = []
for i in _input:
bookname.append(i.text)#取出值并添加到数组
else:
sleep(1)
return bookname
def bbs_booknum(self):#豆瓣评分
_input = self.driver.find_elements(*self.bbs_booknum_loc)# 选择页面上所有的tag name 为input 的元素
booknum = []
for i in _input:
booknum.append(i.text)#取出值并添加到数组
else:
sleep(1)
return booknum
def bbs_bookauther(self):#作者
_input = self.driver.find_elements(*self.bbs_bookauther_loc)# 选择页面上所有的tag name 为input 的元素
bookauther = []
for i in _input:
bookauther.append(i.text)#取出值并添加到数组
else:
sleep(1)
return bookauther
def bbs_booktype(self):#类型
_input = self.driver.find_elements(*self.bbs_booktype_loc)# 选择页面上所有的tag name 为input 的元素
bookautype = []
for i in _input:
bookautype.append(i.text)#取出值并添加到数组
else:
sleep(1)
return bookautype
def bbs_bookcomments(self):#评论
_input = self.driver.find_elements(*self.bbs_bookcomments_loc)# 选择页面上所有的tag name 为input 的元素
bookaucomments = []
for i in _input:
bookaucomments.append(i.text)#取出值并添加到数组
else:
sleep(1)
return bookaucomments
def out_node(self):#排序
list1 = self.bbs_booknum()
list2 = self.bbs_bookname()
list3 = self.bbs_bookauther()
list4 = self.bbs_booktype()
list5 = self.bbs_bookcomments()
list = [[a,b,c,d,e] for a,b,c,d,e in zip(list1,list2,list3,list4,list5)]
team = list[:]
team.sort()
print('从小到大排序：%s'%team)
return team
def writetxt(self):#写入txt文件
a = self.out_node()
f1 = open('b.txt','w',encoding='utf-8')#写入文件编码格式，不写的话有时候会导致解析错误
for i in a:#循环读取内嵌列表
#if isinstance(i,list):#判断是否是一个列表
print(i)#分别打印内嵌列表里的列表
f1.writelines(i)#写入txt文件
f1.write('
')#写入文件时换行
f1.close()
class test_Douban_film(MyTest):
u'''豆瓣读书排序'''
def test1(self):
u'''豆瓣评分按降序排列并保存到txt文件'''
p1 = Douban_film(self.driver)
p1.open()
p1.writetxt()
if __name__ == '__main__':
unittest.main()