我们大概如何抽象公共代码

示例：处理不同类型文件的上传和处理

假设我们有一个应用程序，需要处理上传的不同类型的文件（如文本文件、图片和视频），每种文件类型都需要进行特定的处理和验证。这种情况下，代码中可能存在大量的重复逻辑，特别是在文件验证和保存方面。

非抽象代码示例

在这个简单示例中，我们将看到针对每种文件类型分别进行处理的代码，这些处理步骤在逻辑上非常相似，适合进行抽象。

def upload_text_file(file):
    if not file.endswith('.txt'):
        raise ValueError("Invalid file format")
    # 保存文件逻辑
    print("Text file saved")

def upload_image_file(file):
    if not file.endswith(('.png', '.jpg', '.jpeg')):
        raise ValueError("Invalid file format")
    # 保存文件逻辑
    print("Image file saved")

def upload_video_file(file):
    if not file.endswith(('.mp4', '.avi', '.mov')):
        raise ValueError("Invalid file format")
    # 保存文件逻辑
    print("Video file saved")

使用抽象重构代码

一般来说，抽象出公共代码的思路包括以下步骤：

1. 识别重复代码：查找在多个地方出现的相似代码。

2. 定义通用操作：将这些重复代码转换成一个或多个函数，这些函数可以接受参数来处理不同情况的具体细节。

3. 参数化差异：确定哪些部分是变化的，将这些部分作为参数传递给通用函数。

4. 集成与测试：将抽象的函数集成回原始代码中，并确保它们能够正常工作。

根据以上思路，我们可以重构上述代码，使其更加抽象和可重用：

def upload_file(file, valid_extensions):
    if not file.endswith(valid_extensions):
        raise ValueError("Invalid file format")
    # 保存文件逻辑
    print(f"File saved: {file}")

def main():
    upload_file("example.txt", ('.txt',))
    upload_file("example.png", ('.png', '.jpg', '.jpeg'))
    upload_file("example.mp4", ('.mp4', '.avi', '.mov'))

在这个重构后的版本中，我们创建了一个通用的upload_file函数，它接受文件和一个有效扩展名的元组作为参数。这样，我们就可以使用同一个函数来处理不同类型的文件上传，而具体的文件类型验证逻辑则通过参数传递。这种方法减少了代码冗余，提高了代码的可维护性和可拓展性。

通过这种方式，我们有效地将文件上传和验证的共通逻辑抽象出来，使得代码更加简洁并容易管理。