Lib/idlelib/IOBinding.py

   1 # changes by dscherer@cmu.edu
   2 #   - IOBinding.open() replaces the current window with the opened file,
   3 #     if the current window is both unmodified and unnamed
   4 #   - IOBinding.loadfile() interprets Windows, UNIX, and Macintosh
   5 #     end-of-line conventions, instead of relying on the standard library,
   6 #     which will only understand the local convention.
   7
   8 import os
   9 import tkFileDialog
  10 import tkMessageBox
  11 import re
  12
  13 #$ event <<open-window-from-file>>
  14 #$ win <Control-o>
  15 #$ unix <Control-x><Control-f>
  16
  17 #$ event <<save-window>>
  18 #$ win <Control-s>
  19 #$ unix <Control-x><Control-s>
  20
  21 #$ event <<save-window-as-file>>
  22 #$ win <Alt-s>
  23 #$ unix <Control-x><Control-w>
  24
  25 #$ event <<save-copy-of-window-as-file>>
  26 #$ win <Alt-Shift-s>
  27 #$ unix <Control-x><w>
  28
  29
  30 class IOBinding:
  31
  32     def __init__(self, editwin):
  33         self.editwin = editwin
  34         self.text = editwin.text
  35         self.__id_open = self.text.bind("<<open-window-from-file>>", self.open)
  36         self.__id_save = self.text.bind("<<save-window>>", self.save)
  37         self.__id_saveas = self.text.bind("<<save-window-as-file>>",
  38                                           self.save_as)
  39         self.__id_savecopy = self.text.bind("<<save-copy-of-window-as-file>>",
  40                                             self.save_a_copy)
  41
  42     def close(self):
  43         # Undo command bindings
  44         self.text.unbind("<<open-window-from-file>>", self.__id_open)
  45         self.text.unbind("<<save-window>>", self.__id_save)
  46         self.text.unbind("<<save-window-as-file>>",self.__id_saveas)
  47         self.text.unbind("<<save-copy-of-window-as-file>>", self.__id_savecopy)
  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                 # if the current window has no filename and hasn't been
  80                 #   modified, we replace it's contents (no loss).  Otherwise
  81                 #   we open a new window.
  82                 if not self.filename and self.get_saved():
  83                     self.editwin.flist.open(filename, self.loadfile)
  84                 else:
  85                     self.editwin.flist.open(filename)
  86             else:
  87                 self.text.focus_set()
  88
  89             return "break"
  90         # Code for use outside IDLE:
  91         if self.get_saved():
  92             reply = self.maybesave()
  93             if reply == "cancel":
  94                 self.text.focus_set()
  95                 return "break"
  96         filename = self.askopenfile()
  97         if filename:
  98             self.loadfile(filename)
  99         else:
 100             self.text.focus_set()
 101         return "break"
 102
 103     def loadfile(self, filename):
 104         try:
 105             # open the file in binary mode so that we can handle
 106             #   end-of-line convention ourselves.
 107             f = open(filename,'rb')
 108             chars = f.read()
 109             f.close()
 110         except IOError, msg:
 111             tkMessageBox.showerror("I/O Error", str(msg), master=self.text)
 112             return 0
 113
 114         # We now convert all end-of-lines to '\n's
 115         eol = r"(\r\n)|\n|\r"  # \r\n (Windows), \n (UNIX), or \r (Mac)
 116         chars = re.compile( eol ).sub( r"\n", chars )
 117
 118         self.text.delete("1.0", "end")
 119         self.set_filename(None)
 120         self.text.insert("1.0", chars)
 121         self.reset_undo()
 122         self.set_filename(filename)
 123         self.text.mark_set("insert", "1.0")
 124         self.text.see("insert")
 125         return 1
 126
 127     def maybesave(self):
 128         if self.get_saved():
 129             return "yes"
 130         message = "Do you want to save %s before closing?" % (
 131             self.filename or "this untitled document")
 132         m = tkMessageBox.Message(
 133             title="Save On Close",
 134             message=message,
 135             icon=tkMessageBox.QUESTION,
 136             type=tkMessageBox.YESNOCANCEL,
 137             master=self.text)
 138         reply = m.show()
 139         if reply == "yes":
 140             self.save(None)
 141             if not self.get_saved():
 142                 reply = "cancel"
 143         self.text.focus_set()
 144         return reply
 145
 146     def save(self, event):
 147         if not self.filename:
 148             self.save_as(event)
 149         else:
 150             if self.writefile(self.filename):
 151                 self.set_saved(1)
 152         self.text.focus_set()
 153         return "break"
 154
 155     def save_as(self, event):
 156         filename = self.asksavefile()
 157         if filename:
 158             if self.writefile(filename):
 159                 self.set_filename(filename)
 160                 self.set_saved(1)
 161         self.text.focus_set()
 162         return "break"
 163
 164     def save_a_copy(self, event):
 165         filename = self.asksavefile()
 166         if filename:
 167             self.writefile(filename)
 168         self.text.focus_set()
 169         return "break"
 170
 171     def writefile(self, filename):
 172         self.fixlastline()
 173         try:
 174             f = open(filename, "w")
 175             chars = self.text.get("1.0", "end-1c")
 176             f.write(chars)
 177             f.close()
 178             ## print "saved to", `filename`
 179             return 1
 180         except IOError, msg:
 181             tkMessageBox.showerror("I/O Error", str(msg),
 182                                    master=self.text)
 183             return 0
 184
 185     def fixlastline(self):
 186         c = self.text.get("end-2c")
 187         if c != '\n':
 188             self.text.insert("end-1c", "\n")
 189
 190     opendialog = None
 191     savedialog = None
 192
 193     filetypes = [
 194         ("Python and text files", "*.py *.pyw *.txt", "TEXT"),
 195         ("All text files", "*", "TEXT"),
 196         ("All files", "*"),
 197         ]
 198
 199     def askopenfile(self):
 200         dir, base = self.defaultfilename("open")
 201         if not self.opendialog:
 202             self.opendialog = tkFileDialog.Open(master=self.text,
 203                                                 filetypes=self.filetypes)
 204         return self.opendialog.show(initialdir=dir, initialfile=base)
 205
 206     def defaultfilename(self, mode="open"):
 207         if self.filename:
 208             return os.path.split(self.filename)
 209         else:
 210             try:
 211                 pwd = os.getcwd()
 212             except os.error:
 213                 pwd = ""
 214             return pwd, ""
 215
 216     def asksavefile(self):
 217         dir, base = self.defaultfilename("save")
 218         if not self.savedialog:
 219             self.savedialog = tkFileDialog.SaveAs(master=self.text,
 220                                                   filetypes=self.filetypes)
 221         return self.savedialog.show(initialdir=dir, initialfile=base)
 222
 223
 224 def test():
 225     from Tkinter import *
 226     root = Tk()
 227     class MyEditWin:
 228         def __init__(self, text):
 229             self.text = text
 230             self.flist = None
 231             self.text.bind("<Control-o>", self.open)
 232             self.text.bind("<Control-s>", self.save)
 233             self.text.bind("<Alt-s>", self.save_as)
 234             self.text.bind("<Alt-z>", self.save_a_copy)
 235         def get_saved(self): return 0
 236         def set_saved(self, flag): pass
 237         def reset_undo(self): pass
 238         def open(self, event):
 239             self.text.event_generate("<<open-window-from-file>>")
 240         def save(self, event):
 241             self.text.event_generate("<<save-window>>")
 242         def save_as(self, event):
 243             self.text.event_generate("<<save-window-as-file>>")
 244         def save_a_copy(self, event):
 245             self.text.event_generate("<<save-copy-of-window-as-file>>")
 246     text = Text(root)
 247     text.pack()
 248     text.focus_set()
 249     editwin = MyEditWin(text)
 250     io = IOBinding(editwin)
 251     root.mainloop()
 252
 253 if __name__ == "__main__":
 254     test()