Add source.Range.source_lines().

whitequark · whitequark · commit 72f6c5ff5008 · 2015-07-18T07:51:13.000+03:00
diff --git a/pythonparser/source.py b/pythonparser/source.py
@@ -148,6 +148,13 @@ def source_line(self):
         """
         return self.source_buffer.source_line(self.line())
 
+    def source_lines(self):
+        """
+        Returns the lines of source code containing the entirety of this range.
+        """
+        return [self.source_buffer.source_line(line)
+                for line in range(self.line(), self.end().line() + 1)]
+
     def __str__(self):
         """
         Returns a Clang-style string representation of the beginning of this range.
diff --git a/pythonparser/test/test_source.py b/pythonparser/test/test_source.py
@@ -85,6 +85,10 @@ def test_source(self):
     def test_source_line(self):
         self.assertEqual("line two\n", self.range(9, 9).source_line())
 
+    def test_source_lines(self):
+        self.assertEqual(["line two\n"], self.range(9, 9).source_lines())
+        self.assertEqual(["line two\n", "\n"], self.range(9, 18).source_lines())
+
     def test___str__(self):
         self.assertEqual("<input>:2:1-2:1", str(self.range(9, 9)))
 
