ansible-later/env_27/lib/python2.7/site-packages/flake8_docstrings.py
Robert Kaussow 10aaa8e7e3 fix pytest
2019-04-11 15:56:20 +02:00

103 lines
2.9 KiB
Python

# -*- coding: utf-8 -*-
"""Implementation of pydocstyle integration with Flake8.
pydocstyle docstrings convention needs error code and class parser for be
included as module into flake8
"""
from flake8_polyfill import stdin
import pycodestyle
try:
import pydocstyle as pep257
module_name = 'pydocstyle'
except ImportError:
import pep257
module_name = 'pep257'
__version__ = '1.3.0'
__all__ = ('pep257Checker',)
stdin.monkey_patch('pycodestyle')
class EnvironError(pep257.Error):
def __init__(self, err):
super(EnvironError, self).__init__(
code='D998',
short_desc='EnvironmentError: ' + str(err),
context=None,
)
@property
def line(self):
"""Return 0 as line number for EnvironmentError."""
return 0
class AllError(pep257.Error):
def __init__(self, err):
super(AllError, self).__init__(
code='D999',
short_desc=str(err).partition('\n')[0],
context=None,
)
@property
def line(self):
"""pep257.AllError does not contain line number. Return 0 instead."""
return 0
class pep257Checker(object):
"""Flake8 needs a class to check python file."""
name = 'flake8-docstrings'
version = __version__ + ', {0}: {1}'.format(
module_name, pep257.__version__
)
STDIN_NAMES = set(['stdin', '-', '(none)', None])
def __init__(self, tree, filename='(none)'):
"""Placeholder."""
self.tree = tree
self.filename = filename
self.checker = pep257.ConventionChecker()
self.load_source()
def _check_source(self):
try:
# TODO: Naive fix for `pydocstyle 2.0.0` with default settings.
# Should probably add a proper setting so `ignore_decorators` can
# be set when calling through the CLI
for err in self.checker.check_source(
self.source,
self.filename,
ignore_decorators=None,
):
yield err
except pep257.AllError as err:
yield AllError(err)
except EnvironmentError as err:
yield EnvironError(err)
def run(self):
"""Use directly check() api from pydocstyle."""
checked_codes = pep257.conventions.pep257 | {'D998', 'D999'}
for error in self._check_source():
if isinstance(error, pep257.Error) and error.code in checked_codes:
# NOTE(sigmavirus24): Fixes GitLab#3
message = '%s %s' % (error.code, error.short_desc)
yield (error.line, 0, message, type(self))
def load_source(self):
"""Load the source for the specified file."""
if self.filename in self.STDIN_NAMES:
self.filename = 'stdin'
self.source = pycodestyle.stdin_get_value()
else:
with pep257.tokenize_open(self.filename) as fd:
self.source = fd.read()