菜鸟求助！！ webdriver基于python，封装登录脚本时报错
本帖最后由 kettle_hou 于 2016-7-21 16:27 编辑如题！我想把打开浏览器和用户登录作为两个class，放在同一个模块下使用。在调用用户登录类时，会报错！！
猜想：在def denglu(self):下的应该要调用class browser():创建的driver对象，不知道如何实现。请各位高手指点，谢谢。
代码如下：
#####new.py#####
from public import browser
from public import login
from selenium import webdriver
browser().firefox("此处是url")
yonghu = login("此处是username","此处是password")
yonghu.denglu()
#####public.py#####
from selenium import webdriver
import os
class browser():
# webdriver for chrome
def chrome(self,url,driverpath):
os.environ["webdriver.chrome.driver"] = driverpath
driver = webdriver.Chrome()
driver.get(url)
# webdriver for firefox
def firefox(self,url):
driver = webdriver.Firefox()
driver.get(url)
#login
class login():
def __init__(self,username,password):
self.name = username
self.password = password
def denglu(self):
driver = webdriver()
driver.find_element_by_xpath("//*[@id='TANGRAM__PSP_4__userName']").send_keys(self.name)
driver.find_element_by_id("TANGRAM__PSP_4__password").send_keys(self.password)
driver.find_element_by_xpath("//*[@id='TANGRAM__PSP_4__submit']").click()
text = driver.find_element_by_id("TANGRAM__PSP_4__error").text
报错内容！！
Traceback (most recent call last):
File "F:\Workplace\Python auto testing\src\new1\__init__.py", line 6, in <module>
yonghu = login("此处是username","此处是password")
File "F:\Workplace\Python auto testing\src\new1\public.py", line 19, in __init__
a = self.driver
AttributeError: login instance has no attribute 'driver'
坐等 让denglu（）这个函数带参，将browser().firefox("此处是url") 的driver的值传参进去。 在Browser类中把你创建的driver return出来，或者写个方法getDriver得到driver，然后再登陆类中调用getDriver方法就可以使用了。 清晨一缕阳光 发表于 2016-7-21 17:26
在Browser类中把你创建的driver return出来，或者写个方法getDriver得到driver，然后再登陆类中调用getDriv ...
在Browser类的函数下加了return后！请问我在denglu()这个方法里，怎么获得返回的driver？
driver = self.driver ???? 一个类中的Get ，Set 方法不会吗？
页:
[1]
