~ubuntu-branches/ubuntu/karmic/pypy/karmic

« back to all changes in this revision

Viewing changes to lib-python/2.4.1/fnmatch.py

  • Committer: Bazaar Package Importer
  • Author(s): Alexandre Fayolle
  • Date: 2007-04-13 09:33:09 UTC
  • Revision ID: james.westby@ubuntu.com-20070413093309-yoojh4jcoocu2krz
Tags: upstream-1.0.0
ImportĀ upstreamĀ versionĀ 1.0.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
"""Filename matching with shell patterns.
 
2
 
 
3
fnmatch(FILENAME, PATTERN) matches according to the local convention.
 
4
fnmatchcase(FILENAME, PATTERN) always takes case in account.
 
5
 
 
6
The functions operate by translating the pattern into a regular
 
7
expression.  They cache the compiled regular expressions for speed.
 
8
 
 
9
The function translate(PATTERN) returns a regular expression
 
10
corresponding to PATTERN.  (It does not compile it.)
 
11
"""
 
12
 
 
13
import re
 
14
 
 
15
__all__ = ["filter", "fnmatch","fnmatchcase","translate"]
 
16
 
 
17
_cache = {}
 
18
 
 
19
def fnmatch(name, pat):
 
20
    """Test whether FILENAME matches PATTERN.
 
21
 
 
22
    Patterns are Unix shell style:
 
23
 
 
24
    *       matches everything
 
25
    ?       matches any single character
 
26
    [seq]   matches any character in seq
 
27
    [!seq]  matches any char not in seq
 
28
 
 
29
    An initial period in FILENAME is not special.
 
30
    Both FILENAME and PATTERN are first case-normalized
 
31
    if the operating system requires it.
 
32
    If you don't want this, use fnmatchcase(FILENAME, PATTERN).
 
33
    """
 
34
 
 
35
    import os
 
36
    name = os.path.normcase(name)
 
37
    pat = os.path.normcase(pat)
 
38
    return fnmatchcase(name, pat)
 
39
 
 
40
def filter(names, pat):
 
41
    """Return the subset of the list NAMES that match PAT"""
 
42
    import os,posixpath
 
43
    result=[]
 
44
    pat=os.path.normcase(pat)
 
45
    if not pat in _cache:
 
46
        res = translate(pat)
 
47
        _cache[pat] = re.compile(res)
 
48
    match=_cache[pat].match
 
49
    if os.path is posixpath:
 
50
        # normcase on posix is NOP. Optimize it away from the loop.
 
51
        for name in names:
 
52
            if match(name):
 
53
                result.append(name)
 
54
    else:
 
55
        for name in names:
 
56
            if match(os.path.normcase(name)):
 
57
                result.append(name)
 
58
    return result
 
59
 
 
60
def fnmatchcase(name, pat):
 
61
    """Test whether FILENAME matches PATTERN, including case.
 
62
 
 
63
    This is a version of fnmatch() which doesn't case-normalize
 
64
    its arguments.
 
65
    """
 
66
 
 
67
    if not pat in _cache:
 
68
        res = translate(pat)
 
69
        _cache[pat] = re.compile(res)
 
70
    return _cache[pat].match(name) is not None
 
71
 
 
72
def translate(pat):
 
73
    """Translate a shell PATTERN to a regular expression.
 
74
 
 
75
    There is no way to quote meta-characters.
 
76
    """
 
77
 
 
78
    i, n = 0, len(pat)
 
79
    res = ''
 
80
    while i < n:
 
81
        c = pat[i]
 
82
        i = i+1
 
83
        if c == '*':
 
84
            res = res + '.*'
 
85
        elif c == '?':
 
86
            res = res + '.'
 
87
        elif c == '[':
 
88
            j = i
 
89
            if j < n and pat[j] == '!':
 
90
                j = j+1
 
91
            if j < n and pat[j] == ']':
 
92
                j = j+1
 
93
            while j < n and pat[j] != ']':
 
94
                j = j+1
 
95
            if j >= n:
 
96
                res = res + '\\['
 
97
            else:
 
98
                stuff = pat[i:j].replace('\\','\\\\')
 
99
                i = j+1
 
100
                if stuff[0] == '!':
 
101
                    stuff = '^' + stuff[1:]
 
102
                elif stuff[0] == '^':
 
103
                    stuff = '\\' + stuff
 
104
                res = '%s[%s]' % (res, stuff)
 
105
        else:
 
106
            res = res + re.escape(c)
 
107
    return res + "$"