Merge remote-tracking branch 'origin/new/bug_8636'
[koha.git] / misc / migration_tools / bulkauthimport.pl
index cd80aba..2709db9 100755 (executable)
@@ -2,6 +2,7 @@
 # small script that import an iso2709 file into koha 2.0
 
 use strict;
+#use warnings; FIXME - Bug 2505
 BEGIN {
     # find Koha's Perl modules
     # test carefully before changing this
@@ -10,42 +11,48 @@ BEGIN {
 }
 
 # Koha modules used
+use Unicode::Normalize;
 use MARC::File::USMARC;
+use MARC::File::XML;
 use MARC::Record;
 use MARC::Batch;
 use C4::Context;
+use C4::Charset;
 use C4::AuthoritiesMarc;
 use Time::HiRes qw(gettimeofday);
-
 use Getopt::Long;
+use IO::File;
+
 my ( $input_marc_file, $number) = ('',0);
-my ($version, $delete, $test_parameter,$char_encoding, $verbose);
+my ($version, $delete, $test_parameter,$char_encoding, $verbose, $format, $commit);
+$| = 1;
 GetOptions(
     'file:s'    => \$input_marc_file,
-    'n' => \$number,
+    'n:i' => \$number,
     'h' => \$version,
     'd' => \$delete,
     't' => \$test_parameter,
     'c:s' => \$char_encoding,
     'v:s' => \$verbose,
+    'm:s' => \$format,
+    'commit:f' => \$commit,
 );
 
 if ($version || ($input_marc_file eq '')) {
-       print <<EOF
+    print <<EOF
 small script to import an iso2709 file into Koha.
 parameters :
 \th : this version/help screen
 \tfile /path/to/file/to/dump : the file to dump
-\tv : verbose mode. 1 means "some infos", 2 means "MARC dumping"
-\tn : the number of the record to import. If missing, all the file is imported
-\tt : test mode : parses the file, saying what he would do, but doing nothing.
-\tc : the char encoding. At the moment, only MARC21 and UNIMARC supported. MARC21 by default.
-\td : delete EVERYTHING related to biblio in koha-DB before import  :tables :
-\t\tbiblio, \t\tbiblioitems, \t\tsubjects,\titems
-\tmarc_biblio,
-\t\tmarc_subfield_table, \tmarc_word, \t\tmarc_blob_subfield
-IMPORTANT : don't use this script before you've entered and checked twice (or more) your  MARC parameters tables.
-If you fail this, the import won't work correctly and you will get invalid datas.
+\tv : verbose mode. Valid modes are 1 and 2
+\tn : the number of the record to import. If missing, the whole file is imported
+\tt : test mode : parses the file, saying what it would do, but doing nothing.
+\tc : the MARC flavour. At the moment, MARC21 and UNIMARC supported. MARC21 by default.
+\td : delete EVERYTHING related to authorities in koha-DB before import
+\tm : format, MARCXML or ISO2709
+\tcommit : the number of records to wait before performing a 'commit' operation
+IMPORTANT : don't use this script before you've entered and checked twice (or more) your MARC parameters tables.
+If you fail the test, the import won't work correctly and you will get invalid datas.
 
 SAMPLE : ./bulkmarcimport.pl -file /home/paul/koha.dev/local/npl -n 1
 EOF
@@ -56,70 +63,92 @@ die;
 my $dbh = C4::Context->dbh;
 
 if ($delete) {
-       print "deleting authorities\n";
-       $dbh->do("delete from auth_header");
+    print "deleting authorities\n";
+    $dbh->do("truncate auth_header");
 }
 if ($test_parameter) {
-       print "TESTING MODE ONLY\n    DOING NOTHING\n===============\n";
+    print "TESTING MODE ONLY\n    DOING NOTHING\n===============\n";
 }
 
+my $marcFlavour = C4::Context->preference('marcflavour') || 'MARC21';
 $char_encoding = 'MARC21' unless ($char_encoding);
 print "CHAR : $char_encoding\n" if $verbose;
 my $starttime = gettimeofday;
-my $batch = MARC::Batch->new( 'USMARC', $input_marc_file );
+my $fh = IO::File->new($input_marc_file); # don't let MARC::Batch open the file, as it applies the ':utf8' IO layer
+my $batch;
+if ($format =~ /XML/i) {
+    # ugly hack follows -- MARC::File::XML, when used by MARC::Batch,
+    # appears to try to convert incoming XML records from MARC-8
+    # to UTF-8.  Setting the BinaryEncoding key turns that off
+    # TODO: see what happens to ISO-8859-1 XML files.
+    # TODO: determine if MARC::Batch can be fixed to handle
+    #       XML records properly -- it probably should be
+    #       be using a proper push or pull XML parser to
+    #       extract the records, not using regexes to look
+    #       for <record>.*</record>.
+    $MARC::File::XML::_load_args{BinaryEncoding} = 'utf-8';
+    $batch = MARC::Batch->new( 'XML', $fh );
+} else {
+    $batch = MARC::Batch->new( 'USMARC', $fh );
+}
 $batch->warnings_off();
 $batch->strict_off();
 my $i=0;
-while ( my $record = $batch->next() ) {
-       $i++;
-       #now, parse the record, extract the item fields, and store them in somewhere else.
 
-    ## create an empty record object to populate
-    my $newRecord = MARC::Record->new();
-       $newRecord->leader($record->leader);
-    # go through each field in the existing record
-    foreach my $oldField ( $record->fields() ) {
-               # just reproduce tags < 010 in our new record
-               if ( $oldField->tag() < 10 ) {
-                       $newRecord->append_fields( $oldField );
-                       next();
-               }
-               # store our new subfield data in this list
-               my @newSubfields = ();
-       
-               # go through each subfield code/data pair
-               foreach my $pair ( $oldField->subfields() ) { 
-                       $pair->[1] =~ s/\<//g;
-                       $pair->[1] =~ s/\>//g;
-                       push( @newSubfields, $pair->[0], char_decode($pair->[1],$char_encoding) );
-               }
-       
-               # add the new field to our new record
-               my $newField = MARC::Field->new(
-                       $oldField->tag(),
-                       $oldField->indicator(1),
-                       $oldField->indicator(2),
-                       @newSubfields
-               );
-               $newRecord->append_fields( $newField );
+$dbh->{AutoCommit} = 0;
+my $commitnum = 50;
+if ($commit) {
+    $commitnum = $commit;
+}
+
+RECORD: while ( my $record = $batch->next() ) {
+    $i++;
+    print ".";
+    print "\r$i" unless $i % 100;
+
+    if ($record->encoding() eq 'MARC-8') {
+        my ($guessed_charset, $charset_errors);
+        ($record, $guessed_charset, $charset_errors) = MarcToUTF8Record($record, $marcFlavour);
+        if ($guessed_charset eq 'failed') {
+            warn "ERROR: failed to perform character conversion for record $i\n";
+            next RECORD;            
+        }
     }
-       warn "$i ==>".$newRecord->as_formatted() if $verbose eq 2;
-       my $authtypecode=substr($newRecord->leader(),9,1);
-       $authtypecode="NP" if ($authtypecode eq 'a'); # personnes
-       $authtypecode="CO" if ($authtypecode eq 'b'); # collectivit�
-       $authtypecode="NG" if ($authtypecode eq 'c'); # g�graphique
-       $authtypecode="NM" if ($authtypecode eq 'd'); # marque
-       $authtypecode="NF" if ($authtypecode eq 'e'); # famille
-       $authtypecode="TI" if ($authtypecode eq 'f'); # Titre uniforme
-       $authtypecode="TI" if ($authtypecode eq 'h'); # auteur/titre
-       $authtypecode="MM" if ($authtypecode eq 'j'); # mot mati�e
-       warn "XX => $authtypecode";
-       # now, create biblio and items with NEWnewXX call.
-       unless ($test_parameter) {
-               my ($authid) = AddAuthority($newRecord,0,$authtypecode);
-               warn "ADDED authority NB $authid in DB\n" if $verbose;
-       }
+
+    warn "$i ==>".$record->as_formatted() if $verbose eq 2;
+    my $authtypecode;
+    if (C4::Context->preference('marcflavour') eq 'MARC21') {
+        $authtypecode="PERSO_NAME" if ($record->field('100'));
+        $authtypecode="CORPO_NAME" if ($record->field('110'));
+        $authtypecode="MEETI_NAME" if ($record->field('111'));
+        $authtypecode="UNIF_TITLE" if ($record->field('130'));
+        $authtypecode="CHRON_TERM" if ($record->field('148') or $record->field('182'));
+        $authtypecode="TOPIC_TERM" if ($record->field('150') or $record->field('180'));
+        $authtypecode="GEOGR_NAME" if ($record->field('151') or $record->field('181'));
+        $authtypecode="GENRE/FORM" if ($record->field('155') or $record->field('185'));
+        next unless $authtypecode; # skip invalid records FIXME: far too simplistic
+    }
+    else {
+        $authtypecode=substr($record->leader(),9,1);
+        $authtypecode="NP" if ($authtypecode eq 'a'); # personnes
+        $authtypecode="CO" if ($authtypecode eq 'b'); # collectivit�
+        $authtypecode="NG" if ($authtypecode eq 'c'); # g�graphique
+        $authtypecode="NM" if ($authtypecode eq 'd'); # marque
+        $authtypecode="NF" if ($authtypecode eq 'e'); # famille
+        $authtypecode="TI" if ($authtypecode eq 'f'); # Titre uniforme
+        $authtypecode="TI" if ($authtypecode eq 'h'); # auteur/titre
+        $authtypecode="MM" if ($authtypecode eq 'j'); # mot mati�e
+    }
+
+    unless ($test_parameter) {
+        my ($authid) = AddAuthority($record,0,$authtypecode);
+        warn "ADDED authority NB $authid in DB\n" if $verbose;
+        $dbh->commit() if (0 == $i % $commitnum);
+    }
+
+    last if $i ==  $number;
 }
-# $dbh->do("unlock tables");
+$dbh->commit();
+
 my $timeneeded = gettimeofday - $starttime;
-print "$i MARC record done in $timeneeded seconds";
+print "\n$i MARC record done in $timeneeded seconds\n";