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:
raise
def remove_files(name):
- for f in _iter_files(name):
+ for f in _files(name):
unlink(f)