iomenu.py

importio
importos
importshlex
importsys
importtempfile
importtokenize

fromtkinterimportfiledialog
fromtkinterimportmessagebox
fromtkinter.simpledialogimportaskstring# loadfile encoding.

fromidlelib.configimportidleConf
fromidlelib.utilimportpy_extensions

py_extensions=' '.join("*"+extforextinpy_extensions)
encoding='utf-8'
errors='surrogatepass'ifsys.platform=='win32'else'surrogateescape'


classIOBinding:
# One instance per editor Window so methods know which to save, close.
# Open returns focus to self.editwin if aborted.
# EditorWindow.open_module, others, belong here.

def__init__(self, editwin):
self.editwin=editwin
self.text=editwin.text
self.__id_open=self.text.bind("<<open-window-from-file>>", self.open)
self.__id_save=self.text.bind("<<save-window>>", self.save)
self.__id_saveas=self.text.bind("<<save-window-as-file>>",
self.save_as)
self.__id_savecopy=self.text.bind("<<save-copy-of-window-as-file>>",
self.save_a_copy)
self.fileencoding='utf-8'
self.__id_print=self.text.bind("<<print-window>>", self.print_window)

defclose(self):
# Undo command bindings
self.text.unbind("<<open-window-from-file>>", self.__id_open)
self.text.unbind("<<save-window>>", self.__id_save)
self.text.unbind("<<save-window-as-file>>",self.__id_saveas)
self.text.unbind("<<save-copy-of-window-as-file>>", self.__id_savecopy)
self.text.unbind("<<print-window>>", self.__id_print)
# Break cycles
self.editwin=None
self.text=None
self.filename_change_hook=None

defget_saved(self):
returnself.editwin.get_saved()

defset_saved(self, flag):
self.editwin.set_saved(flag)

defreset_undo(self):
self.editwin.reset_undo()

filename_change_hook=None

defset_filename_change_hook(self, hook):
self.filename_change_hook=hook

filename=None
dirname=None

defset_filename(self, filename):
iffilenameandos.path.isdir(filename):
self.filename=None
self.dirname=filename
else:
self.filename=filename
self.dirname=None
self.set_saved(1)
ifself.filename_change_hook:
self.filename_change_hook()

defopen(self, event=None, editFile=None):
flist=self.editwin.flist
# Save in case parent window is closed (ie, during askopenfile()).
ifflist:
ifnoteditFile:
filename=self.askopenfile()
else:
filename=editFile
iffilename:
# If editFile is valid and already open, flist.open will
# shift focus to its existing window.
# If the current window exists and is a fresh unnamed,
# unmodified editor window (not an interpreter shell),
# pass self.loadfile to flist.open so it will load the file
# in the current window (if the file is not already open)
# instead of a new window.
if (self.editwinand
notgetattr(self.editwin, 'interp', None) and
notself.filenameand
self.get_saved()):
flist.open(filename, self.loadfile)
else:
flist.open(filename)
else:
ifself.text:
self.text.focus_set()
return"break"

# Code for use outside IDLE:
ifself.get_saved():
reply=self.maybesave()
ifreply=="cancel":
self.text.focus_set()
return"break"
ifnoteditFile:
filename=self.askopenfile()
else:
filename=editFile
iffilename:
self.loadfile(filename)
else:
self.text.focus_set()
return"break"

eol_convention=os.linesep# default

defloadfile(self, filename):
try:
try:
withtokenize.open(filename) asf:
chars=f.read()
fileencoding=f.encoding
eol_convention=f.newlines
converted=False
except (UnicodeDecodeError, SyntaxError):
# Wait for the editor window to appear
self.editwin.text.update()
enc=askstring(
"Specify file encoding",
"The file's encoding is invalid for Python 3.x.\n"
"IDLE will convert it to UTF-8.\n"
"What is the current encoding of the file?",
initialvalue='utf-8',
parent=self.editwin.text)
withopen(filename, encoding=enc) asf:
chars=f.read()
fileencoding=f.encoding
eol_convention=f.newlines
converted=True
exceptOSErroraserr:
messagebox.showerror("I/O Error", str(err), parent=self.text)
returnFalse
exceptUnicodeDecodeError:
messagebox.showerror("Decoding Error",
"File %s\nFailed to Decode"%filename,
parent=self.text)
returnFalse

ifnotisinstance(eol_convention, str):
# If the file does not contain line separators, it is None.
# If the file contains mixed line separators, it is a tuple.
ifeol_conventionisnotNone:
messagebox.showwarning("Mixed Newlines",
"Mixed newlines detected.\n"
"The file will be changed on save.",
parent=self.text)
converted=True
eol_convention=os.linesep# default

self.text.delete("1.0", "end")
self.set_filename(None)
self.fileencoding=fileencoding
self.eol_convention=eol_convention
self.text.insert("1.0", chars)
self.reset_undo()
self.set_filename(filename)
ifconverted:
# We need to save the conversion results first
# before being able to execute the code
self.set_saved(False)
self.text.mark_set("insert", "1.0")
self.text.yview("insert")
self.updaterecentfileslist(filename)
returnTrue

defmaybesave(self):
"""Return 'yes', 'no', 'cancel' as appropriate.

 Tkinter messagebox.askyesnocancel converts these tk responses
 to True, False, None. Convert back, as now expected elsewhere.
 """
ifself.get_saved():
return"yes"
message= ("Do you want to save "
f"{self.filenameor'this untitled document'}"
" before closing?")
confirm=messagebox.askyesnocancel(
title="Save On Close",
message=message,
default=messagebox.YES,
parent=self.text)
ifconfirm:
self.save(None)
reply="yes"ifself.get_saved() else"cancel"
else: reply="cancel"ifconfirmisNoneelse"no"
self.text.focus_set()
returnreply

defsave(self, event):
ifnotself.filename:
self.save_as(event)
else:
ifself.writefile(self.filename):
self.set_saved(True)
try:
self.editwin.store_file_breaks()
exceptAttributeError: # may be a PyShell
pass
self.text.focus_set()
return"break"

defsave_as(self, event):
filename=self.asksavefile()
iffilename:
ifself.writefile(filename):
self.set_filename(filename)
self.set_saved(1)
try:
self.editwin.store_file_breaks()
exceptAttributeError:
pass
self.text.focus_set()
self.updaterecentfileslist(filename)
return"break"

defsave_a_copy(self, event):
filename=self.asksavefile()
iffilename:
self.writefile(filename)
self.text.focus_set()
self.updaterecentfileslist(filename)
return"break"

defwritefile(self, filename):
text=self.fixnewlines()
chars=self.encode(text)
try:
withopen(filename, "wb") asf:
f.write(chars)
f.flush()
os.fsync(f.fileno())
returnTrue
exceptOSErrorasmsg:
messagebox.showerror("I/O Error", str(msg),
parent=self.text)
returnFalse

deffixnewlines(self):
"""Return text with os eols.

 Add prompts if shell else final \n if missing.
 """

ifhasattr(self.editwin, "interp"): # Saving shell.
text=self.editwin.get_prompt_text('1.0', self.text.index('end-1c'))
else:
ifself.text.get("end-2c") !='\n':
self.text.insert("end-1c", "\n") # Changes 'end-1c' value.
text=self.text.get('1.0', "end-1c")
ifself.eol_convention!="\n":
text=text.replace("\n", self.eol_convention)
returntext

defencode(self, chars):
ifisinstance(chars, bytes):
# This is either plain ASCII, or Tk was returning mixed-encoding
# text to us. Don't try to guess further.
returnchars
# Preserve a BOM that might have been present on opening
ifself.fileencoding=='utf-8-sig':
returnchars.encode('utf-8-sig')
# See whether there is anything non-ASCII in it.
# If not, no need to figure out the encoding.
try:
returnchars.encode('ascii')
exceptUnicodeEncodeError:
pass
# Check if there is an encoding declared
try:
encoded=chars.encode('ascii', 'replace')
enc, _=tokenize.detect_encoding(io.BytesIO(encoded).readline)
returnchars.encode(enc)
exceptSyntaxErroraserr:
failed=str(err)
exceptUnicodeEncodeError:
failed="Invalid encoding '%s'"%enc
messagebox.showerror(
"I/O Error",
"%s.\nSaving as UTF-8"%failed,
parent=self.text)
# Fallback: save as UTF-8, with BOM - ignoring the incorrect
# declared encoding
returnchars.encode('utf-8-sig')

defprint_window(self, event):
confirm=messagebox.askokcancel(
title="Print",
message="Print to Default Printer",
default=messagebox.OK,
parent=self.text)
ifnotconfirm:
self.text.focus_set()
return"break"
tempfilename=None
saved=self.get_saved()
ifsaved:
filename=self.filename
# shell undo is reset after every prompt, looks saved, probably isn't
ifnotsavedorfilenameisNone:
 (tfd, tempfilename) =tempfile.mkstemp(prefix='IDLE_tmp_')
filename=tempfilename
os.close(tfd)
ifnotself.writefile(tempfilename):
os.unlink(tempfilename)
return"break"
platform=os.name
printPlatform=True
ifplatform=='posix': #posix platform
command=idleConf.GetOption('main','General',
'print-command-posix')
command=command+" 2>&1"
elifplatform=='nt': #win32 platform
command=idleConf.GetOption('main','General','print-command-win')
else: #no printing for this platform
printPlatform=False
ifprintPlatform: #we can try to print for this platform
command=command%shlex.quote(filename)
pipe=os.popen(command, "r")
# things can get ugly on NT if there is no printer available.
output=pipe.read().strip()
status=pipe.close()
ifstatus:
output="Printing failed (exit status 0x%x)\n"% \
status+output
ifoutput:
output="Printing command: %s\n"%repr(command) +output
messagebox.showerror("Print status", output, parent=self.text)
else: #no printing for this platform
message="Printing is not enabled for this platform: %s"%platform
messagebox.showinfo("Print status", message, parent=self.text)
iftempfilename:
os.unlink(tempfilename)
return"break"

opendialog=None
savedialog=None

filetypes= (
 ("Python files", py_extensions, "TEXT"),
 ("Text files", "*.txt", "TEXT"),
 ("All files", "*"),
 )

defaultextension='.py'ifsys.platform=='darwin'else''

defaskopenfile(self):
dir, base=self.defaultfilename("open")
ifnotself.opendialog:
self.opendialog=filedialog.Open(parent=self.text,
filetypes=self.filetypes)
filename=self.opendialog.show(initialdir=dir, initialfile=base)
returnfilename

defdefaultfilename(self, mode="open"):
ifself.filename:
returnos.path.split(self.filename)
elifself.dirname:
returnself.dirname, ""
else:
try:
pwd=os.getcwd()
exceptOSError:
pwd=""
returnpwd, ""

defasksavefile(self):
dir, base=self.defaultfilename("save")
ifnotself.savedialog:
self.savedialog=filedialog.SaveAs(
parent=self.text,
filetypes=self.filetypes,
defaultextension=self.defaultextension)
filename=self.savedialog.show(initialdir=dir, initialfile=base)
returnfilename

defupdaterecentfileslist(self,filename):
"Update recent file list on all editor windows"
ifself.editwin.flist:
self.editwin.update_recent_files_list(filename)


def_io_binding(parent): # htest #
fromtkinterimportToplevel, Text

top=Toplevel(parent)
top.title("Test IOBinding")
x, y=map(int, parent.geometry().split('+')[1:])
top.geometry("+%d+%d"% (x, y+175))

classMyEditWin:
def__init__(self, text):
self.text=text
self.flist=None
self.text.bind("<Control-o>", self.open)
self.text.bind('<Control-p>', self.print)
self.text.bind("<Control-s>", self.save)
self.text.bind("<Alt-s>", self.saveas)
self.text.bind('<Control-c>', self.savecopy)
defget_saved(self): return0
defset_saved(self, flag): pass
defreset_undo(self): pass
defopen(self, event):
self.text.event_generate("<<open-window-from-file>>")
defprint(self, event):
self.text.event_generate("<<print-window>>")
defsave(self, event):
self.text.event_generate("<<save-window>>")
defsaveas(self, event):
self.text.event_generate("<<save-window-as-file>>")
defsavecopy(self, event):
self.text.event_generate("<<save-copy-of-window-as-file>>")

text=Text(top)
text.pack()
text.focus_set()
editwin=MyEditWin(text)
IOBinding(editwin)


if__name__=="__main__":
fromunittestimportmain
main('idlelib.idle_test.test_iomenu', verbosity=2, exit=False)

fromidlelib.idle_test.htestimportrun
run(_io_binding)