Fix reporting of UndefinedLocal pyflakes error

This commit is contained in:
Anthony Sottile 2019-01-31 21:53:19 -08:00
parent 79148a02ee
commit 8df38c92b9
2 changed files with 23 additions and 5 deletions

View file

@ -57,10 +57,7 @@ def patch_pyflakes():
"""Add error codes to Pyflakes messages."""
for name, obj in vars(pyflakes.messages).items():
if name[0].isupper() and obj.message:
obj.flake8_msg = "%s %s" % (
FLAKE8_PYFLAKES_CODES.get(name, "F999"),
obj.message,
)
obj.flake8_code = FLAKE8_PYFLAKES_CODES.get(name, "F999")
patch_pyflakes()
@ -188,6 +185,9 @@ class FlakesChecker(pyflakes.checker.Checker):
yield (
message.lineno,
col,
(message.flake8_msg % message.message_args),
"{} {}".format(
message.flake8_code,
message.message % message.message_args,
),
message.__class__,
)