maccache.py 1008 Bytes
Newer Older
Guido van Rossum's avatar
Guido van Rossum committed
1 2 3
# Module 'maccache'
#
# Maintain a cache of listdir(), isdir(), isfile() or exists() outcomes.
4
# XXX Should merge with module statcache
Guido van Rossum's avatar
Guido van Rossum committed
5

6
import os
Guido van Rossum's avatar
Guido van Rossum committed
7 8 9 10 11 12 13 14 15 16 17


# The cache.
# Keys are absolute pathnames;
# values are 0 (nothing), 1 (file) or [...] (dir).
#
cache = {}


# Current working directory.
#
18
cwd = os.getcwd()
Guido van Rossum's avatar
Guido van Rossum committed
19 20 21 22 23 24 25 26 27


# Constants.
#
NONE = 0
FILE = 1
LISTTYPE = type([])

def _stat(name):
28
	name = os.path.join(cwd, name)
Guido van Rossum's avatar
Guido van Rossum committed
29 30
	if cache.has_key(name):
		return cache[name]
31
	if os.path.isfile(name):
Guido van Rossum's avatar
Guido van Rossum committed
32 33 34
		cache[name] = FILE
		return FILE
	try:
35
		list = os.listdir(name)
Guido van Rossum's avatar
Guido van Rossum committed
36 37 38 39
	except:
		cache[name] = NONE
		return NONE
	cache[name] = list
Guido van Rossum's avatar
Guido van Rossum committed
40
	if name[-1:] == ':': cache[name[:-1]] = list
Guido van Rossum's avatar
Guido van Rossum committed
41 42 43 44 45
	else: cache[name+':'] = list
	return list

def isdir(name):
	st = _stat(name)
Guido van Rossum's avatar
Guido van Rossum committed
46
	return type(st) == LISTTYPE
Guido van Rossum's avatar
Guido van Rossum committed
47 48 49

def isfile(name):
	st = _stat(name)
Guido van Rossum's avatar
Guido van Rossum committed
50
	return st == FILE
Guido van Rossum's avatar
Guido van Rossum committed
51 52 53 54 55 56 57

def exists(name):
	st = _stat(name)
	return st <> NONE

def listdir(name):
	st = _stat(name)
Guido van Rossum's avatar
Guido van Rossum committed
58
	if type(st) == LISTTYPE:
Guido van Rossum's avatar
Guido van Rossum committed
59 60 61
		return st
	else:
		raise RuntimeError, 'list non-directory'