some extra arguments for mig-stagebibs to make it easier to, for example, load patron...

[migration-tools.git] / mig-bin / mig-stagebibs

diff --git a/mig-bin/mig-stagebibs b/mig-bin/mig-stagebibs
index e64fd82..f045a13 100755 (executable)
--- a/mig-bin/mig-stagebibs
+++ b/mig-bin/mig-stagebibs
@@ -11,7 +11,13 @@ so records should be checked before hand and cleaned.
 
 Takes three optional arguments:
 
---source
+
+--source 
+
+Takes a numeric value and set the x_source of the bib record to that.  Defaults to 
+2 which is local system.
+
+--x_source
 
 Sets an x_source value on the staging table to the one supplied instead of the 
 default of none.
@@ -54,11 +60,14 @@ pod2usage(-verbose => 2) if defined $ARGV[0] && $ARGV[0] eq '--help';
 pod2usage(-verbose => 1) if ! $ARGV[1];
 
 my $append = 0;
-my $base_table = 'biblio_record_entry';
-my $stage_table = 'biblio_record_entry_legacy';
+my $base_table;
+my $stage_table;
+my $marc_column = 'marc';
 my $auth = '';
 my $serial = '';
-my $source = 'default';
+my $source = 2;
+my $x_source = 'default';
+my $no_source_or_last_xact_id;
 my $dbh = Mig::db_connect();
 my $infile;
 my $i = 0;
@@ -69,26 +78,35 @@ my $ret = GetOptions(
     'file:s'              => \$infile,
     'serial:s'            => \$serial,
     'auth:s'              => \$auth,
-    'source:s'            => \$source,
+    'x_source:s'          => \$x_source,
+    'source:i'            => \$source,
     'base_table:s'        => \$base_table,
-    'stage_table:s'       => \$stage_table
+    'stage_table:s'       => \$stage_table,
+    'marc_column:s'       => \$marc_column,
+    'no_source_or_last_xact_id' => \$no_source_or_last_xact_id
 );
 
 #if in file is empty then fail
 #if auth and serial = 1 fail 
 
 if ($serial == 1) { 
-    $base_table = 'authority_record_entry';
-    $stage_table = 'authority_record_entry_legacy';
+    $base_table = 'm_authority_record_entry';
 }
 
 if ($auth == 1) {
-    $base_table = 'serial_record_entry';
-    $stage_table = 'serial_record_entry_legacy';       
+    $base_table = 'm_serial_record_entry';
 }
 
 if ($auth == 1 and $serial == 1) { abort('are you sure you want to load these as authorities and serials?'); }
 
+if (!$base_table) {
+    $base_table = 'm_biblio_record_entry';
+}
+
+if (!$stage_table) {
+    $stage_table = $base_table . '_legacy';
+}
+
 my $bre_test = check_for_table($dbh,$base_table);
 my $bre_legacy_test = check_for_table($dbh,$stage_table);
 if ($bre_test == 0 and $bre_legacy_test == 0 ) { create_bre($dbh); create_child_bre($dbh); }
@@ -97,8 +115,12 @@ if ($bre_test == 1 and $bre_legacy_test == 0 ) { create_child_bre($dbh); }
 my $xmig_test = check_for_column($dbh,$stage_table,'x_migrate');
 if ($xmig_test == 0) { add_column($dbh,$stage_table,'x_migrate','BOOLEAN DEFAULT TRUE'); }
 
-my $xsource_test = check_for_column($dbh,$stage_table,'x_source');
-if ($xsource_test == 0) { add_column($dbh,$stage_table,'x_source','TEXT'); }
+my $xx_source_test = check_for_column($dbh,$stage_table,'x_source');
+if ($xx_source_test == 0) { add_column($dbh,$stage_table,'x_source','TEXT'); }
+
+my $xmarc_test = check_for_column($dbh,$stage_table,$marc_column);
+if ($xmarc_test == 0) { add_column($dbh,$stage_table,$marc_column,'TEXT'); }
+
 
 #flatten out MARC XML FILE
 open my $xml, "<:encoding(utf8)", $infile or abort('could not open MARC XML file');
@@ -109,7 +131,7 @@ while(my $line = <$xml>) {
        chomp $line;
        $record = $record . $line;
        if ($line =~ /<\/record>$/) {
-               stage_record($dbh,$record,$source); 
+               stage_record($dbh,$record,$x_source,$source); 
                $record = '';
                $i++;
                if (($i % 100) == 0) { report_progress('Records stage', $i); }
@@ -165,12 +187,20 @@ sub report_progress {
 sub stage_record {
     my $dbh = shift;
     my $record = shift;
-       my $source = shift;
+       my $x_source = shift;
+    my $source = shift;
        my $last_xact = "'$MIGSCHEMA'";
        $record = '$_$' . $record . '$_$';
        my $sql;
-       if ($source eq 'default') { $sql = "INSERT INTO $MIGSCHEMA.$stage_table (last_xact_id,marc) VALUES ($last_xact,$record);"; }
-               else { $sql = "INSERT INTO $MIGSCHEMA.$stage_table (last_xact_id,marc,x_source) VALUES ($last_xact,$record,'$source');";  }
+    if ($no_source_or_last_xact_id) {
+        $sql = "INSERT INTO $MIGSCHEMA.$stage_table ($marc_column) VALUES ($record);";
+    } else {
+        if ($x_source eq 'default') {
+            $sql = "INSERT INTO $MIGSCHEMA.$stage_table (last_xact_id,$marc_column,source) VALUES ($last_xact,$record,$source);";
+        } else {
+            $sql = "INSERT INTO $MIGSCHEMA.$stage_table (last_xact_id,$marc_column,x_source,source) VALUES ($last_xact,$record,'$x_source',$source);";
+        }
+    }
     my $sth = $dbh->prepare($sql);
     $sth->execute();
        return;