Selenium+Python 支持多项目集中配置文件管理Yaml

在自动化测试项目管理经常会遇到2个问题：
1.我手头上有多个需要做自动化的项目，我每个项目都要新建一个项目文件吗？
2.一些基础的配置信息如何高效的进行管理？

争对这2个问题，这边采用了Yaml 进行配置文件的一个管理：YAML 是一种简洁的非标记语言。YAML以数据为中心，使用空白，缩进，分行组织数据，从而使得表示更加简洁易读。

基本规则

YAML有以下基本规则：
1、大小写敏感
2、使用缩进表示层级关系
3、禁止使用tab缩进，只能使用空格键
4、缩进长度没有限制，只要元素对齐就表示这些元素属于一个层级。
5、使用#表示注释
6、字符串可以不用引号标注

其他具体的语法需要大家百度一下下了！我们直接上源码！

给个文件试例：如图右3个项目配置文件（apk;cm,gf），一个整体配置文件

读取文件源码：（file_reader.py）里面还可以放入其它文件类型的读取方法（后话）

""
文件读取。YamlReader读取yaml文件
"""
import yaml
import os
from xlrd import open_workbook


class YamlReader:
    def __init__(self, yamlf):
        if os.path.exists(yamlf):
            self.yamlf = yamlf
        else:
            raise FileNotFoundError('文件不存在！')
        self._data = None

    @property
    def data(self):
        # 如果是第一次调用data，读取yaml文档，否则直接返回之前保存的数据
        if not self._data:
            with open(self.yamlf, 'rb') as f:
                self._data = list(yaml.safe_load_all(f))  # load后是个generator，用list组织成列表
        return self._data

之后就是我们不同的调用了：（config.py）

"""
读取配置。这里配置文件用的yaml，也可用其他如XML,INI等，需在file_reader中添加相应的Reader进行处理。
"""
import os
from util_s.file_reader import YamlReader

# 通过当前文件的绝对路径，其父级目录一定是框架的base目录，然后确定各层的绝对路径。如果你的结构不同，可自行修改。
# 之前直接拼接的路径，修改了一下，用现在下面这种方法，可以支持linux和windows等不同的平台，也建议大家多用os.path.split()和os.path.join()，不要直接+'\\xxx\\ss'这样
BASE_PATH = os.path.split(os.path.dirname(os.path.abspath(__file__)))[0]
CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'config.yml')
GF_CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'gf_config.yml')
CM_CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'cm_config.yml')
APK_CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'apk_config.yml')


class Config:
    def __init__(self, config=""):
        if config == "gf":
            self.config = YamlReader(GF_CONFIG_FILE).data
        if config == "cm":
            self.config = YamlReader(CM_CONFIG_FILE).data
        if config == "apk":
            self.config = YamlReader(APK_CONFIG_FILE).data
        else:
            self.config = YamlReader(CONFIG_FILE).data

    def get(self, element, index=0):
        """
        yaml是可以通过'---'分节的。用YamlReader读取返回的是一个list，第一项是默认的节，如果有多个节，可以传入index来获取。
        这样我们其实可以把框架相关的配置放在默认节，其他的关于项目的配置放在其他节中。可以在框架中实现多个项目的测试。
        """
        return self.config[index].get(element)

最后就是每次使用的方法啦！（在其他邮件相关文章中有用到）

from util_s.config import Config

config = Config("gf")
mail_host = config.get("mail_host")
my_sender = config.get("my_sender")  # 发件人邮箱账号
my_pass = config.get("my_pass")  # 发件人邮箱密码
my_user = config.get("my_user")  # 收件人邮箱账号，我这边发送给自己

具体Yaml 文件内容贴图一个

Selenium+Python 支持多项目集中配置文件管理Yaml

基本规则

猜你喜欢