def __init__(self): self.b = AnotherBrowser()
class Login(object): def __init__(self): self.b = AnotherBrowser() def isLogin(self): """判断是否已经登录 打开另外一个浏览器,看首页是否有填写用户名和密码的输入框 如果有,则表示未登录""" self.b.get(Parameter.HOST) login = False try: self.b.find_element_by_id('username') except: login = True return login def login(self): """登录系统 要判断是否有填写问卷调查的页面,如果有则把问卷调查先填了""" self.b.get(Parameter.HOST) try: self.b.find_element_by_id('into_system') except: username = self.b.find_element_by_id('username') username.send_keys(Parameter.USERNAME + Keys.TAB) password = self.b.find_element_by_id('password') password.send_keys(Parameter.PASSWORD + Keys.ENTER) try: profile_scale = self.b.find_element_by_id('profile_scale') # 职业类型 type_select = self.b.find_element_by_id('type_select') type_select.select_by_index(1) # 职位名称 job_title_select = self.b.find_element_by_id('job_title_select') job_title_select.select_by_index(1) # 领域 prac = self.b.find_element_by_css_selector('input[type=checkbox][class=all_1]') prac.check() # submit submit = self.b.find_element_by_id('submit') submit.click() # 判断表单是否提交成功 def is_submited(d): url = d.current_url return 'getprofile.php' not in url WebDriverWait(self.b, 60).until(is_submited) except Exception: pass def close(self): self.b.close()