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