Tools/idle/IOBinding.py

   1 import os
   2 import tempfile
   3 import tkFileDialog
   4 import tkMessageBox
   5 from IdleConf import idleconf
   6
   7 #$ event <<open-window-from-file>>
   8 #$ win <Control-o>
   9 #$ unix <Control-x><Control-f>
  10
  11 #$ event <<save-window>>
  12 #$ win <Control-s>
  13 #$ unix <Control-x><Control-s>
  14
  15 #$ event <<save-window-as-file>>
  16 #$ win <Alt-s>
  17 #$ unix <Control-x><Control-w>
  18
  19 #$ event <<save-copy-of-window-as-file>>
  20 #$ win <Alt-Shift-s>
  21 #$ unix <Control-x><w>
  22
  23 #$ event <<print-window>>
  24 #$ win <Control-p>
  25 #$ unix <Control-x><Control-p>
  26
  27
  28 class IOBinding:
  29
  30     def __init__(self, editwin):
  31         self.editwin = editwin
  32         self.text = editwin.text
  33         self.__id_open = self.text.bind("<<open-window-from-file>>", self.open)
  34         self.__id_save = self.text.bind("<<save-window>>", self.save)
  35         self.__id_saveas = self.text.bind("<<save-window-as-file>>",
  36                                           self.save_as)
  37         self.__id_savecopy = self.text.bind("<<save-copy-of-window-as-file>>",
  38                                             self.save_a_copy)
  39         self.__id_print = self.text.bind("<<print-window>>", self.print_window)
  40
  41     def close(self):
  42         # Undo command bindings
  43         self.text.unbind("<<open-window-from-file>>", self.__id_open)
  44         self.text.unbind("<<save-window>>", self.__id_save)
  45         self.text.unbind("<<save-window-as-file>>",self.__id_saveas)
  46         self.text.unbind("<<save-copy-of-window-as-file>>", self.__id_savecopy)
  47         self.text.unbind("<<print-window>>", self.__id_print)
  48         # Break cycles
  49         self.editwin = None
  50         self.text = None
  51         self.filename_change_hook = None
  52
  53     def get_saved(self):
  54         return self.editwin.get_saved()
  55
  56     def set_saved(self, flag):
  57         self.editwin.set_saved(flag)
  58
  59     def reset_undo(self):
  60         self.editwin.reset_undo()
  61
  62     filename_change_hook = None
  63
  64     def set_filename_change_hook(self, hook):
  65         self.filename_change_hook = hook
  66
  67     filename = None
  68
  69     def set_filename(self, filename):
  70         self.filename = filename
  71         self.set_saved(1)
  72         if self.filename_change_hook:
  73             self.filename_change_hook()
  74
  75     def open(self, event):
  76         if self.editwin.flist:
  77             filename = self.askopenfile()
  78             if filename:
  79                 self.editwin.flist.open(filename)
  80             else:
  81                 self.text.focus_set()
  82             return "break"
  83         # Code for use outside IDLE:
  84         if self.get_saved():
  85             reply = self.maybesave()
  86             if reply == "cancel":
  87                 self.text.focus_set()
  88                 return "break"
  89         filename = self.askopenfile()
  90         if filename:
  91             self.loadfile(filename)
  92         else:
  93             self.text.focus_set()
  94         return "break"
  95
  96     def loadfile(self, filename):
  97         try:
  98             f = open(filename)
  99             chars = f.read()
 100             f.close()
 101         except IOError, msg:
 102             tkMessageBox.showerror("I/O Error", str(msg), master=self.text)
 103             return False
 104         self.text.delete("1.0", "end")
 105         self.set_filename(None)
 106         self.text.insert("1.0", chars)
 107         self.reset_undo()
 108         self.set_filename(filename)
 109         self.text.mark_set("insert", "1.0")
 110         self.text.see("insert")
 111         return True
 112
 113     def maybesave(self):
 114         if self.get_saved():
 115             return "yes"
 116         message = "Do you want to save %s before closing?" % (
 117             self.filename or "this untitled document")
 118         m = tkMessageBox.Message(
 119             title="Save On Close",
 120             message=message,
 121             icon=tkMessageBox.QUESTION,
 122             type=tkMessageBox.YESNOCANCEL,
 123             master=self.text)
 124         reply = m.show()
 125         if reply == "yes":
 126             self.save(None)
 127             if not self.get_saved():
 128                 reply = "cancel"
 129         self.text.focus_set()
 130         return reply
 131
 132     def save(self, event):
 133         if not self.filename:
 134             self.save_as(event)
 135         else:
 136             if self.writefile(self.filename):
 137                 self.set_saved(1)
 138         self.text.focus_set()
 139         return "break"
 140
 141     def save_as(self, event):
 142         filename = self.asksavefile()
 143         if filename:
 144             if self.writefile(filename):
 145                 self.set_filename(filename)
 146                 self.set_saved(1)
 147         self.text.focus_set()
 148         return "break"
 149
 150     def save_a_copy(self, event):
 151         filename = self.asksavefile()
 152         if filename:
 153             self.writefile(filename)
 154         self.text.focus_set()
 155         return "break"
 156
 157     def print_window(self, event):
 158         tempfilename = None
 159         if self.get_saved():
 160             filename = self.filename
 161         else:
 162             filename = tempfilename = tempfile.mktemp()
 163             if not self.writefile(filename):
 164                 os.unlink(tempfilename)
 165                 return "break"
 166         edconf = idleconf.getsection('EditorWindow')
 167         command = edconf.get('print-command')
 168         command = command % filename
 169         if os.name == 'posix':
 170             command = command + " 2>&1"
 171         pipe = os.popen(command, "r")
 172         output = pipe.read().strip()
 173         status = pipe.close()
 174         if status:
 175             output = "Printing failed (exit status 0x%x)\n" % status + output
 176         if output:
 177             output = "Printing command: %s\n" % repr(command) + output
 178             tkMessageBox.showerror("Print status", output, master=self.text)
 179         return "break"
 180
 181     def writefile(self, filename):
 182         self.fixlastline()
 183         chars = str(self.text.get("1.0", "end-1c"))
 184         try:
 185             f = open(filename, "w")
 186             f.write(chars)
 187             f.close()
 188             ## print "saved to", `filename`
 189             return True
 190         except IOError, msg:
 191             tkMessageBox.showerror("I/O Error", str(msg),
 192                                    master=self.text)
 193             return False
 194
 195     def fixlastline(self):
 196         c = self.text.get("end-2c")
 197         if c != '\n':
 198             self.text.insert("end-1c", "\n")
 199
 200     opendialog = None
 201     savedialog = None
 202
 203     filetypes = [
 204         ("Python and text files", "*.py *.pyw *.txt", "TEXT"),
 205         ("All text files", "*", "TEXT"),
 206         ("All files", "*"),
 207         ]
 208
 209     def askopenfile(self):
 210         dir, base = self.defaultfilename("open")
 211         if not self.opendialog:
 212             self.opendialog = tkFileDialog.Open(master=self.text,
 213                                                 filetypes=self.filetypes)
 214         return self.opendialog.show(initialdir=dir, initialfile=base)
 215
 216     def defaultfilename(self, mode="open"):
 217         if self.filename:
 218             return os.path.split(self.filename)
 219         else:
 220             try:
 221                 pwd = os.getcwd()
 222             except os.error:
 223                 pwd = ""
 224             return pwd, ""
 225
 226     def asksavefile(self):
 227         dir, base = self.defaultfilename("save")
 228         if not self.savedialog:
 229             self.savedialog = tkFileDialog.SaveAs(master=self.text,
 230                                                   filetypes=self.filetypes)
 231         return self.savedialog.show(initialdir=dir, initialfile=base)
 232
 233
 234 def test():
 235     root = Tk()
 236     class MyEditWin:
 237         def __init__(self, text):
 238             self.text = text
 239             self.flist = None
 240             self.text.bind("<Control-o>", self.open)
 241             self.text.bind("<Control-s>", self.save)
 242             self.text.bind("<Alt-s>", self.save_as)
 243             self.text.bind("<Alt-z>", self.save_a_copy)
 244         def get_saved(self): return 0
 245         def set_saved(self, flag): pass
 246         def reset_undo(self): pass
 247         def open(self, event):
 248             self.text.event_generate("<<open-window-from-file>>")
 249         def save(self, event):
 250             self.text.event_generate("<<save-window>>")
 251         def save_as(self, event):
 252             self.text.event_generate("<<save-window-as-file>>")
 253         def save_a_copy(self, event):
 254             self.text.event_generate("<<save-copy-of-window-as-file>>")
 255     text = Text(root)
 256     text.pack()
 257     text.focus_set()
 258     editwin = MyEditWin(text)
 259     io = IOBinding(editwin)
 260     root.mainloop()
 261
 262 if __name__ == "__main__":
 263     from Tkinter import *
 264     test()