Merge pull request #7 from mcepl/PR7-theospears-add-location

mcepl · mcepl · commit ed5b72e9444c · 2013-12-04T08:55:29.000-08:00
Add Location.line_content property to provide context for locations
diff --git a/ropemode/interface.py b/ropemode/interface.py
@@ -552,6 +552,18 @@ def lineno(self):
             return self.location.lineno
         return self.location.resource.read().count('\n', 0, self.offset) + 1
 
+    @property
+    def line_content(self):
+        resource_contents = self.location.resource.read()
+        # rfind returns -1 for start of string, so by adding 1 we get the start
+        # of the string. When a match is found we want to exclude the matching
+        # character so again we add 1.
+        line_start = resource_contents.rfind("\n", 0, self.offset) + 1
+        line_end = resource_contents.find("\n", self.offset)
+        if line_end < 0:
+            line_end = len(resource_contents)
+        return resource_contents[line_start:line_end]
+
 
 class _CodeAssist(object):
 
