from pages.register_page import RegisterPage
from selenium import webdriver

class RegisterHandle:
    def __init__(self,driver):
        self.driver = driver
        self.driver_h = RegisterPage(self.driver)

    #输入昵称
    def send_nickname(self,nickname):
        self.driver_h.get_nickname_element().send_keys(nickname)

    #输入邮箱
    def send_email(self,email):
        self.driver_h.get_email_element().send_keys(email)

    #输入密码
    def send_password(self,password):
        self.driver_h.get_password_element().send_keys(password)

    #输入验证码
    def send_code(self,code):
        self.driver_h.get_code_element().send_keys(code)

    #点击注册
    def click_register_button(self):
        self.driver_h.get_register_button().click()

    #获取文字信息
    def get_error_text(self,error_text):
        text = self.driver_h.get_error_element().text
        if text in error_text:
            return text
        else:
            return None

    #获取登陆注册按钮
    def get_login_button_text(self):
        text = self.driver_h.get_login_button().text
        print(text)

if __name__ == '__main__':
    driver = webdriver.Chrome()
    driver.get('https://www.500d.me/register/')
    rh = RegisterHandle(driver)
    rh.get_login_button_text()

我的page页面跑过，页面找不到元素时，返回的None，说明我的find_element页面方法是没有问题的，但是handle层我要返回text的时候，就报错了