From 8d350c66622ec8a2c3ee75d1fd2ed60f249d78a8 Mon Sep 17 00:00:00 2001
From: Avery Pennarun <apenwarr@gmail.com>
Date: Wed, 28 Jul 2010 02:37:38 -0400
Subject: [PATCH] cmd/{random,memtest}: use the new options.py default value
 support.

---
 cmd/memtest-cmd.py | 14 ++++++--------
 cmd/random-cmd.py  |  4 ++--
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/cmd/memtest-cmd.py b/cmd/memtest-cmd.py
index 41ffbfc..745550c 100755
--- a/cmd/memtest-cmd.py
+++ b/cmd/memtest-cmd.py
@@ -3,6 +3,7 @@ import sys, re, struct, mmap
 from bup import git, options
 from bup.helpers import *
 
+handle_ctrl_c()
 
 def s_from_bytes(bytes):
     clist = [chr(b) for b in bytes]
@@ -27,8 +28,8 @@ def report(count):
 optspec = """
 bup memtest [-n elements] [-c cycles]
 --
-n,number=  number of objects per cycle
-c,cycles=  number of cycles to run
+n,number=  number of objects per cycle [10000]
+c,cycles=  number of cycles to run [100]
 ignore-midx  ignore .midx files, use only .idx files
 """
 o = options.Options('bup memtest', optspec)
@@ -42,15 +43,12 @@ git.ignore_midx = opt.ignore_midx
 git.check_repo_or_die()
 m = git.PackIdxList(git.repo('objects/pack'))
 
-cycles = opt.cycles or 100
-number = opt.number or 10000
-
 report(-1)
 f = open('/dev/urandom')
 a = mmap.mmap(-1, 20)
 report(0)
-for c in xrange(cycles):
-    for n in xrange(number):
+for c in xrange(opt.cycles):
+    for n in xrange(opt.number):
         b = f.read(3)
         if 0:
             bytes = list(struct.unpack('!BBB', b)) + [0]*17
@@ -62,4 +60,4 @@ for c in xrange(cycles):
             bin = str(a[0:20])
         #print bin.encode('hex')
         m.exists(bin)
-    report((c+1)*number)
+    report((c+1)*opt.number)
diff --git a/cmd/random-cmd.py b/cmd/random-cmd.py
index 7e3eaa4..362a19c 100755
--- a/cmd/random-cmd.py
+++ b/cmd/random-cmd.py
@@ -6,7 +6,7 @@ from bup.helpers import *
 optspec = """
 bup random [-S seed] <numbytes>
 --
-S,seed=   optional random number seed (default 1)
+S,seed=   optional random number seed [1]
 f,force   print random data to stdout even if it's a tty
 """
 o = options.Options('bup random', optspec)
@@ -21,7 +21,7 @@ handle_ctrl_c()
 
 if opt.force or (not os.isatty(1) and
                  not atoi(os.environ.get('BUP_FORCE_TTY')) & 1):
-    _hashsplit.write_random(sys.stdout.fileno(), total, opt.seed or 0)
+    _hashsplit.write_random(sys.stdout.fileno(), total, opt.seed)
 else:
     log('error: not writing binary data to a terminal. Use -f to force.\n')
     sys.exit(1)
-- 
2.39.2