Demo/tkinter/guido/ShellWindow.py

   1 import os
   2 import sys
   3 import string
   4 from Tkinter import *
   5 from ScrolledText import ScrolledText
   6 from Dialog import Dialog
   7 import signal
   8
   9 BUFSIZE = 512
  10
  11 class ShellWindow(ScrolledText):
  12
  13         def __init__(self, master=None, shell=None, **cnf):
  14                 if not shell:
  15                         try:
  16                                 shell = os.environ['SHELL']
  17                         except KeyError:
  18                                 shell = '/bin/sh'
  19                         shell = shell + ' -i'
  20                 args = string.split(shell)
  21                 shell = args[0]
  22
  23                 apply(ScrolledText.__init__, (self, master), cnf)
  24                 self.pos = '1.0'
  25                 self.bind('<Return>', self.inputhandler)
  26                 self.bind('<Control-c>', self.sigint)
  27                 self.bind('<Control-t>', self.sigterm)
  28                 self.bind('<Control-k>', self.sigkill)
  29                 self.bind('<Control-d>', self.sendeof)
  30
  31                 self.pid, self.fromchild, self.tochild = spawn(shell, args)
  32                 self.tk.createfilehandler(self.fromchild, READABLE,
  33                                           self.outputhandler)
  34
  35         def outputhandler(self, file, mask):
  36                 data = os.read(file, BUFSIZE)
  37                 if not data:
  38                         self.tk.deletefilehandler(file)
  39                         pid, sts = os.waitpid(self.pid, 0)
  40                         print 'pid', pid, 'status', sts
  41                         self.pid = None
  42                         detail = sts>>8
  43                         cause = sts & 0xff
  44                         if cause == 0:
  45                                 msg = "exit status %d" % detail
  46                         else:
  47                                 msg = "killed by signal %d" % (cause & 0x7f)
  48                                 if cause & 0x80:
  49                                         msg = msg + " -- core dumped"
  50                         Dialog(self.master,
  51                                text=msg,
  52                                title="Exit status",
  53                                bitmap='warning',
  54                                default=0,
  55                                strings=('OK',))
  56                         return
  57                 self.insert(END, data)
  58                 self.pos = self.index("end - 1 char")
  59                 self.yview_pickplace(END)
  60
  61         def inputhandler(self, *args):
  62                 if not self.pid:
  63                         self.no_process()
  64                         return "break"
  65                 self.insert(END, "\n")
  66                 line = self.get(self.pos, "end - 1 char")
  67                 self.pos = self.index(END)
  68                 os.write(self.tochild, line)
  69                 return "break"
  70
  71         def sendeof(self, *args):
  72                 if not self.pid:
  73                         self.no_process()
  74                         return "break"
  75                 os.close(self.tochild)
  76                 return "break"
  77
  78         def sendsig(self, sig):
  79                 if not self.pid:
  80                         self.no_process()
  81                         return "break"
  82                 os.kill(self.pid, sig)
  83                 return "break"
  84
  85         def sigint(self, *args):
  86                 return self.sendsig(signal.SIGINT)
  87
  88         def sigquit(self, *args):
  89                 return self.sendsig(signal.SIGQUIT)
  90
  91         def sigterm(self, *args):
  92                 return self.sendsig(signal.SIGTERM)
  93
  94         def sigkill(self, *args):
  95                 return self.sendsig(signal.SIGKILL)
  96
  97         def no_process(self):
  98                 Dialog(self.master,
  99                        text="No active process",
 100                        title="No process",
 101                        bitmap='error',
 102                        default=0,
 103                        strings=('OK',))
 104
 105 MAXFD = 100     # Max number of file descriptors (os.getdtablesize()???)
 106
 107 def spawn(prog, args):
 108         p2cread, p2cwrite = os.pipe()
 109         c2pread, c2pwrite = os.pipe()
 110         pid = os.fork()
 111         if pid == 0:
 112                 # Child
 113                 for i in 0, 1, 2:
 114                     try:
 115                         os.close(i)
 116                     except os.error:
 117                         pass
 118                 if os.dup(p2cread) <> 0:
 119                         sys.stderr.write('popen2: bad read dup\n')
 120                 if os.dup(c2pwrite) <> 1:
 121                         sys.stderr.write('popen2: bad write dup\n')
 122                 if os.dup(c2pwrite) <> 2:
 123                         sys.stderr.write('popen2: bad write dup\n')
 124                 for i in range(3, MAXFD):
 125                         try:
 126                                 os.close(i)
 127                         except:
 128                                 pass
 129                 try:
 130                         os.execvp(prog, args)
 131                 finally:
 132                         sys.stderr.write('execvp failed\n')
 133                         os._exit(1)
 134         os.close(p2cread)
 135         os.close(c2pwrite)
 136         return pid, c2pread, p2cwrite
 137
 138 def test():
 139         shell = string.join(sys.argv[1:])
 140         root = Tk()
 141         root.minsize(1, 1)
 142         if shell:
 143             w = ShellWindow(root, shell=shell)
 144         else:
 145             w = ShellWindow(root)
 146         w.pack(expand=1, fill=BOTH)
 147         w.focus_set()
 148         w.tk.mainloop()
 149
 150 if __name__ == '__main__':
 151         test()