第2章：Python基础回顾

好，咱们直接进入正题。这一章我打算带大家快速过一遍Python的核心基础。别觉得枯燥，这些内容在自动化框架里天天都要用。我自己带团队时发现，很多同学框架写不好，根子就在基础不牢。

2.1 Python环境搭建

先说环境。我个人习惯用pyenv来管理Python版本。为什么？因为项目多了你就知道，有的依赖Python 3.8，有的要3.10，手动切换太痛苦了。

安装完记得验证一下：

python --version
pip --version

2.2 基础语法速览

Python的语法，说白了就是「优雅」。我见过太多人把Python写成Java，那真叫一个难受。

变量与数据类型

Python是动态类型，你不需要声明类型。但注意，这不代表你可以乱来。

name = "自动化框架"  # 字符串
version = 2.0        # 浮点数
is_ready = True      # 布尔值
modules = []         # 列表

控制流

if-else、for、while，这些大家都熟。我重点说一个：列表推导式。这玩意儿写好了，代码能少一半。

# 传统写法
result = []
for i in range(10):
    if i % 2 == 0:
        result.append(i * 2)

# 推导式写法
result = [i * 2 for i in range(10) if i % 2 == 0]

2.3 面向对象编程

自动化框架里，面向对象是核心。你想想看，一个测试用例、一个页面对象、一个配置管理，哪个不是类？

类与实例

class TestCase:
    def __init__(self, name, steps):
        self.name = name
        self.steps = steps
    
    def run(self):
        print(f"执行测试用例：{self.name}")
        for step in self.steps:
            step.execute()

继承与多态

我在项目中遇到过这样的场景：有Web测试、API测试、移动端测试，但它们的执行流程很像。这时候用继承就对了。

class BaseTest:
    def setup(self):
        print("初始化环境")
    
    def teardown(self):
        print("清理环境")
    
    def execute(self):
        self.setup()
        self.run_test()
        self.teardown()

class WebTest(BaseTest):
    def run_test(self):
        print("执行Web测试")

class APITest(BaseTest):
    def run_test(self):
        print("执行API测试")

2.4 异常处理

自动化测试跑起来，什么妖魔鬼怪都可能遇到。网络超时、元素找不到、数据格式不对...没有异常处理，脚本就是一次性用品。

try-except 结构

try:
    element = driver.find_element(By.ID, "submit")
    element.click()
except NoSuchElementException:
    print("元素没找到，截图保存")
    driver.save_screenshot("error.png")
except TimeoutException:
    print("页面加载超时，重试")
    driver.refresh()
except Exception as e:
    print(f"未知错误：{e}")
    raise
finally:
    print("清理资源")

自定义异常

框架大了，内置异常不够用。我习惯定义自己的异常类：

class FrameworkException(Exception):
    """框架基础异常"""
    pass

class ConfigError(FrameworkException):
    """配置错误"""
    pass

class TestDataError(FrameworkException):
    """测试数据错误"""
    pass

2.5 文件操作

自动化框架离不开文件：配置文件、测试数据、日志、报告...全是文件。

读写文本文件

with语句是标配，它会自动关闭文件，省心。

# 读文件
with open("config.ini", "r", encoding="utf-8") as f:
    content = f.read()

# 写文件
with open("report.html", "w", encoding="utf-8") as f:
    f.write("<h1>测试报告</h1>")

JSON操作

自动化框架里，JSON是最常用的数据格式。配置、测试数据、接口响应，全是JSON。

import json

# 读取JSON配置
with open("config.json", "r") as f:
    config = json.load(f)

# 写入测试结果
result = {"passed": 10, "failed": 2, "skipped": 1}
with open("result.json", "w") as f:
    json.dump(result, f, indent=2, ensure_ascii=False)

class FileUtil:
    @staticmethod
    def read_json(path):
        with open(path, "r") as f:
            return json.load(f)
    
    @staticmethod
    def write_json(path, data):
        with open(path, "w") as f:
            json.dump(data, f, indent=2)

路径处理

路径问题坑过我好几次。Windows用反斜杠，Linux用正斜杠，硬编码就是找死。

import os
from pathlib import Path

# 推荐用pathlib
base_dir = Path(__file__).parent
config_path = base_dir / "config" / "test_config.json"

# 或者用os.path
config_path = os.path.join(os.path.dirname(__file__), "config", "test_config.json")