Spaces:
Running
Running
| """ | |
| Customized Mixin2to3 support: | |
| - adds support for converting doctests | |
| This module raises an ImportError on Python 2. | |
| """ | |
| from distutils.util import Mixin2to3 as _Mixin2to3 | |
| from distutils import log | |
| from lib2to3.refactor import RefactoringTool, get_fixers_from_package | |
| import setuptools | |
| class DistutilsRefactoringTool(RefactoringTool): | |
| def log_error(self, msg, *args, **kw): | |
| log.error(msg, *args) | |
| def log_message(self, msg, *args): | |
| log.info(msg, *args) | |
| def log_debug(self, msg, *args): | |
| log.debug(msg, *args) | |
| class Mixin2to3(_Mixin2to3): | |
| def run_2to3(self, files, doctests=False): | |
| # See of the distribution option has been set, otherwise check the | |
| # setuptools default. | |
| if self.distribution.use_2to3 is not True: | |
| return | |
| if not files: | |
| return | |
| log.info("Fixing " + " ".join(files)) | |
| self.__build_fixer_names() | |
| self.__exclude_fixers() | |
| if doctests: | |
| if setuptools.run_2to3_on_doctests: | |
| r = DistutilsRefactoringTool(self.fixer_names) | |
| r.refactor(files, write=True, doctests_only=True) | |
| else: | |
| _Mixin2to3.run_2to3(self, files) | |
| def __build_fixer_names(self): | |
| if self.fixer_names: | |
| return | |
| self.fixer_names = [] | |
| for p in setuptools.lib2to3_fixer_packages: | |
| self.fixer_names.extend(get_fixers_from_package(p)) | |
| if self.distribution.use_2to3_fixers is not None: | |
| for p in self.distribution.use_2to3_fixers: | |
| self.fixer_names.extend(get_fixers_from_package(p)) | |
| def __exclude_fixers(self): | |
| excluded_fixers = getattr(self, 'exclude_fixers', []) | |
| if self.distribution.use_2to3_exclude_fixers is not None: | |
| excluded_fixers.extend(self.distribution.use_2to3_exclude_fixers) | |
| for fixer_name in excluded_fixers: | |
| if fixer_name in self.fixer_names: | |
| self.fixer_names.remove(fixer_name) | |