8个Python中可复用函数的最佳实践分享

# 不好的实践 - 一个函数完成多个任务
def process_data_and_save_to_db(data):
    # 处理数据
    processed_data = process(data)
    # 保存至数据库
    save_to_database(processed_data)

# 好的实践 - 将功能拆分为独立函数
def process_data(data):
    return process(data)

def save_to_db(data):
    save_to_database(data)

# 函数名不清晰，缺少文档字符串
def func(a, b):
    return a * b

# 函数名清晰，包含文档字符串
def multiply(a, b):
    """
    返回a和b的乘积。

    :param a: 第一个参数
    :param b: 第二个参数
    :return: 乘积结果
    """
    return a * b

# 不好的实践 - 参数不清晰，缺乏默认值
def generate_report(data, title):
    # 处理数据生成报告
    pass

# 好的实践 - 清晰的参数命名和默认值设置
def generate_report(data, title='Report'):
    # 处理数据生成报告
    pass

# 不好的实践 - 没有异常处理
def divide(a, b):
    return a / b

# 好的实践 - 添加异常处理
def divide(a, b):
    try:
        return a / b
    except ZeroDivisionError as e:
        print("除数不能为零：", e)

# 不好的实践 - 使用全局变量
total = 0

def add_to_total(number):
    global total
    total += number

# 好的实践 - 不使用全局变量
def add_to_total(total, number):
    return total + number

# 使用unittest模块编写测试用例
import unittest

def add(a, b):
    return a + b

class TestAddFunction(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(3, 4), 7)
        self.assertEqual(add(-1, 1), 0)

if __name__ == '__main__':
    unittest.main()

# 日志记录装饰器
def log(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print(f"函数 {func.__name__} 被调用")
        return result
    return wrapper

@log
def greet(name):
    return f"Hello, {name}!"

print(greet("Alice"))

# 使用匿名函数和高阶函数
def apply_operation(operation, x, y):
    return operation(x, y)

add = lambda a, b: a + b
multiply = lambda a, b: a * b

print(apply_operation(add, 3, 4))
print(apply_operation(multiply, 3, 4))