home *** CD-ROM | disk | FTP | other *** search
/ Mac-Source 1994 July / Mac-Source_July_1994.iso / Other Langs / python / Lib / cmd.py < prev    next >
Encoding:
Python Source  |  1994-04-01  |  1.2 KB  |  64 lines  |  [TEXT/R*ch]
  1. # A generic class to build line-oriented command interpreters
  2.  
  3. import string
  4. import sys
  5. import linecache
  6.  
  7. PROMPT = '(Cmd) '
  8. IDENTCHARS = string.letters + string.digits + '_'
  9.  
  10. class Cmd:
  11.  
  12.     def __init__(self):
  13.         self.prompt = PROMPT
  14.         self.identchars = IDENTCHARS
  15.         self.lastcmd = ''
  16.  
  17.     def cmdloop(self):
  18.         stop = None
  19.         while not stop:
  20.             try:
  21.                 line = raw_input(self.prompt)
  22.             except EOFError:
  23.                 line = 'EOF'
  24.             stop = self.onecmd(line)
  25.  
  26.     def onecmd(self, line):
  27.         line = string.strip(line)
  28.         if not line:
  29.             line = self.lastcmd
  30.         else:
  31.             self.lastcmd = line
  32.         i, n = 0, len(line)
  33.         while i < n and line[i] in self.identchars: i = i+1
  34.         cmd, arg = line[:i], string.strip(line[i:])
  35.         if cmd == '':
  36.             return self.default(line)
  37.         else:
  38.             try:
  39.                 func = getattr(self, 'do_' + cmd)
  40.             except AttributeError:
  41.                 return self.default(line)
  42.             return func(arg)
  43.  
  44.     def default(self, line):
  45.         print '*** Unknown syntax:', line
  46.  
  47.     def do_help(self, arg):
  48.         if arg:
  49.             # XXX check arg syntax
  50.             try:
  51.                 func = getattr(self, 'help_' + arg)
  52.             except:
  53.                 print '*** No help on', `arg`
  54.                 return
  55.             func()
  56.         else:
  57.             import newdir
  58.             names = newdir.dir(self.__class__)
  59.             cmds = []
  60.             for name in names:
  61.                 if name[:3] == 'do_':
  62.                     cmds.append(name[3:])
  63.             print cmds
  64.