方法封装

目标

1. 深入理解方法封装的思想

2. 能够使用方法封装的思想对代码进行优化

1. 方法封装

方法封装：是将一些有共性的或多次被使用的代码提取到一个方法中，供其他地方调用。封装的好处：

避免代码冗余容易维护

隐藏代码实现的细节

目的：用最少的代码实现最多的功能

2. V3版本

使用方法封装的思想，对代码进行优化。

定义获取驱动对象的工具类封装“获取弹出框的提示消息”

 

2.1 定义获取驱动对象的工具类

对登录流程的代码进行优化，定义获取驱动对象的工具

#  utils.py
 
class  DriverUtil: """
浏览器驱动工具类
"""
_driver  =  None @classmethod
def  get_driver(cls): """
获取浏览器驱动对象，并完成初始化设置
:return:  浏览器驱动对象
"""
if  cls._driver  is  None:
cls._driver  =  webdriver.Firefox() cls._driver.maximize_window() cls._driver.implicitly_wait(10) cls._driver.get("http://localhost")
return  cls._driver
 
@classmethod
def  quit_driver(cls): """
关闭浏览器驱动
"""
if  cls._driver: cls._driver.quit() cls._driver  =  None

2.2 封装“获取弹出框的提示消息”

对登录流程的代码进行优化，封装‘获取弹出框的提示消息’的方法

#  utils.py
 
def  get_tips_msg(): """
获取弹出框的提示消息
:return:  消息文本内容
"""
msg  =  DriverUtil.get_driver().find_element_by_class_name("layui-layer-content").tex
t
return  msg