Commit 59f7f4b4 authored by Jason R. Coombs's avatar Jason R. Coombs

Issue #12885: Revert commits in 3.4 branch which is security-only fixes.

parent 747d37af
...@@ -6,7 +6,6 @@ and building lists of files. ...@@ -6,7 +6,6 @@ and building lists of files.
import os, re import os, re
import fnmatch import fnmatch
import functools
from distutils.util import convert_path from distutils.util import convert_path
from distutils.errors import DistutilsTemplateError, DistutilsInternalError from distutils.errors import DistutilsTemplateError, DistutilsInternalError
from distutils import log from distutils import log
...@@ -243,28 +242,35 @@ class FileList: ...@@ -243,28 +242,35 @@ class FileList:
# ---------------------------------------------------------------------- # ----------------------------------------------------------------------
# Utility functions # Utility functions
def _find_all_simple(path): def findall(dir=os.curdir):
""" """Find all files under 'dir' and return the list of full filenames
Find all files under 'path' (relative to 'dir').
""" """
results = ( from stat import ST_MODE, S_ISREG, S_ISDIR, S_ISLNK
os.path.join(base, file)
for base, dirs, files in os.walk(path, followlinks=True)
for file in files
)
return filter(os.path.isfile, results)
list = []
stack = [dir]
pop = stack.pop
push = stack.append
def findall(dir=os.curdir): while stack:
""" dir = pop()
Find all files under 'dir' and return the list of full filenames. names = os.listdir(dir)
Unless dir is '.', return full filenames with dir prepended.
""" for name in names:
files = _find_all_simple(dir) if dir != os.curdir: # avoid the dreaded "./" syndrome
if dir == os.curdir: fullname = os.path.join(dir, name)
make_rel = functools.partial(os.path.relpath, start=dir) else:
files = map(make_rel, files) fullname = name
return list(files)
# Avoid excess stat calls -- just one will do, thank you!
stat = os.stat(fullname)
mode = stat[ST_MODE]
if S_ISREG(mode):
list.append(fullname)
elif S_ISDIR(mode) and not S_ISLNK(mode):
push(fullname)
return list
def glob_to_re(pattern): def glob_to_re(pattern):
......
...@@ -6,10 +6,8 @@ from distutils import debug ...@@ -6,10 +6,8 @@ from distutils import debug
from distutils.log import WARN from distutils.log import WARN
from distutils.errors import DistutilsTemplateError from distutils.errors import DistutilsTemplateError
from distutils.filelist import glob_to_re, translate_pattern, FileList from distutils.filelist import glob_to_re, translate_pattern, FileList
from distutils import filelist
import test.support from test.support import captured_stdout
from test.support import captured_stdout, run_unittest
from distutils.tests import support from distutils.tests import support
MANIFEST_IN = """\ MANIFEST_IN = """\
...@@ -294,47 +292,5 @@ class FileListTestCase(support.LoggingSilencer, ...@@ -294,47 +292,5 @@ class FileListTestCase(support.LoggingSilencer,
self.assertWarnings() self.assertWarnings()
class FindAllTestCase(unittest.TestCase):
@test.support.skip_unless_symlink
def test_missing_symlink(self):
with test.support.temp_cwd():
os.symlink('foo', 'bar')
self.assertEqual(filelist.findall(), [])
def test_basic_discovery(self):
"""
When findall is called with no parameters or with
'.' as the parameter, the dot should be omitted from
the results.
"""
with test.support.temp_cwd():
os.mkdir('foo')
file1 = os.path.join('foo', 'file1.txt')
test.support.create_empty_file(file1)
os.mkdir('bar')
file2 = os.path.join('bar', 'file2.txt')
test.support.create_empty_file(file2)
expected = [file2, file1]
self.assertEqual(sorted(filelist.findall()), expected)
def test_non_local_discovery(self):
"""
When findall is called with another path, the full
path name should be returned.
"""
with test.support.temp_dir() as temp_dir:
file1 = os.path.join(temp_dir, 'file1.txt')
test.support.create_empty_file(file1)
expected = [file1]
self.assertEqual(filelist.findall(temp_dir), expected)
def test_suite():
return unittest.TestSuite([
unittest.makeSuite(FileListTestCase),
unittest.makeSuite(FindAllTestCase),
])
if __name__ == "__main__": if __name__ == "__main__":
run_unittest(test_suite()) unittest.main()
...@@ -13,8 +13,6 @@ Core and Builtins ...@@ -13,8 +13,6 @@ Core and Builtins
Library Library
------- -------
- Issue #12885: Fix error when distutils encounters symlink.
- In the curses module, raise an error if window.getstr() or window.instr() is - In the curses module, raise an error if window.getstr() or window.instr() is
passed a negative value. passed a negative value.
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment