Add end coordinates for column and line number

Author: krassowski

pyflakes/checker.py

@@ -1389,9 +1389,13 @@ def getDocstring(self, node):
     def handleNode(self, node, parent):
         if node is None:
             return
-        if self.offset and getattr(node, 'lineno', None) is not None:
-            node.lineno += self.offset[0]
-            node.col_offset += self.offset[1]
+        if self.offset:
+            if getattr(node, 'lineno', None) is not None:
+                node.lineno += self.offset[0]
+                node.col_offset += self.offset[1]
+            if getattr(node, 'end_lineno', None) is not None:
+                node.end_lineno += self.offset[0]
+                node.end_col_offset += self.offset[1]
         if self.futuresAllowed and not (isinstance(node, ast.ImportFrom) or
                                         self.isDocstring(node)):
             self.futuresAllowed = False

pyflakes/messages.py

@@ -11,6 +11,8 @@ def __init__(self, filename, loc):
         self.filename = filename
         self.lineno = loc.lineno
         self.col = getattr(loc, 'col_offset', 0)
+        self.end_col = getattr(loc, 'end_col_offset', None)
+        self.end_lineno = getattr(loc, 'end_lineno', None)
 
     def __str__(self):
         return '%s:%s:%s %s' % (self.filename, self.lineno, self.col+1,

pyflakes/test/test_doctests.py

@@ -298,6 +298,11 @@ def doctest_stuff():
         ''', m.UndefinedName).messages[0]
         self.assertEqual(exc.lineno, 5)
         self.assertEqual(exc.col, 20)
+        # end_col_offset and end_lineno are new in Python 3.8
+        if sys.version_info >= (3, 8):
+            # check that offsets are also added to end_col and end_lineno
+            self.assertEqual(exc.end_lineno, 5)
+            self.assertEqual(exc.end_col, 21)
 
     def test_offsetAfterDoctests(self):
         exc = self.flakes('''