cmd/random-cmd.py

   1 #!/bin/sh
   2 """": # -*-python-*-
   3 bup_python="$(dirname "$0")/bup-python" || exit $?
   4 exec "$bup_python" "$0" ${1+"$@"}
   5 """
   6 # end of bup preamble
   7
   8 from __future__ import absolute_import
   9 import os, sys
  10
  11 from bup import options, _helpers
  12 from bup.helpers import atoi, handle_ctrl_c, log, parse_num
  13
  14
  15 optspec = """
  16 bup random [-S seed] <numbytes>
  17 --
  18 S,seed=   optional random number seed [1]
  19 f,force   print random data to stdout even if it's a tty
  20 v,verbose print byte counter to stderr
  21 """
  22 o = options.Options(optspec)
  23 (opt, flags, extra) = o.parse(sys.argv[1:])
  24
  25 if len(extra) != 1:
  26     o.fatal("exactly one argument expected")
  27
  28 total = parse_num(extra[0])
  29
  30 handle_ctrl_c()
  31
  32 if opt.force or (not os.isatty(1) and
  33                  not atoi(os.environ.get('BUP_FORCE_TTY')) & 1):
  34     _helpers.write_random(sys.stdout.fileno(), total, opt.seed,
  35                           opt.verbose and 1 or 0)
  36 else:
  37     log('error: not writing binary data to a terminal. Use -f to force.\n')
  38     sys.exit(1)