zzdummy.py

"""Example extension, also used for testing.

See extend.txt for more details on creating an extension.
See config-extension.def for configuring an extension.
"""

fromidlelib.configimportidleConf
fromfunctoolsimportwraps


defformat_selection(format_line):
"Apply a formatting function to all of the selected lines."

@wraps(format_line)
defapply(self, event=None):
head, tail, chars, lines=self.formatter.get_region()
forposinrange(len(lines) -1):
line=lines[pos]
lines[pos] =format_line(self, line)
self.formatter.set_region(head, tail, chars, lines)
return'break'

returnapply


classZzDummy:
"""Prepend or remove initial text from selected lines."""

# Extend the format menu.
menudefs= [
 ('format', [
 ('Z in', '<<z-in>>'),
 ('Z out', '<<z-out>>'),
 ] )
 ]

def__init__(self, editwin):
"Initialize the settings for this extension."
self.editwin=editwin
self.text=editwin.text
self.formatter=editwin.fregion

@classmethod
defreload(cls):
"Load class variables from config."
cls.ztext=idleConf.GetOption('extensions', 'ZzDummy', 'z-text')

@format_selection
defz_in_event(self, line):
"""Insert text at the beginning of each selected line.

 This is bound to the <<z-in>> virtual event when the extensions
 are loaded.
 """
returnf'{self.ztext}{line}'

@format_selection
defz_out_event(self, line):
"""Remove specific text from the beginning of each selected line.

 This is bound to the <<z-out>> virtual event when the extensions
 are loaded.
 """
zlength=0ifnotline.startswith(self.ztext) elselen(self.ztext)
returnline[zlength:]


ZzDummy.reload()


if__name__=="__main__":
importunittest
unittest.main('idlelib.idle_test.test_zzdummy', verbosity=2, exit=False)