PageRenderTime 10ms CodeModel.GetById 1ms app.highlight 7ms RepoModel.GetById 1ms app.codeStats 0ms

/Lib/glob.py

http://unladen-swallow.googlecode.com/
Python | 78 lines | 57 code | 7 blank | 14 comment | 20 complexity | 6d3ac906ba8cd3f97a569f3f6279cfd8 MD5 | raw file
 1"""Filename globbing utility."""
 2
 3import sys
 4import os
 5import re
 6import fnmatch
 7
 8__all__ = ["glob", "iglob"]
 9
10def glob(pathname):
11    """Return a list of paths matching a pathname pattern.
12
13    The pattern may contain simple shell-style wildcards a la fnmatch.
14
15    """
16    return list(iglob(pathname))
17
18def iglob(pathname):
19    """Return an iterator which yields the paths matching a pathname pattern.
20
21    The pattern may contain simple shell-style wildcards a la fnmatch.
22
23    """
24    if not has_magic(pathname):
25        if os.path.lexists(pathname):
26            yield pathname
27        return
28    dirname, basename = os.path.split(pathname)
29    if not dirname:
30        for name in glob1(os.curdir, basename):
31            yield name
32        return
33    if has_magic(dirname):
34        dirs = iglob(dirname)
35    else:
36        dirs = [dirname]
37    if has_magic(basename):
38        glob_in_dir = glob1
39    else:
40        glob_in_dir = glob0
41    for dirname in dirs:
42        for name in glob_in_dir(dirname, basename):
43            yield os.path.join(dirname, name)
44
45# These 2 helper functions non-recursively glob inside a literal directory.
46# They return a list of basenames. `glob1` accepts a pattern while `glob0`
47# takes a literal basename (so it only has to check for its existence).
48
49def glob1(dirname, pattern):
50    if not dirname:
51        dirname = os.curdir
52    if isinstance(pattern, unicode) and not isinstance(dirname, unicode):
53        dirname = unicode(dirname, sys.getfilesystemencoding() or
54                                   sys.getdefaultencoding())
55    try:
56        names = os.listdir(dirname)
57    except os.error:
58        return []
59    if pattern[0] != '.':
60        names = filter(lambda x: x[0] != '.', names)
61    return fnmatch.filter(names, pattern)
62
63def glob0(dirname, basename):
64    if basename == '':
65        # `os.path.split()` returns an empty basename for paths ending with a
66        # directory separator.  'q*x/' should match only directories.
67        if os.path.isdir(dirname):
68            return [basename]
69    else:
70        if os.path.lexists(os.path.join(dirname, basename)):
71            return [basename]
72    return []
73
74
75magic_check = re.compile('[*?[]')
76
77def has_magic(s):
78    return magic_check.search(s) is not None