mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-06-29 13:40:59 +02:00
63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
"""Monkey-patching for pep8 and pycodestyle."""
|
|
try:
|
|
import pep8
|
|
except ImportError:
|
|
pep8 = None
|
|
|
|
try:
|
|
import pycodestyle
|
|
except ImportError:
|
|
pycodestyle = None
|
|
|
|
from flake8_polyfill import version
|
|
|
|
__all__ = ('monkey_patch',)
|
|
|
|
modules = {
|
|
'pep8': [pep8],
|
|
'pycodestyle': [pycodestyle],
|
|
'all': [pep8, pycodestyle],
|
|
}
|
|
|
|
|
|
def monkey_patch(which):
|
|
"""Monkey-patch the specified module with the appropriate stdin.
|
|
|
|
On Flake8 2.5 and lower, Flake8 would would monkey-patch
|
|
``pep8.stdin_get_value`` for everyone. This avoided problems where
|
|
stdin might be exhausted.
|
|
|
|
On Flake8 2.6, Flake8 stopped patching ``pep8`` and started
|
|
monkey-patching ``pycodestyle.stdin_get_value``.
|
|
|
|
On Flake8 3.x, Flake8 has no need to monkey patch either ``pep8`` or
|
|
``pycodestyle``.
|
|
|
|
This function accepts three parameters:
|
|
|
|
- pep8
|
|
- pycodestyle
|
|
- all
|
|
|
|
"all" is a special value that will monkey-patch both "pep8" and
|
|
"pycodestyle".
|
|
|
|
:param str which:
|
|
The name of the module to patch.
|
|
:returns:
|
|
Nothing.
|
|
:rtype:
|
|
NoneType
|
|
"""
|
|
if (2, 0) <= version.version_info < (3, 0):
|
|
from flake8.engine import pep8 as _pep8
|
|
stdin_get_value = _pep8.stdin_get_value
|
|
elif (3, 0) <= version.version_info < (4, 0):
|
|
from flake8 import utils
|
|
stdin_get_value = utils.stdin_get_value
|
|
|
|
for module in modules[which]:
|
|
if module is None:
|
|
continue
|
|
module.stdin_get_value = stdin_get_value
|