From: Shawn Boyette <sboyette@esilibrary.com>
Date: Mon, 25 Aug 2008 16:49:48 +0000 (+0000)
Subject: * getopt in effect
X-Git-Url: http://git.equinoxoli.org/?p=migration-tools.git;a=commitdiff_plain;h=827b82a5249ac11c483cbee7f031aa02fd21e58e

* getopt in effect
* code reflow
---

diff --git a/set_record_ids.pl b/set_record_ids.pl
index 911ed10..2f7d094 100755
--- a/set_record_ids.pl
+++ b/set_record_ids.pl
@@ -1,38 +1,99 @@
 #!/usr/bin/perl
+use strict;
+use warnings;
+
 use MARC::Batch;
-use MARC::Record;
-use MARC::File::XML ( BinaryEncoding => 'utf-8' );
-use MARC::Field;
+use Getopt::Long;
+#use MARC::Record;
+#use MARC::File::XML ( BinaryEncoding => 'utf-8' );
+#use MARC::Field;
 
-my $initial_record_number = $ARGV[0];
-my $record_tag_number = $ARGV[1];
-my $record_tag_subfield = $ARGV[2];
 my $count = 0;
+my $conf  = {}; # configuration hashref
+initialize($conf);
 
 binmode(STDOUT, ':utf8');
 binmode(STDIN, ':utf8');
 
-foreach $argnum ( 3 .. $#ARGV ) {
+foreach my $input ( @ARGV ) {
+    print STDERR "Processing $input, starting record id at ",
+      $conf->{'renumber-from'},"\n";
+
+    my $batch = MARC::Batch->new('XML', $input);
+    while ( my $record = $batch->next ) {
+        $count++;
+        my @warnings = $batch->warnings;
+        print STDERR "WARNINGS: Record $count : ",
+          join(":",@warnings), " : continuing...\n"
+            if ( @warnings );
+
+        while ($record->field($conf->{tag}))
+          { $record->delete_field( $record->field($conf->{tag}) ) }
+        my $new_id = $conf->{'renumber-from'} + $count - 1;
+        my $new_id_field = MARC::Field->new( $conf->{tag},
+                                             ' ',
+                                             ' ',
+                                             $conf->{subfield} => $new_id );
+        $record->append_fields($new_id_field);
+        print $record->as_xml;
+    }
+    print STDERR "Processed $count records.  Last record id at ",
+      ($conf->{'renumber-from'} + $count - 1), "\n";
+}
 
-	print STDERR "Processing " . $ARGV[$argnum] . ", starting record id at $initial_record_number\n";
 
-	my $batch = MARC::Batch->new('XML',$ARGV[$argnum]);
-	#$batch->strict_off();
-	#$batch->warnings_off();
+=head2 initialize
 
-	while ( my $record = $batch->next() ) {
+Performs boring script initialization. Handles argument parsing,
+mostly.
 
-        $count++;
+=cut
 
-        print STDERR "WARNINGS: Record $count : " . join(":",@warnings) . " : continuing...\n" if ( @warnings );
+sub initialize {
+    my ($c) = @_;
+    my @missing = ();
 
-        while ($record->field($record_tag_number)) { $record->delete_field( $record->field($record_tag_number) ); }
-        my $new_id = $initial_record_number + $count - 1;
-        my $new_id_field = MARC::Field->new( $record_tag_number, ' ', ' ', $record_tag_subfield => $new_id);
-        $record->append_fields($new_id_field);
+    # set mode on existing filehandles
+    binmode(STDIN, ':utf8');
+
+    my $rc = GetOptions( $c,
+                         'output|o=s',
+                         'renumber-from|rf=i',
+                         'subfield|s=s',
+                         'tag|t=s',
+                         'help|h',
+                       );
+    show_help() unless $rc;
+
+    my @keys = keys %{$c};
+    show_help() unless (@ARGV and @keys);
+    for my $key ('renumber-from', 'tag', 'subfield', 'output')
+      { push @missing, $key unless $c->{$key} }
+    if (@missing) {
+        print "Required option: ", join(', ', @missing), " missing!\n";
+        show_help();
+    }
+
+    show_help() if ($c->{help});
+}
+
+
+=head2 show_help
+
+Display usage message when things go wrong
+
+=cut
 
-        print $record->as_xml();
-	}
+sub show_help {
+print <<HELP;
+Usage is: $0 [REQUIRED ARGS] <filelist>
+Req'd Arguments
+  --renumber-from=N        -rf First id# of new sequence
+  --tag=N                  -t  Which tag to use
+  --subfield=X             -s  Which subfield to use
+  --output=<file>          -o  Output filename
 
-    print STDERR "Processed $count records.  Last record id at " . ($initial_record_number + $count - 1) . "\n";
+Any number of input files may be specified; one output file will result.
+HELP
+exit 1;
 }