Issue #29326: Ignores blank lines in ._pth files (Patch by Alexey Izbyshev)

This commit is contained in:
Steve Dower 2017-02-04 15:19:46 -08:00
commit 824c6fdfaa
4 changed files with 31 additions and 14 deletions

View File

@ -511,16 +511,30 @@ class StartupImportTests(unittest.TestCase):
os.unlink(_pth_file) os.unlink(_pth_file)
os.unlink(exe_file) os.unlink(exe_file)
@classmethod
def _calc_sys_path_for_underpth_nosite(self, sys_prefix, lines):
sys_path = []
for line in lines:
if not line or line[0] == '#':
continue
abs_path = os.path.abspath(os.path.join(sys_prefix, line))
sys_path.append(abs_path)
return sys_path
@unittest.skipUnless(sys.platform == 'win32', "only supported on Windows") @unittest.skipUnless(sys.platform == 'win32', "only supported on Windows")
def test_underpth_nosite_file(self): def test_underpth_nosite_file(self):
libpath = os.path.dirname(os.path.dirname(encodings.__file__)) libpath = os.path.dirname(os.path.dirname(encodings.__file__))
exe_prefix = os.path.dirname(sys.executable) exe_prefix = os.path.dirname(sys.executable)
exe_file = self._create_underpth_exe([ pth_lines = [
'fake-path-name', 'fake-path-name',
*[libpath for _ in range(200)], *[libpath for _ in range(200)],
'',
'# comment', '# comment',
'import site' ]
]) exe_file = self._create_underpth_exe(pth_lines)
sys_path = self._calc_sys_path_for_underpth_nosite(
os.path.dirname(exe_file),
pth_lines)
try: try:
env = os.environ.copy() env = os.environ.copy()
@ -529,14 +543,11 @@ class StartupImportTests(unittest.TestCase):
rc = subprocess.call([exe_file, '-c', rc = subprocess.call([exe_file, '-c',
'import sys; sys.exit(sys.flags.no_site and ' 'import sys; sys.exit(sys.flags.no_site and '
'len(sys.path) > 200 and ' 'len(sys.path) > 200 and '
'%r in sys.path and %r in sys.path and %r not in sys.path)' % ( 'sys.path == %r)' % sys_path,
os.path.join(sys.prefix, 'fake-path-name'), ], env=env)
libpath,
os.path.join(sys.prefix, 'from-env'),
)], env=env)
finally: finally:
self._cleanup_underpth_exe(exe_file) self._cleanup_underpth_exe(exe_file)
self.assertEqual(rc, 0) self.assertTrue(rc, "sys.path is incorrect")
@unittest.skipUnless(sys.platform == 'win32', "only supported on Windows") @unittest.skipUnless(sys.platform == 'win32', "only supported on Windows")
def test_underpth_file(self): def test_underpth_file(self):
@ -545,23 +556,26 @@ class StartupImportTests(unittest.TestCase):
exe_file = self._create_underpth_exe([ exe_file = self._create_underpth_exe([
'fake-path-name', 'fake-path-name',
*[libpath for _ in range(200)], *[libpath for _ in range(200)],
'',
'# comment', '# comment',
'import site' 'import site'
]) ])
sys_prefix = os.path.dirname(exe_file)
try: try:
env = os.environ.copy() env = os.environ.copy()
env['PYTHONPATH'] = 'from-env' env['PYTHONPATH'] = 'from-env'
env['PATH'] = '{};{}'.format(exe_prefix, os.getenv('PATH')) env['PATH'] = '{};{}'.format(exe_prefix, os.getenv('PATH'))
rc = subprocess.call([exe_file, '-c', rc = subprocess.call([exe_file, '-c',
'import sys; sys.exit(not sys.flags.no_site and ' 'import sys; sys.exit(not sys.flags.no_site and '
'%r in sys.path and %r in sys.path and %r not in sys.path)' % ( '%r in sys.path and %r in sys.path and %r not in sys.path and '
os.path.join(sys.prefix, 'fake-path-name'), 'all("\\r" not in p and "\\n" not in p for p in sys.path))' % (
os.path.join(sys_prefix, 'fake-path-name'),
libpath, libpath,
os.path.join(sys.prefix, 'from-env'), os.path.join(sys_prefix, 'from-env'),
)], env=env) )], env=env)
finally: finally:
self._cleanup_underpth_exe(exe_file) self._cleanup_underpth_exe(exe_file)
self.assertEqual(rc, 0) self.assertTrue(rc, "sys.path is incorrect")
if __name__ == "__main__": if __name__ == "__main__":

View File

@ -684,6 +684,7 @@ John Interrante
Bob Ippolito Bob Ippolito
Roger Irwin Roger Irwin
Atsuo Ishimoto Atsuo Ishimoto
Alexey Izbyshev
Kasia Jachim Kasia Jachim
Adam Jackson Adam Jackson
Ben Jackson Ben Jackson

View File

@ -871,6 +871,8 @@ Library
Windows Windows
------- -------
- Issue #29326: Ignores blank lines in ._pth files (Patch by Alexey Izbyshev)
- Issue #28164: Correctly handle special console filenames (patch by Eryk Sun) - Issue #28164: Correctly handle special console filenames (patch by Eryk Sun)
- Issue #29409: Implement PEP 529 for io.FileIO (Patch by Eryk Sun) - Issue #29409: Implement PEP 529 for io.FileIO (Patch by Eryk Sun)

View File

@ -560,7 +560,7 @@ read_pth_file(const wchar_t *path, wchar_t *prefix, int *isolated, int *nosite)
char *p = fgets(line, MAXPATHLEN + 1, sp_file); char *p = fgets(line, MAXPATHLEN + 1, sp_file);
if (!p) if (!p)
break; break;
if (*p == '\0' || *p == '#') if (*p == '\0' || *p == '\r' || *p == '\n' || *p == '#')
continue; continue;
while (*++p) { while (*++p) {
if (*p == '\r' || *p == '\n') { if (*p == '\r' || *p == '\n') {