closes bpo-37675: Use pkgutil.iter_modules to find fixers in a package rather than listdir. (14942)

Author: benjaminp

Lib/lib2to3/refactor.py

@@ -14,6 +14,7 @@
 # Python imports
 import io
 import os
+import pkgutil
 import sys
 import logging
 import operator
@@ -30,13 +31,12 @@
 def get_all_fix_names(fixer_pkg, remove_prefix=True):
  """Return a sorted list of all available fix names in the given package."""
  pkg = __import__(fixer_pkg, [], [], ["*"])
- fixer_dir = os.path.dirname(pkg.__file__)
  fix_names = []
- for name in sorted(os.listdir(fixer_dir)):
- if name.startswith("fix_") and name.endswith(".py"):
+ for finder, name, ispkg in pkgutil.iter_modules(pkg.__path__):
+ if name.startswith("fix_"):
  if remove_prefix:
  name = name[4:]
- fix_names.append(name[:-3])
+ fix_names.append(name)
  return fix_names
 
 

Misc/NEWS.d/next/Tools-Demos/2019-07-24-16-20-54.bpo-37675.951Cvf.rst

@@ -0,0 +1 @@
+2to3 now works when run from a zipped standard library.