X-Git-Url: https://arthur.barton.de/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=cmd%2Frandom-cmd.py;h=3eef820143be8ed3f506ce4af8381c6511fdca71;hb=053869284db064aa21b2249981ec6affa489e296;hp=46fd4f2673db0cfe563f9f97f8eb0945cf0a07e7;hpb=d254bde8d204a02fb3f675dad7051fb2e5d58381;p=bup.git

diff --git a/cmd/random-cmd.py b/cmd/random-cmd.py
index 46fd4f2..3eef820 100755
--- a/cmd/random-cmd.py
+++ b/cmd/random-cmd.py
@@ -1,15 +1,25 @@
-#!/usr/bin/env python
-import sys, mmap
-from bup import options, _faster
-from bup.helpers import *
+#!/bin/sh
+"""": # -*-python-*-
+bup_python="$(dirname "$0")/bup-python" || exit $?
+exec "$bup_python" "$0" ${1+"$@"}
+"""
+# end of bup preamble
+
+from __future__ import absolute_import
+import os, sys
+
+from bup import options, _helpers
+from bup.helpers import atoi, handle_ctrl_c, log, parse_num
+
 
 optspec = """
 bup random [-S seed] <numbytes>
 --
 S,seed=   optional random number seed [1]
 f,force   print random data to stdout even if it's a tty
+v,verbose print byte counter to stderr
 """
-o = options.Options('bup random', optspec)
+o = options.Options(optspec)
 (opt, flags, extra) = o.parse(sys.argv[1:])
 
 if len(extra) != 1:
@@ -21,7 +31,8 @@ handle_ctrl_c()
 
 if opt.force or (not os.isatty(1) and
                  not atoi(os.environ.get('BUP_FORCE_TTY')) & 1):
-    _faster.write_random(sys.stdout.fileno(), total, opt.seed)
+    _helpers.write_random(sys.stdout.fileno(), total, opt.seed,
+                          opt.verbose and 1 or 0)
 else:
     log('error: not writing binary data to a terminal. Use -f to force.\n')
     sys.exit(1)