来源:http://code.activestate.com/recipes/215418-watching-a-directory-tree-on-unix/
watch_directories() 函数获取一个路径列表和一个可调用对象,然后反复遍历以这些路径为根的目录树,监视被删除或修改时间更改的文件。然后将可调用对象传递给两个列表,其中包含已更改的文件和已删除的文件。
from __future__ import nested_scopes
import os, time
def watch_directories (paths, func, delay=1.0):
"""(paths:[str], func:callable, delay:float)
Continuously monitors the paths and their subdirectories
for changes. If any files or directories are modified,
the callable 'func' is called with a list of the modified paths of both
files and directories. 'func' can return a Boolean value
for rescanning; if it returns True, the directory tree will be
rescanned without calling func() for any found changes.
(This is so func() can write changes into the tree and prevent itself
from being immediately called again.)
"""
# Basic principle: all_files is a dictionary mapping paths to
# modification times. We repeatedly crawl through the directory
# tree rooted at 'path', doing a stat() on each file and comparing
# the modification time.
all_files = {}
def f (unused, dirname, files):
# Traversal function for directories
for filename in files:
path = os.path.join(dirname, filename)
try:
t = os.stat(path)
except os.error:
# If a file has been deleted between os.path.walk()
# scanning the directory and now, we'll get an
# os.error here. Just ignore it -- we'll report
# the deletion on the next pass through the main loop.
continue
mtime = remaining_files.get(path)
if mtime is not None:
# Record this file as having been seen
del remaining_files[path]
# File's mtime has been changed since we last looked at it.
if t.st_mtime > mtime:
changed_list.append(path)
else:
# No recorded modification time, so it must be
# a brand new file.
changed_list.append(path)
# Record current mtime of file.
all_files[path] = t.st_mtime
# Main loop
rescan = False
while True:
changed_list = []
remaining_files = all_files.copy()
all_files = {}
for path in paths:
os.path.walk(path, f, None)
removed_list = remaining_files.keys()
if rescan:
rescan = False
elif changed_list or removed_list:
rescan = func(changed_list, removed_list)
time.sleep(delay)
if __name__ == '__main__':
def f (changed_files, removed_files):
print changed_files
print 'Removed', removed_files
watch_directories(['.'], f, 1)
如果您希望通过某种方式将作业发送到守护程序,但又不想使用某些 IPC 机制(例如套接字或管道),则此秘诀很有用。相反,守护进程可以坐下来监视提交目录,并且可以通过将文件或目录拖放到提交目录中来提交作业。
不考虑锁定。 watch_directories() 函数本身并不需要锁定;如果它错过了一次修改,它会在下一次通过时注意到它。但是,如果将作业直接写入监视目录,则可调用对象可能会在作业文件仅写入一半时开始运行。要解决这个问题,您可以使用锁定文件;可调用对象在运行时必须获取锁,提交者在希望添加新作业时必须获取锁。一种更简单的方法是依赖 rename() 系统调用是原子的:将作业写入不被监视的临时目录中,一旦文件完成,使用 os.rename() 将其移动到提交目录中。