cmd/mux-cmd.py

   1 #!/usr/bin/env python
   2 import os, sys, subprocess, struct
   3 from bup import options
   4 from bup.helpers import *
   5
   6 optspec = """
   7 bup mux command [command arguments...]
   8 --
   9 """
  10 o = options.Options(optspec)
  11 (opt, flags, extra) = o.parse(sys.argv[1:])
  12 if len(extra) < 1:
  13     o.fatal('command is required')
  14
  15 cmdpath, cmdfn = os.path.split(__file__)
  16 subcmd = extra
  17 subcmd[0] = os.path.join(cmdpath, 'bup-' + subcmd[0])
  18
  19 debug2('bup mux: starting %r\n' % (extra,))
  20
  21 outr, outw = os.pipe()
  22 errr, errw = os.pipe()
  23 def close_fds():
  24     os.close(outr)
  25     os.close(errr)
  26 p = subprocess.Popen(subcmd, stdout=outw, stderr=errw, preexec_fn=close_fds)
  27 os.close(outw)
  28 os.close(errw)
  29 sys.stdout.write('BUPMUX')
  30 sys.stdout.flush()
  31 mux(p, sys.stdout.fileno(), outr, errr)
  32 os.close(outr)
  33 os.close(errr)
  34 prv = p.wait()
  35
  36 if prv:
  37     debug1('%s exited with code %d\n' % (extra[0], prv))
  38
  39 debug1('bup mux: done\n')
  40
  41 sys.exit(prv)