def click_sel(self, selector):
self.logger = Logger(loggername=self.classname).get_logger()
el = self.find_element(selector)
try:
self.logger.info("The element {} was clicked." .format(el.text))
el.click()
except NameError as e:
self.logger.error("Failed to click the element with {}".format(e))
class Picture(Basepage):
def rec_create_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到新建广告位按钮')
return self.find_element('css_selector=>div.panel_page_button_text')
def click_create_btn(self):
self.rec_create_btn().click()
self.logger.debug(u'点击新建广告位按钮')
def set_value(self,element,text):
self.logger = Logger(loggername=self.classname).get_logger()
element.clear()
try:
element.send_keys(text)
self.logger.info("Had type {} in inputBox".format(text))
except NameError as e:
self.logger.error("Failed to type in input box with {}".format(e))
self.get_windows_img()
def clear(self,selector):
self.logger = Logger(loggername=self.classname).get_logger()
el = self.find_element(selector)
try:
el.clear()
self.logger.info("Clear text in input box before typing.")
except NameError as e:
self.logger.error("Failed to clear in input box with {}".format(e))
self.get_windows_img()
class AdPosition(Basepage):
def rec_picture_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到图文按钮')
return self.find_element('css_selector=>div#adPos_adKind_tuwen')
def click_picture_btn(self):
self.rec_picture_btn().click()
self.logger.debug(u'点击图文按钮')
return Picture(self.driver)
def find_element(self,selector):
self.logger = Logger(loggername=self.classname).get_logger()
element = ''
if '=>' not in selector:
return self.driver.find_element_by_id(selector)
selector_by = selector.split('=>')[0]
selector_value = selector.split('=>')[1]
if selector_by == "i" or selector_by == 'id':
try:
element = self.driver.find_element_by_id(selector_value)
self.logger.info("Had find the element {}successful "
"by {} via value: {} ".format(element.text, selector_by, selector_value))
except NoSuchElementException as e:
self.logger.error("NoSuchElementException: {}".format(e))
self.get_windows_img() # take screenshot
elif selector_by == "n" or selector_by == 'name':
element = self.driver.find_element_by_name(selector_value)
elif selector_by == "c" or selector_by == 'class_name':
element = self.driver.find_element_by_class_name(selector_value)
elif selector_by == "l" or selector_by == 'link_text':
element = self.driver.find_element_by_link_text(selector_value)
elif selector_by == "p" or selector_by == 'partial_link_text':
element = self.driver.find_element_by_partial_link_text(selector_value)
elif selector_by == "t" or selector_by == 'tag_name':
element = self.driver.find_element_by_tag_name(selector_value)
elif selector_by == "x" or selector_by == 'xpath':
try:
element = self.driver.find_element_by_xpath(selector_value)
self.logger.info("Had find the element {} successful "
"by {} via value: {} ".format(element.text, selector_by, selector_value))
except NoSuchElementException as e:
self.logger.error("NoSuchElementException: {}".format(e))
self.get_windows_img()
elif selector_by == "s" or selector_by == 'css_selector':
element = self.driver.find_element_by_css_selector(selector_value)
else:
raise NameError("Please enter a valid type of targeting elements.")
return element
class SearchPage(Basepage):
def rec_search_input(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到搜索Input.')
return self.find_element(selector='id=>kw')
def rec_search_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到百度一下Input.')
return self.find_element(selector='id=>su')
def click_search_btn(self):
self.rec_search_btn().click()
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'点击"立即登录 "按钮.')
def rec_picture_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到图文按钮')
return self.find_element('css_selector=>div#adPos_adKind_tuwen')
def click_login_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'点击"立即登录 "按钮.')
self.click_sel(selector='id=>login_dialog_login_button')
def rec_login_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"立即登录"按钮.')
return self.find_element(selector='id=>login_dialog_login_button')
def rec_search_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到百度一下Input.')
return self.find_element(selector='id=>su')
class LoginPage(Basepage):
def rec_user_input(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"登录用户名"输入input.')
return self.find_element(selector='id=>login_dialog_input_user_mail')
def rec_passwd_input(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"密码"输入input.')
return self.find_element(selector='id=>login_dialog_input_user_pwd')
def rec_rmb_user_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"记住用户名"按钮.')
return self.find_element(selector='id=>login_dialog_rmb_user')
def rec_rmb_pwd_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"记住密码"按钮.')
return self.find_element(selector='id=>login_dialog_rmb_pwd')
def rec_login_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"立即登录"按钮.')
return self.find_element(selector='id=>login_dialog_login_button')
def click_login_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'点击"立即登录 "按钮.')
self.click_sel(selector='id=>login_dialog_login_button')
def get_main_page(self):
return MainPage(self.driver)
def sleep(self,seconds):
self.logger = Logger(loggername=self.classname).get_logger()
time.sleep(seconds)
self.logger.info("Sleep for {} seconds" .format(seconds))
def get_page_title(self):
self.logger = Logger(loggername=self.classname).get_logger()
self.logger.info("Current page title is {}".format(self.driver.title))
return self.driver.title
def rec_create_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到新建广告位按钮')
return self.find_element('css_selector=>div.panel_page_button_text')
def rec_passwd_input(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"密码"输入input.')
return self.find_element(selector='id=>login_dialog_input_user_pwd')
def rec_rmb_pwd_btn(self):
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'找到"记住密码"按钮.')
return self.find_element(selector='id=>login_dialog_rmb_pwd')
def click_search_btn(self):
self.rec_search_btn().click()
self.logger = Logger(loggername=self.__class__.__name__).get_logger()
self.logger.debug(u'点击"立即登录 "按钮.')
class Basepage(object):
url = None
''''
定义一个基类，封装常用的页面方法
'''
def __init__(self,driver):
self.driver = driver
self.classname = self.__class__.__name__
#浏览器最大化
def max_browser(self):
self.driver.maximize_window()
#设置浏览器的宽,高
def set_width(self,width,height):
self.driver.set_window_size(width,height)
#关闭浏览器
def quit_browser(self):
self.driver.quit()
#浏览器前进
def forward(self):
self.driver.forward()
#浏览器后退
def back(self):
self.driver.back()
# 隐式等待
def wait(self,seconds):
self.driver.implicitly_wait(seconds)
# 关闭当前窗口
def close(self):
try:
self.driver.close()
except NameError as e:
print('Failed to close')
def visit(self):
self.driver.get(self.url)
# 保存图片
def get_windows_img(self):
file_path = r'D:/python/Daemon_/report/screenpicture/'
rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
screen_name = file_path + rq + '.png'
try:
self.driver.get_screenshot_as_file(screen_name)
self.logger.info("Had take screenshot and save to folder : /screenshots")
print(screen_name)
except NameError as e:
self.logger.error("Failed to take screenpicture! {}".format(e))
self.get_windows_img()
# 定位元素方法
def find_element(self,selector):
self.logger = Logger(loggername=self.classname).get_logger()
element = ''
if '=>' not in selector:
return self.driver.find_element_by_id(selector)
selector_by = selector.split('=>')[0]
selector_value = selector.split('=>')[1]
if selector_by == "i" or selector_by == 'id':
try:
element = self.driver.find_element_by_id(selector_value)
self.logger.info("Had find the element {}successful "
"by {} via value: {} ".format(element.text, selector_by, selector_value))
except NoSuchElementException as e:
self.logger.error("NoSuchElementException: {}".format(e))
self.get_windows_img() # take screenshot
elif selector_by == "n" or selector_by == 'name':
element = self.driver.find_element_by_name(selector_value)
elif selector_by == "c" or selector_by == 'class_name':
element = self.driver.find_element_by_class_name(selector_value)
elif selector_by == "l" or selector_by == 'link_text':
element = self.driver.find_element_by_link_text(selector_value)
elif selector_by == "p" or selector_by == 'partial_link_text':
element = self.driver.find_element_by_partial_link_text(selector_value)
elif selector_by == "t" or selector_by == 'tag_name':
element = self.driver.find_element_by_tag_name(selector_value)
elif selector_by == "x" or selector_by == 'xpath':
try:
element = self.driver.find_element_by_xpath(selector_value)
self.logger.info("Had find the element {} successful "
"by {} via value: {} ".format(element.text, selector_by, selector_value))
except NoSuchElementException as e:
self.logger.error("NoSuchElementException: {}".format(e))
self.get_windows_img()
elif selector_by == "s" or selector_by == 'css_selector':
element = self.driver.find_element_by_css_selector(selector_value)
else:
raise NameError("Please enter a valid type of targeting elements.")
return element
# 文本框输入
def set_value(self,element,text):
self.logger = Logger(loggername=self.classname).get_logger()
element.clear()
try:
element.send_keys(text)
self.logger.info("Had type {} in inputBox".format(text))
except NameError as e:
self.logger.error("Failed to type in input box with {}".format(e))
self.get_windows_img()
# 文本框清除
def clear(self,selector):
self.logger = Logger(loggername=self.classname).get_logger()
el = self.find_element(selector)
try:
el.clear()
self.logger.info("Clear text in input box before typing.")
except NameError as e:
self.logger.error("Failed to clear in input box with {}".format(e))
self.get_windows_img()
# 点击元素
def click_sel(self, selector):
self.logger = Logger(loggername=self.classname).get_logger()
el = self.find_element(selector)
try:
self.logger.info("The element {} was clicked." .format(el.text))
el.click()
except NameError as e:
self.logger.error("Failed to click the element with {}".format(e))
# 获取网页标题
def get_page_title(self):
self.logger = Logger(loggername=self.classname).get_logger()
self.logger.info("Current page title is {}".format(self.driver.title))
return self.driver.title
#sleep
def sleep(self,seconds):
self.logger = Logger(loggername=self.classname).get_logger()
time.sleep(seconds)
self.logger.info("Sleep for {} seconds" .format(seconds))
#
def switch_alert(self):
alert = self.driver.switch_to_alert()
alert.accept()
#frame切换
def switch_frame(self,elem):
pass