Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 1 | 'Provides "Strip trailing whitespace" under the "Format" menu.' |
| 2 | |
Srinivas Reddy Thatiparthy (శ్రీనివాస్ రెడ్డి తాటిపర్తి) | 9bb9223 | 2018-06-20 13:12:13 +0530 | [diff] [blame] | 3 | class Rstrip: |
Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 4 | |
Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 5 | def __init__(self, editwin): |
| 6 | self.editwin = editwin |
Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 7 | |
| 8 | def do_rstrip(self, event=None): |
| 9 | |
| 10 | text = self.editwin.text |
| 11 | undo = self.editwin.undo |
| 12 | |
| 13 | undo.undo_block_start() |
| 14 | |
Terry Jan Reedy | 4f133e2 | 2013-07-13 02:34:43 -0400 | [diff] [blame] | 15 | end_line = int(float(text.index('end'))) |
Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 16 | for cur in range(1, end_line): |
Terry Jan Reedy | 4f133e2 | 2013-07-13 02:34:43 -0400 | [diff] [blame] | 17 | txt = text.get('%i.0' % cur, '%i.end' % cur) |
| 18 | raw = len(txt) |
Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 19 | cut = len(txt.rstrip()) |
Terry Jan Reedy | 4f133e2 | 2013-07-13 02:34:43 -0400 | [diff] [blame] | 20 | # Since text.delete() marks file as changed, even if not, |
| 21 | # only call it when needed to actually delete something. |
| 22 | if cut < raw: |
| 23 | text.delete('%i.%i' % (cur, cut), '%i.end' % cur) |
Raymond Hettinger | eb81695 | 2009-05-29 01:22:18 +0000 | [diff] [blame] | 24 | |
| 25 | undo.undo_block_stop() |
Terry Jan Reedy | 4f133e2 | 2013-07-13 02:34:43 -0400 | [diff] [blame] | 26 | |
| 27 | if __name__ == "__main__": |
Terry Jan Reedy | 4d92158 | 2018-06-19 19:12:52 -0400 | [diff] [blame] | 28 | from unittest import main |
| 29 | main('idlelib.idle_test.test_rstrip', verbosity=2,) |