cpython/Lib/statcache.py

76 lines
1.5 KiB
Python
Raw Normal View History

"""Maintain a cache of stat() information on files.
There are functions to reset the cache or to selectively remove items.
"""
1990-10-13 16:23:40 -03:00
1992-03-31 15:04:48 -04:00
import os
from stat import *
1990-10-13 16:23:40 -03:00
# The cache.
1992-03-31 15:04:48 -04:00
# Keys are pathnames, values are `os.stat' outcomes.
1990-10-13 16:23:40 -03:00
#
cache = {}
def stat(path):
"""Stat a file, possibly out of the cache."""
if cache.has_key(path):
1990-10-13 16:23:40 -03:00
return cache[path]
1992-03-31 15:04:48 -04:00
cache[path] = ret = os.stat(path)
1990-10-13 16:23:40 -03:00
return ret
def reset():
"""Reset the cache completely."""
global cache
cache = {}
1990-10-13 16:23:40 -03:00
def forget(path):
"""Remove a given item from the cache, if it exists."""
if cache.has_key(path):
1990-10-13 16:23:40 -03:00
del cache[path]
def forget_prefix(prefix):
"""Remove all pathnames with a given prefix."""
1990-10-13 16:23:40 -03:00
n = len(prefix)
for path in cache.keys():
1992-01-01 15:35:13 -04:00
if path[:n] == prefix:
1990-10-13 16:23:40 -03:00
del cache[path]
def forget_dir(prefix):
"""Forget about a directory and all entries in it, but not about
entries in subdirectories."""
if prefix[-1:] == '/' and prefix != '/':
1990-10-13 16:23:40 -03:00
prefix = prefix[:-1]
forget(prefix)
if prefix[-1:] != '/':
1990-10-13 16:23:40 -03:00
prefix = prefix + '/'
n = len(prefix)
for path in cache.keys():
1992-01-01 15:35:13 -04:00
if path[:n] == prefix:
1990-10-13 16:23:40 -03:00
rest = path[n:]
1992-01-01 15:35:13 -04:00
if rest[-1:] == '/': rest = rest[:-1]
1990-10-13 16:23:40 -03:00
if '/' not in rest:
del cache[path]
def forget_except_prefix(prefix):
"""Remove all pathnames except with a given prefix.
Normally used with prefix = '/' after a chdir()."""
1990-10-13 16:23:40 -03:00
n = len(prefix)
for path in cache.keys():
if path[:n] != prefix:
1990-10-13 16:23:40 -03:00
del cache[path]
def isdir(path):
"""Check for directory."""
1990-10-13 16:23:40 -03:00
try:
st = stat(path)
1992-03-31 15:04:48 -04:00
except os.error:
1990-10-13 16:23:40 -03:00
return 0
return S_ISDIR(st[ST_MODE])