cpython/Lib/fnmatch.py

57 lines
1.2 KiB
Python
Raw Normal View History

1991-01-01 14:11:14 -04:00
# module 'fnmatch' -- filename matching with shell patterns
1992-01-12 19:29:29 -04:00
# This version translates the pattern to a regular expression
# and moreover caches the expressions.
import os
import regex
cache = {}
1991-01-01 14:11:14 -04:00
def fnmatch(name, pat):
1992-01-12 19:29:29 -04:00
name = os.path.normcase(name)
pat = os.path.normcase(pat)
if not cache.has_key(pat):
res = translate(pat)
save_syntax = regex.set_syntax(0)
cache[pat] = regex.compile(res)
save_syntax = regex.set_syntax(save_syntax)
return cache[pat].match(name) == len(name)
1991-01-01 14:11:14 -04:00
1992-01-12 19:29:29 -04:00
def translate(pat):
1992-01-01 15:35:13 -04:00
i, n = 0, len(pat)
1992-01-12 19:29:29 -04:00
res = ''
1992-01-01 15:35:13 -04:00
while i < n:
1991-01-01 14:11:14 -04:00
c = pat[i]
1992-01-12 19:29:29 -04:00
i = i+1
1992-01-01 15:35:13 -04:00
if c == '*':
1992-01-12 19:29:29 -04:00
res = res + '.*'
1992-01-01 15:35:13 -04:00
elif c == '?':
1992-01-12 19:29:29 -04:00
res = res + '.'
1992-01-01 15:35:13 -04:00
elif c == '[':
1992-01-12 19:29:29 -04:00
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]
i = j+1
if stuff[0] == '!':
stuff = '[^' + stuff[1:] + ']'
elif stuff == '^'*len(stuff):
stuff = '\\^'
else:
while stuff[0] == '^':
stuff = stuff[1:] + stuff[0]
stuff = '[' + stuff + ']'
res = res + stuff
elif c in '\\.+^$':
res = res + ('\\' + c)
1991-01-01 14:11:14 -04:00
else:
1992-01-12 19:29:29 -04:00
res = res + c
1991-01-01 14:11:14 -04:00
return res