add -k option for pkgpart
authorivan <ivan>
Sun, 24 Aug 2008 21:52:34 +0000 (21:52 +0000)
committerivan <ivan>
Sun, 24 Aug 2008 21:52:34 +0000 (21:52 +0000)
bin/customer-faker

index 9076df0..236a412 100755 (executable)
@@ -12,15 +12,16 @@ use FS::svc_acct;
 
 my $refnum = 1;
 
-my @pkgs = ( 2, 3, 4 );
 #my @pkgs = ( 4, 5, 6 );
 my $svcpart = 2;
 
-use vars qw( $opt_p $opt_a );
-getopts('p:a:');
+use vars qw( $opt_p $opt_a $opt_k );
+getopts('p:a:k:');
 
 my $agentnum = $opt_a || 1;
 
+my @pkgs = $opt_k ? split(/,\s*/, $opt_k) : ( 2, 3, 4 );
+
 my $user = shift or die &usage;
 my $num = shift or die &usage;
 adminsuidsetup($user);
@@ -119,5 +120,5 @@ print "$onum customers inserted in $sec seconds ($persec customers/sec)\n";
 #---
 
 sub usage {
-  die "Usage:\n\n  customer-faker [ -p payby ] [ -a agentnum ] user num_fakes\n";
+  die "Usage:\n\n  customer-faker [ -p payby ] [ -a agentnum ] [ -k pkgpart,pkgpart,pkgpart... ] user num_fakes\n";
 }