traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

No reason to use a generator here

This commit is contained in:
Nick Coghlan 2012-10-19 23:59:35 +10:00
parent 097708aef6
commit 33b0fc2304
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Lib/test/test_import.py
View File

@ -15,16 +15,15 @@ from test.test_support import (unlink, TESTFN, unload, run_unittest, rmtree,
from test import symlink_support
from test import script_helper
def _iter_files(name):
for f in (name + os.extsep + "py",
name + os.extsep + "pyc",
name + os.extsep + "pyo",
name + os.extsep + "pyw",
name + "$py.class"):
yield f
def _files(name):
return (name + os.extsep + "py",
name + os.extsep + "pyc",
name + os.extsep + "pyo",
name + os.extsep + "pyw",
name + "$py.class")
def chmod_files(name):
for f in _iter_files(name):
for f in _files(name):
try:
os.chmod(f, 0600)
except OSError as exc:
@ -32,7 +31,7 @@ def chmod_files(name):
raise
def remove_files(name):
for f in _iter_files(name):
for f in _files(name):
unlink(f)