cpython/Lib/fnmatch.py

110 lines
3.1 KiB
Python
Raw Normal View History

1995-01-26 22:41:45 -04:00
"""Filename matching with shell patterns.
1992-01-12 19:29:29 -04:00
1995-01-26 22:41:45 -04:00
fnmatch(FILENAME, PATTERN) matches according to the local convention.
fnmatchcase(FILENAME, PATTERN) always takes case in account.
1992-01-12 19:29:29 -04:00
1995-01-26 22:41:45 -04:00
The functions operate by translating the pattern into a regular
expression. They cache the compiled regular expressions for speed.
The function translate(PATTERN) returns a regular expression
corresponding to PATTERN. (It does not compile it.)
"""
import os
import posixpath
import re
2010-08-13 13:26:40 -03:00
import functools
2010-08-13 13:26:40 -03:00
__all__ = ["filter", "fnmatch", "fnmatchcase", "translate"]
1991-01-01 14:11:14 -04:00
def fnmatch(name, pat):
2001-01-14 19:36:06 -04:00
"""Test whether FILENAME matches PATTERN.
Patterns are Unix shell style:
* matches everything
? matches any single character
[seq] matches any character in seq
[!seq] matches any char not in seq
An initial period in FILENAME is not special.
Both FILENAME and PATTERN are first case-normalized
if the operating system requires it.
If you don't want this, use fnmatchcase(FILENAME, PATTERN).
"""
name = os.path.normcase(name)
pat = os.path.normcase(pat)
return fnmatchcase(name, pat)
1995-01-26 22:41:45 -04:00
2013-02-17 05:33:37 -04:00
@functools.lru_cache(maxsize=256, typed=True)
2011-10-20 13:22:10 -03:00
def _compile_pattern(pat):
if isinstance(pat, bytes):
2010-08-13 13:26:40 -03:00
pat_str = str(pat, 'ISO-8859-1')
res_str = translate(pat_str)
res = bytes(res_str, 'ISO-8859-1')
else:
res = translate(pat)
return re.compile(res).match
def filter(names, pat):
"""Return the subset of the list NAMES that match PAT."""
result = []
pat = os.path.normcase(pat)
2011-10-20 13:22:10 -03:00
match = _compile_pattern(pat)
if os.path is posixpath:
# normcase on posix is NOP. Optimize it away from the loop.
for name in names:
if match(name):
result.append(name)
else:
for name in names:
if match(os.path.normcase(name)):
result.append(name)
return result
1995-01-26 22:41:45 -04:00
def fnmatchcase(name, pat):
2001-01-14 19:36:06 -04:00
"""Test whether FILENAME matches PATTERN, including case.
This is a version of fnmatch() which doesn't case-normalize
its arguments.
"""
2011-10-20 13:22:10 -03:00
match = _compile_pattern(pat)
return match(name) is not None
1991-01-01 14:11:14 -04:00
1992-01-12 19:29:29 -04:00
def translate(pat):
2001-01-14 19:36:06 -04:00
"""Translate a shell PATTERN to a regular expression.
There is no way to quote meta-characters.
"""
i, n = 0, len(pat)
res = ''
while i < n:
c = pat[i]
i = i+1
if c == '*':
res = res + '.*'
elif c == '?':
res = res + '.'
elif c == '[':
j = i
if j < n and pat[j] == '!':
j = j+1
if j < n and pat[j] == ']':
j = j+1
while j < n and pat[j] != ']':
j = j+1
if j >= n:
res = res + '\\['
else:
stuff = pat[i:j].replace('\\','\\\\')
2001-01-14 19:36:06 -04:00
i = j+1
if stuff[0] == '!':
stuff = '^' + stuff[1:]
elif stuff[0] == '^':
stuff = '\\' + stuff
res = '%s[%s]' % (res, stuff)
2001-01-14 19:36:06 -04:00
else:
res = res + re.escape(c)
return r'(?s:%s)\Z' % res