Skip to content

Commit

Permalink
Re-apply r83871.
Browse files Browse the repository at this point in the history
  • Loading branch information
pitrou committed Aug 13, 2010
1 parent e81c806 commit 6fdb74f
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 60 deletions.
7 changes: 0 additions & 7 deletions Doc/library/fnmatch.rst
Expand Up @@ -84,13 +84,6 @@ patterns.
<_sre.SRE_Match object at 0x...>


.. function:: purge()

Clear the internal pattern cache.

.. versionadded:: 3.2


.. seealso::

Module :mod:`glob`
Expand Down
44 changes: 13 additions & 31 deletions Lib/fnmatch.py
Expand Up @@ -12,19 +12,9 @@
import os
import posixpath
import re
import functools

__all__ = ["filter", "fnmatch", "fnmatchcase", "purge", "translate"]

_cache = {} # Maps text patterns to compiled regexen.
_cacheb = {} # Ditto for bytes patterns.
_MAXCACHE = 100 # Maximum size of caches.


def purge():
"""Clear the pattern cache."""
_cache.clear()
_cacheb.clear()

__all__ = ["filter", "fnmatch", "fnmatchcase", "translate"]

def fnmatch(name, pat):
"""Test whether FILENAME matches PATTERN.
Expand All @@ -45,28 +35,21 @@ def fnmatch(name, pat):
pat = os.path.normcase(pat)
return fnmatchcase(name, pat)


def _compile_pattern(pat):
cache = _cacheb if isinstance(pat, bytes) else _cache
regex = cache.get(pat)
if regex is None:
if isinstance(pat, bytes):
pat_str = str(pat, 'ISO-8859-1')
res_str = translate(pat_str)
res = bytes(res_str, 'ISO-8859-1')
else:
res = translate(pat)
if len(cache) >= _MAXCACHE:
cache.clear()
cache[pat] = regex = re.compile(res)
return regex.match

@functools.lru_cache(maxsize=250)
def _compile_pattern(pat, is_bytes=False):
if is_bytes:
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)
match = _compile_pattern(pat)
match = _compile_pattern(pat, isinstance(pat, bytes))
if os.path is posixpath:
# normcase on posix is NOP. Optimize it away from the loop.
for name in names:
Expand All @@ -78,14 +61,13 @@ def filter(names, pat):
result.append(name)
return result


def fnmatchcase(name, pat):
"""Test whether FILENAME matches PATTERN, including case.
This is a version of fnmatch() which doesn't case-normalize
its arguments.
"""
match = _compile_pattern(pat)
match = _compile_pattern(pat, isinstance(pat, bytes))
return match(name) is not None


Expand Down
23 changes: 1 addition & 22 deletions Lib/test/test_fnmatch.py
Expand Up @@ -3,15 +3,10 @@
from test import support
import unittest

from fnmatch import (fnmatch, fnmatchcase, _MAXCACHE, _cache, _cacheb, purge,
translate, filter)

from fnmatch import fnmatch, fnmatchcase, translate, filter

class FnmatchTestCase(unittest.TestCase):

def tearDown(self):
purge()

def check_match(self, filename, pattern, should_match=1, fn=fnmatch):
if should_match:
self.assertTrue(fn(filename, pattern),
Expand Down Expand Up @@ -65,22 +60,6 @@ def test_bytes(self):
self.check_match(b'test\xff', b'te*\xff')
self.check_match(b'foo\nbar', b'foo*')

def test_cache_clearing(self):
# check that caches do not grow too large
# http://bugs.python.org/issue7846

# string pattern cache
for i in range(_MAXCACHE + 1):
fnmatch('foo', '?' * i)

self.assertLessEqual(len(_cache), _MAXCACHE)

# bytes pattern cache
for i in range(_MAXCACHE + 1):
fnmatch(b'foo', b'?' * i)
self.assertLessEqual(len(_cacheb), _MAXCACHE)


class TranslateTestCase(unittest.TestCase):

def test_translate(self):
Expand Down

0 comments on commit 6fdb74f

Please sign in to comment.