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