Python中利用distutils.cmd实现自定义的清理命令

在Python中，我们可以使用distutils.cmd模块来自定义命令。distutils.cmd提供了一个Command类，我们可以继承这个类来实现自定义的命令。以下是一个例子，展示了如何利用distutils.cmd实现自定义的清理命令。

from distutils.core import setup
from distutils.cmd import Command
import os
import shutil

# 定义一个自定义的清理命令
class CleanCommand(Command):
    description = "Custom clean command"
    user_options = [('all', None, 'remove all')]
    
    def initialize_options(self):
        self.all = None
    
    def finalize_options(self):
        pass
    
    def run(self):
        # 删除build目录
        build_dir = os.path.join(os.getcwd(), 'build')
        if os.path.exists(build_dir):
            self._remove_dir(build_dir)
        
        # 删除dist目录
        dist_dir = os.path.join(os.getcwd(), 'dist')
        if os.path.exists(dist_dir):
            self._remove_dir(dist_dir)
        
        # 删除.egg-info目录
        egg_info_dir = self.get_finalized_command('egg_info').egg_info_dir
        if os.path.exists(egg_info_dir):
            self._remove_dir(egg_info_dir)
        
        # 删除__pycache__目录
        for root, dirs, files in os.walk(os.getcwd()):
            if '__pycache__' in dirs:
                self._remove_dir(os.path.join(root, '__pycache__'))
    
    def _remove_dir(self, dir_path):
        print(f"Removing directory {dir_path}")
        shutil.rmtree(dir_path)

# 设置setup函数
setup(
    name='my_package',
    version='1.0',
    cmdclass={
        'clean': CleanCommand
    }
)

在上面的代码中，我们创建了一个自定义的命令CleanCommand，继承自Command类。我们重写了initialize_options、finalize_options和run方法来实现自定义的清理逻辑。在run方法中，我们首先判断如果传入了--all选项，则删除所有构建的目录，否则只删除特定目录。_remove_dir方法用于删除目录。

在setup函数中，我们通过cmdclass参数注册了我们的自定义命令，这样我们就可以在命令行中使用python setup.py clean来执行自定义的清理命令。

该例子中自定义的清理命令会删除build、dist、.egg-info和__pycache__目录。

希望这个例子能帮助你理解如何利用distutils.cmd实现自定义的清理命令。