基于Python Watchdog库实现文件系统监控_Python

watchdog是一个优秀的python库，用于监控文件系统事件。它可以检测文件或目录的创建、修改、删除和移动等操作，并触发相应的回调函数。本文将介绍watchdog的基本用法，并通过一个实际案例展示如何用它来监控下载目录并自动转换ico文件为png格式。

watchdog简介

watchdog库的主要组件包括：

observer - 监控文件系统变化的核心类
filesystemeventhandler - 处理文件系统事件的基类
各种事件类 - 如filecreatedevent, filemodifiedevent等

安装watchdog

pip install watchdog

实际案例：自动转换ico文件为png

下面是一个完整的示例，它监控用户的下载目录，当检测到新的ico文件时，自动将其转换为png格式（使用imagemagick的convert工具）。

import os
import time
import subprocess
import shutil
from watchdog.observers import observer
from watchdog.events import filesystemeventhandler

class icohandler(filesystemeventhandler):
    def __init__(self, download_dir, convert_exe):
        self.download_dir = download_dir
        self.processed_files = set()
        self.convert_exe = convert_exe
        
        # 初始化时处理已存在的ico文件
        for filename in os.listdir(download_dir):
            if filename.lower().endswith('.ico'):
                self.processed_files.add(filename)
                self.convert_ico_to_png(filename)
    
    def on_created(self, event):
        if not event.is_directory and event.src_path.lower().endswith('.ico'):
            filename = os.path.basename(event.src_path)
            if filename not in self.processed_files:
                self.processed_files.add(filename)
                print(f"检测到新的ico文件: {filename}")
                self.convert_ico_to_png(filename)
    
    def convert_ico_to_png(self, ico_filename):
        ico_path = os.path.join(self.download_dir, ico_filename)
        base_name = os.path.splitext(ico_filename)[0]
        
        # 临时输出目录
        temp_dir = os.path.join(self.download_dir, f"{base_name}_temp")
        os.makedirs(temp_dir, exist_ok=true)
        
        # 临时输出路径模板
        temp_output = os.path.join(temp_dir, f"{base_name}_%d.png")
        
        try:
            # 使用imagemagick的convert命令转换ico到png
            cmd = [
                self.convert_exe,
                ico_path,
                temp_output
            ]
            subprocess.run(cmd, check=true)
            print(f"已转换: {ico_filename} -> 多个png文件")
            
            # 找出最大的png文件
            largest_file = self.find_largest_png(temp_dir)
            if largest_file:
                final_output = os.path.join(self.download_dir, f"{base_name}.png")
                os.rename(largest_file, final_output)
                print(f"已保存最大的png文件: {final_output}")
            
            # 清理临时目录
            self.cleanup_temp_dir(temp_dir)
            
        except subprocess.calledprocesserror as e:
            print(f"转换失败: {e}")
            self.cleanup_temp_dir(temp_dir)
        except exception as e:
            print(f"发生错误: {e}")
            self.cleanup_temp_dir(temp_dir)
    
    def find_largest_png(self, directory):
        largest_size = 0
        largest_file = none
        
        for filename in os.listdir(directory):
            if filename.lower().endswith('.png'):
                filepath = os.path.join(directory, filename)
                file_size = os.path.getsize(filepath)
                if file_size > largest_size:
                    largest_size = file_size
                    largest_file = filepath
        
        return largest_file
    
    def cleanup_temp_dir(self, temp_dir):
        try:
            for filename in os.listdir(temp_dir):
                filepath = os.path.join(temp_dir, filename)
                os.remove(filepath)
            os.rmdir(temp_dir)
        except exception as e:
            print(f"清理临时目录时出错: {e}")

def main():
    # 获取用户下载目录
    download_dir = os.path.expanduser('~/downloads')

    # 在这里设置imagemagick的convert.exe的完整路径
    # 如果'convert'在您的系统path中，您可以直接写'convert'
    # 否则，请提供完整路径，例如: r'c:\program files\imagemagick-7.1.1-q16-hdri\convert.exe'
    convert_exe = 'convert'
    
    # 查找并打印convert命令的完整路径
    # 如果用户没有提供绝对路径，则在path中搜索
    if convert_exe == 'convert' and not os.path.isabs(convert_exe):
        full_path = shutil.which(convert_exe)
        if full_path:
            print(f"查找到 'convert' 的完整路径是: {full_path}")
            convert_exe = full_path  # 使用完整路径以提高可靠性
        else:
            print(f"警告: 在系统path中找不到 '{convert_exe}'。程序可能会失败。")

    # 确保imagemagick已安装
    try:
        subprocess.run([convert_exe, '--version'], check=true, stdout=subprocess.pipe, stderr=subprocess.pipe)
    except (subprocess.calledprocesserror, filenotfounderror):
        print(f"错误: 无法执行 '{convert_exe}'。")
        print("请检查路径是否正确，或imagemagick是否已安装并添加到系统path中。")
        print("您可以修改脚本中 'convert_exe' 变量的值为 convert.exe 的完整路径。")
        print("imagemagick下载地址: https://imagemagick.org/script/download.php")
        return
    
    print(f"开始监控目录: {download_dir}")
    print("按ctrl+c停止监控")
    
    event_handler = icohandler(download_dir, convert_exe)
    observer = observer()
    observer.schedule(event_handler, download_dir, recursive=false)
    observer.start()
    
    try:
        while true:
            time.sleep(1)
    except keyboardinterrupt:
        observer.stop()
    observer.join()

if __name__ == "__main__":
    main()