Python全局方法定义及自动化登录代码优化问题咨询

嘿，我来帮你一步步解决这两个问题！

在Python里，有几种实用的方式可以实现这个需求，结合你的自动化项目场景，我重点推荐两种：

1. 用类封装（最适合你的项目） #

因为你后续的代码是基于类编写的，把变量作为实例属性会更规范，避免全局变量的混乱。你可以在setup方法里初始化这些变量，然后通过self.变量名让类里的其他方法直接调用：

class Demo:
    def setup(self):
        # 把需要共享的变量定义为实例属性
        self.driver = webdriver.Chrome("C:\\temp\\chromedriver.exe")
        self.url = 'site-address'
        self.username = 'Name'
        self.password = 'pas'
        self.wait = WebDriverWait(self.driver, 120)
        # 初始化页面操作
        self.driver.get(self.url)
        self.driver.maximize_window()
    
    def func1(self):
        # 直接用self访问setup里的变量
        user = self.driver.find_element_by_xpath("//")
        user.send_keys(self.username)
        # 其他操作...

2. 全局变量方式（适合简单场景） #

如果是小型脚本，也可以在方法外部定义全局变量，在方法内部用global关键字声明后修改，其他方法就能直接访问：

# 定义全局变量
global_username = "default"

def setup_global():
    global global_username  # 声明要使用全局变量
    global_username = "Humble_boy"

def use_var():
    print(f"当前用户名：{global_username}")

# 调用顺序
setup_global()
use_var()  # 输出：当前用户名：Humble_boy

你的核心痛点是重复代码冗余和首次登录失败的重试逻辑，我帮你重构代码，把初始化、登录操作都封装成独立方法，彻底解决这些问题：

重构后的完整代码 #

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

class Demo:
    def setup(self):
        # 一次性初始化所有公共变量，作为实例属性
        self.driver = webdriver.Chrome("C:\\temp\\chromedriver.exe")
        self.url = 'site-address'
        self.username = 'Name'
        self.password = 'pas'
        self.wait = WebDriverWait(self.driver, 120)
        # 初始化页面
        self.driver.get(self.url)
        self.driver.maximize_window()
    
    def perform_login(self):
        # 把登录操作单独封装，重复调用时直接用
        user = self.driver.find_element_by_xpath("//")  # 替换成你实际的xpath
        user.clear()  # 避免重试时输入框内容叠加
        user.send_keys(self.username)
        
        pas = self.driver.find_element_by_xpath("//")  # 替换成你实际的xpath
        pas.clear()
        pas.send_keys(self.password)
        
        btn = self.driver.find_element_by_xpath("//")  # 替换成你实际的xpath
        btn.click()
    
    def func1(self):
        self.setup()  # 先执行初始化
        # 第一次尝试登录
        self.perform_login()
        
        try:
            # 等待验证元素出现
            chk = self.wait.until(EC.presence_of_element_located((By.ID, 'lop')))
            assert chk.text == "Some Text"
            print('PAGE VERIFIED : ', chk.text)
        except:
            print('LOGIN UN-SUCCESSFUL')
            print('Re-Trying...')
            # 直接调用封装好的登录方法重试
            self.perform_login()
            # 重试后再验证一次，确保结果明确
            try:
                chk = self.wait.until(EC.presence_of_element_located((By.ID, 'lop')))
                assert chk.text == "Some Text"
                print('RETRY LOGIN SUCCESSFUL : ', chk.text)
            except:
                print('RETRY LOGIN FAILED')
                self.driver.quit()

# 运行测试
if __name__ == "__main__":
    test = Demo()
    test.func1()

关键优化说明 #

• setup方法：一次性完成driver、URL、账号密码等所有公共资源的初始化，后续方法直接通过self调用，不用重复定义。
• perform_login方法：把登录的核心操作（输入账号、密码、点击按钮）封装起来，不管第一次登录还是重试，都只需调用这个方法，彻底消除重复代码。
• 增加clear()操作：输入账号密码前清空输入框，避免重试时输入框已有内容导致输入错误。
• 完善重试验证：在重试后再次验证页面，让测试结果更明确，避免重试后不知道是否成功。
• 结构更清晰：每个方法只负责一件事，后续维护和扩展会更方便。

内容的提问来源于stack exchange，提问作者Humble_boy