r1393@llin: dpavlin | 2007-10-31 00:52:52 +0100
[webpac2] / run.pl
diff --git a/run.pl b/run.pl
index 6b7311f..03de8b3 100755 (executable)
--- a/run.pl
+++ b/run.pl
@@ -7,12 +7,12 @@ use File::Temp qw/tempdir/;
 use lib './lib';
 
 use WebPAC::Common 0.02;
-use WebPAC::Parser 0.04;
-use WebPAC::Input 0.13;
-use WebPAC::Store 0.10;
-use WebPAC::Normalize 0.11;
+use WebPAC::Parser 0.08;
+use WebPAC::Input 0.16;
+use WebPAC::Store 0.15;
+use WebPAC::Normalize 0.22;
 use WebPAC::Output::TT;
-use WebPAC::Validate 0.06;
+use WebPAC::Validate 0.11;
 use WebPAC::Output::MARC;
 use WebPAC::Config;
 use Getopt::Long;
@@ -21,6 +21,7 @@ use Time::HiRes qw/time/;
 use File::Slurp;
 use Data::Dump qw/dump/;
 use Storable qw/dclone/;
+use Pod::Usage qw/pod2usage/;
 
 use Proc::Queue size => 1;
 use POSIX ":sys_wait_h"; # imports WNOHANG
@@ -31,7 +32,7 @@ run.pl - start WebPAC indexing
 
 B<this command will probably go away. Don't get used to it!>
 
-Options:
+=head1 OPTIONS
 
 =over 4
 
@@ -67,17 +68,22 @@ and subfield usage for each input
 
 turn on extra validation of imput records, see L<WebPAC::Validation>
 
-=item --marc-normalize conf/normalize/mapping.pl
+You can use special variables C<$database> and $C<$input> in this parametar
+like C<--validate 'conf/validate/$database-$input'> to construct filename
 
-This option specifies normalisation file for MARC creation
+=item --validate-delimiters path/to/validate_delimiters_file
 
-=item --marc-output out/marc/test.marc
+this option is used with C<--validate> to turn on extra validation of
+delimiters. If file is non existant, it will be created on first run.
 
-Optional path to output file
+=item --marc-generate
+
+Generate MARC file. This will automatically be on if file contains C<marc*> directives.
+You can use this option as C<--no-marc-generate> to disable MARC generation.
 
 =item --marc-lint
 
-By default turned on if C<--marc-normalize> is used. You can disable lint
+By default turned on if normalisation file has C<marc*> directives. You can disable lint
 messages with C<--no-marc-lint>.
 
 =item --marc-dump
@@ -110,12 +116,14 @@ my $debug = 0;
 my $only_filter;
 my $stats = 0;
 my $validate_path;
-my ($marc_normalize, $marc_output);
+my $validate_delimiters_path;
+my $marc_generate = 1;
 my $marc_lint = 1;
 my $marc_dump = 0;
 my $parallel = 0;
 my $only_links = 0;
 my $merge = 0;
+my $help;
 
 my $log = _new WebPAC::Common()->_get_logger();
 
@@ -129,15 +137,20 @@ GetOptions(
        "debug+" => \$debug,
        "stats" => \$stats,
        "validate=s" => \$validate_path,
-       "marc-normalize=s" => \$marc_normalize,
-       "marc-output=s" => \$marc_output,
+       "validate-delimiters=s" => \$validate_delimiters_path,
+       "marc-generate!" => \$marc_generate,
        "marc-lint!" => \$marc_lint,
        "marc-dump!" => \$marc_dump,
        "parallel=i" => \$parallel,
        "only-links!" => \$only_links,
        "merge" => \$merge,
+       "help" => \$help,
 );
 
+$marc_generate = 0 if ( $validate_delimiters_path );
+
+pod2usage(-verbose => 2) if ($help);
+
 my $config = new WebPAC::Config( path => $config_path );
 
 #print "config = ",dump($config) if ($debug);
@@ -146,6 +159,12 @@ die "no databases in config file!\n" unless ($config->databases);
 
 $log->info( "-" x 79 );
 
+my $log_file = 'log';
+
+if (-e $log_file ) {   # && -s $log_file > 5 * 1024 * 1024) {
+       $log->info("moved old log with ", -s $log_file, " bytes to '${log_file}.old'");
+       rename $log_file, "${log_file}.old" || $log->logwarn("can't rename $log_file to ${log_file}.old: $!");
+}
 
 my $estcmd_fh;
 my $estcmd_path = './estcmd-merge.sh';
@@ -156,24 +175,20 @@ if ($merge) {
        $log->info("created merge batch file $estcmd_path");
 }
 
-
 my $validate;
 $validate = new WebPAC::Validate(
-       path => $validate_path,
-) if ($validate_path);
-
+       delimiters => $config->webpac('delimiters'),
+) if ($validate_path || $validate_delimiters_path);
 
 my $use_indexer = $config->use_indexer;
+$stats ||= $validate;
 if ($stats) {
-       $log->debug("option --stats disables update of indexing engine...");
+       $log->debug("disabled indexing for stats collection");
        $use_indexer = undef;
-} else {
+} elsif ( $use_indexer ) {
        $log->info("using $use_indexer indexing engine...");
 }
 
-# disable indexing when creating marc
-$use_indexer = undef if ($marc_normalize);
-
 # parse normalize files and create source files for lookup and normalization
 
 my $parser = new WebPAC::Parser( config => $config );
@@ -197,7 +212,8 @@ sub create_ds_config {
        return $c;
 }
 
-while (my ($database, $db_config) = each %{ $config->databases }) {
+foreach my $database ( sort keys %{ $config->databases } ) {
+       my $db_config = $config->databases->{$database};
 
        my ($only_database,$only_input) = split(m#/#, $only_filter) if ($only_filter);
        next if ($only_database && $database !~ m/$only_database/i);
@@ -212,7 +228,7 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
        }
 
        my $indexer;
-       if ($use_indexer) {
+       if ($use_indexer && $parser->have_rules( 'search', $database )) {
 
                my $cfg_name = $use_indexer;
                $cfg_name =~ s/\-.*$//;
@@ -228,27 +244,24 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
                if ($use_indexer eq 'hyperestraier') {
 
                        # open Hyper Estraier database
-                       use WebPAC::Output::Estraier '0.10';
+                       require WebPAC::Output::Estraier;
                        $indexer = new WebPAC::Output::Estraier( %{ $indexer_config } );
                
                } elsif ($use_indexer eq 'hyperestraier-native') {
 
                        # open Hyper Estraier database
-                       use WebPAC::Output::EstraierNative;
+                       require WebPAC::Output::EstraierNative;
                        $indexer = new WebPAC::Output::EstraierNative( %{ $indexer_config } );
 
                } elsif ($use_indexer eq 'kinosearch') {
 
-                       # open KinoSearch
-                       use WebPAC::Output::KinoSearch;
-                       $indexer_config->{clean} = 1 unless (-e $indexer_config->{index_path});
-                       $indexer = new WebPAC::Output::KinoSearch( %{ $indexer_config } );
+                       die "no longer supported";
 
                } else {
                        $log->logdie("unknown use_indexer: $use_indexer");
                }
 
-               $log->logide("can't continue without valid indexer") unless ($indexer);
+               $log->logdie("can't continue without valid indexer") unless ($indexer);
        }
 
 
@@ -282,37 +295,60 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
        #
        # now WebPAC::Store
        #
-       my $abs_path = abs_path($0);
-       $abs_path =~ s#/[^/]*$#/#;
+       my $store = new WebPAC::Store({
+               debug => $debug,
+       });
+
+       sub iterate_over {
+               my ( $what, $error ) = @_;
+               my @result;
+               if ( ref( $what ) eq 'ARRAY' ) {
+                       @result = @{ $what };
+               } elsif ($db_config->{input}) {
+                       @result =  ( $what );
+               } else {
+                       $error->() if ref($error) eq 'CODE';
+               }
+               return @result;
+       }
 
-       my $db_path = $config->get('webpac')->{db_path} . '/' . $database;
+       #
+       # prepare output
+       #
+       my @outputs = iterate_over( $db_config->{output}, sub {
+               $log->error("Database $database doesn't have any outputs defined. Do you want to remove it from configuration?" );
+       } );
 
-       if ($clean) {
-               $log->info("creating new database '$database' in $db_path");
-               rmtree( $db_path ) || $log->warn("can't remove $db_path: $!");
-       } else {
-               $log->info("working on database '$database' in $db_path");
-       }
+       my @output_modules;
 
-       my $store = new WebPAC::Store(
-               path => $db_path,
-               database => $database,
-               debug => $debug,
-       );
+       foreach my $output ( @outputs ) {
+
+warn '## output = ',dump( $output );
+
+               my $module = $output->{module} || $log->logdie("need module in output section of $database");
+               $module = 'WebPAC::Output::' . $module unless $module =~ m/::/;
+       
+               $log->debug("loading output module $module");
+               eval "require $module";
+       
+               $output->{database} = $database;
+
+               $log->debug("calling $module->new(",dump( $output ),")");
+               my $out = new $module->new( $output );
+               $out->init;
+
+               push @output_modules, $out;
+       }
 
 
        #
        # now, iterate through input formats
        #
 
-       my @inputs;
-       if (ref($db_config->{input}) eq 'ARRAY') {
-               @inputs = @{ $db_config->{input} };
-       } elsif ($db_config->{input}) {
-               push @inputs, $db_config->{input};
-       } else {
+
+       my @inputs = iterate_over( $db_config->{input}, sub {
                $log->info("database $database doesn't have inputs defined");
-       }
+       } );
 
        foreach my $input (@inputs) {
 
@@ -347,6 +383,7 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
                        stats => $stats,
                        modify_records => $input->{modify_records},
                        modify_file => $input->{modify_file},
+                       input_config => $input,
                );
                $log->logdie("can't create input using $input_module") unless ($input);
 
@@ -355,7 +392,6 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
                        delete( $input->{lookup} );
                }
 
-               my $lookup;
                my $lookup_coderef;
 
                if (@lookups) {
@@ -369,11 +405,10 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
                                WebPAC::Normalize::data_structure(
                                        row => $rec,
                                        rules => $rules,
-                                       lookup => $lookup,
                                        config => create_ds_config( $db_config, $database, $input, $mfn ),
                                );
 
-                               warn "current lookup = ", dump($lookup) if ($lookup);
+                               #warn "current lookup: ", dump(WebPAC::Normalize::_get_lookup());
                        };
 
                        WebPAC::Normalize::_set_lookup( undef );
@@ -382,20 +417,46 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
 
                };
 
+               my $lookup_jar;
+
                my $maxmfn = $input_db->open(
                        path => $input->{path},
                        code_page => $input->{encoding},        # database encoding
                        lookup_coderef => $lookup_coderef,
+                       lookup => $lookup_jar,
                        %{ $input },
+                       load_row => sub {
+                               my $a = shift;
+                               return $store->load_row(
+                                       database => $database,
+                                       input => $input_name,
+                                       id => $a->{id},
+                               );
+                       },
+                       save_row => sub {
+                               my $a = shift;
+                               return $store->save_row(
+                                       database => $database,
+                                       input => $input_name,
+                                       id => $a->{id},
+                                       row => $a->{row},
+                               );
+                       },
+
                );
 
                my $lookup_data = WebPAC::Normalize::_get_lookup();
 
                if (defined( $lookup_data->{$database}->{$input_name} )) {
-                       $log->debug("created following lookups: ", dump( $lookup_data ));
+                       $log->debug("created following lookups: ", sub { dump( $lookup_data ) } );
 
                        foreach my $key (keys %{ $lookup_data->{$database}->{$input_name} }) {
-                               $store->save_lookup( $database, $input_name, $key, $lookup_data->{$database}->{$input_name}->{$key} );
+                               $store->save_lookup(
+                                       database => $database,
+                                       input => $input_name,
+                                       key => $key,
+                                       data => $lookup_data->{$database}->{$input_name}->{$key},
+                               );
                        }
                }
 
@@ -407,126 +468,167 @@ while (my ($database, $db_config) = each %{ $config->databases }) {
                        print $report_fh "Report for database '$database' input '$input_name' records ",
                                $offset || 1, "-", $limit || $input->{limit} || $maxmfn, "\n\n";
                        $log->info("Generating report file $path");
-               }
-
-               my @norm_array = ref($input->{normalize}) eq 'ARRAY' ?
-                       @{ $input->{normalize} } : ( $input->{normalize} );
 
-               if ($marc_normalize) {
-                       @norm_array = ( {
-                               path => $marc_normalize,
-                               output => $marc_output || "out/marc/${database}-${input_name}.marc",
-                       } );
+                       if ( $validate ) {
+                               $validate->read_validate_file( $validate->fill_in( $validate_path, database => $database, input => $input_name ) ) if ( $validate_path );
+                               $validate->read_validate_delimiters_file( $validate->fill_in( $validate_delimiters_path, database => $database, input => $input_name ) ) if ( $validate_delimiters_path );
+                       }
                }
 
-               foreach my $normalize (@norm_array) {
-
-                       my $normalize_path = $normalize->{path} || $log->logdie("can't find normalize path in config");
-
-                       $log->logdie("Found '$normalize_path' as normalization file which isn't supported any more!") unless ( $normalize_path =~ m!\.pl$!i );
-
-                       my $rules = read_file( $normalize_path ) or die "can't open $normalize_path: $!";
-
-                       $log->info("Using $normalize_path for normalization...");
-
-                       my $marc = new WebPAC::Output::MARC(
-                               path => $normalize->{output},
+               my $marc;
+               if ($marc_generate && $parser->have_rules( 'marc', $database, $input_name )) {
+                       $marc = new WebPAC::Output::MARC(
+                               path => "out/marc/${database}-${input_name}.marc",
                                lint => $marc_lint,
                                dump => $marc_dump,
-                       ) if ($normalize->{output});
+                       );
+               }
 
-                       # reset position in database
-                       $input_db->seek(1);
+               my $rules = $parser->normalize_rules($database,$input_name) || $log->logdie("no normalize rules found for $database/$input_name");
+               $log->debug("parsed normalize rules:\n$rules");
+
+               # reset position in database
+               $input_db->seek(1);
+
+               # generate name of config key for indexer (strip everything after -)
+               my $indexer_config = $use_indexer;
+               $indexer_config =~ s/^(\w+)-?.*$/$1/g if ($indexer_config);
+
+               my $lookup_hash;
+               my $depends = $parser->depends($database,$input_name);
+       
+               if ($depends) {
+                       $log->debug("$database/$input_name depends on: ", dump($depends)) if ($depends);
+                       $log->logdie("parser->depends didn't return HASH") unless (ref($depends) eq 'HASH');
+
+                       foreach my $db (keys %$depends) {
+                               foreach my $i (keys %{$depends->{$db}}) {
+                                       foreach my $k (keys %{$depends->{$db}->{$i}}) {
+                                               my $t = time();
+                                               $log->debug("loading lookup $db/$i");
+                                               $lookup_hash->{$db}->{$i}->{$k} = $store->load_lookup(
+                                                       database => $db,
+                                                       input => $i,
+                                                       key => $k,
+                                               );
+                                               $log->debug(sprintf("lookup $db/$i took %.2fs", time() - $t));
+                                       }
+                               }
+                       }
 
-                       # generate name of config key for indexer (strip everything after -)
-                       my $indexer_config = $use_indexer;
-                       $indexer_config =~ s/^(\w+)-?.*$/$1/g if ($indexer_config);
+                       $log->debug("lookup_hash = ", sub { dump( $lookup_hash ) });
+               }
 
-                       foreach my $pos ( 0 ... $input_db->size ) {
 
-                               my $row = $input_db->fetch || next;
+               foreach my $pos ( 0 ... $input_db->size ) {
 
-                               my $mfn = $row->{'000'}->[0];
+                       my $row = $input_db->fetch || next;
 
-                               if (! $mfn || $mfn !~ m#^\d+$#) {
-                                       $log->warn("record $pos doesn't have valid MFN but '$mfn', using $pos");
-                                       $mfn = $pos;
-                                       push @{ $row->{'000'} }, $pos;
-                               }
+                       $total_rows++;
 
+                       my $mfn = $row->{'000'}->[0];
 
-                               if ($validate) {
-                                       if ( my $errors = $validate->validate_errors( $row, $input_db->dump ) ) {
-                                               $log->error( "MFN $mfn validation error:\n",
-                                                       $validate->report_error( $errors )
-                                               );
-                                       }
-                               }
+                       if (! $mfn || $mfn !~ m{^\d+$}) {
+                               $log->warn("record $pos doesn't have valid MFN but '$mfn', using $pos");
+                               $mfn = $pos;
+                               push @{ $row->{'000'} }, $pos;
+                       }
 
-                               my $ds = WebPAC::Normalize::data_structure(
-                                       row => $row,
-                                       rules => $rules,
-                                       lookup => $lookup ? $lookup->lookup_hash : undef,
-                                       config => create_ds_config( $db_config, $database, $input, $mfn ),
-                                       marc_encoding => 'utf-8',
-                               );
 
-                               $store->save_ds(
-                                       id => $mfn,
-                                       ds => $ds,
-                                       prefix => $input_name,
-                               ) if ($ds && !$stats);
-
-                               $indexer->add(
-                                       id => "${input_name}/${mfn}",
-                                       ds => $ds,
-                                       type => $config->get($indexer_config)->{type},
-                               ) if ($indexer && $ds);
-
-                               if ($marc) {
-                                       my $i = 0;
-
-                                       while (my $fields = WebPAC::Normalize::_get_marc_fields( fetch_next => 1 ) ) {
-                                               $marc->add(
-                                                       id => $mfn . ( $i ? "/$i" : '' ),
-                                                       fields => $fields,
-                                                       leader => WebPAC::Normalize::marc_leader(),
-                                                       row => $row,
-                                               );
-                                               $i++;
-                                       }
+                       if ($validate) {
+                               if ( my $errors = $validate->validate_rec( $row, $input_db->dump_ascii ) ) {
+                                       $log->error( "MFN $mfn validation error:\n",
+                                               $validate->report_error( $errors )
+                                       );
+                               }
+                               next;   # validation doesn't create any output
+                       }
 
-                                       $log->info("Created $i instances of MFN $mfn\n") if ($i > 1);
+                       my $ds = WebPAC::Normalize::data_structure(
+                               row => $row,
+                               rules => $rules,
+                               lookup => $lookup_hash,
+                               config => create_ds_config( $db_config, $database, $input, $mfn ),
+                               marc_encoding => 'utf-8',
+                               load_row_coderef => sub {
+                                       my ($database,$input,$mfn) = @_;
+                                       return $store->load_row(
+                                               database => $database,
+                                               input => $input,
+                                               id => $mfn,
+                                       );
+                               },
+                       );
+
+                       $log->debug("ds = ", sub { dump($ds) }) if ($ds);
+
+                       $store->save_ds(
+                               database => $database,
+                               input => $input_name,
+                               id => $mfn,
+                               ds => $ds,
+                       ) if ($ds && !$stats);
+
+                       $indexer->add(
+                               id => "${input_name}/${mfn}",
+                               ds => $ds,
+                               type => $config->get($indexer_config)->{type},
+                       ) if ($indexer && $ds);
+
+                       if ($marc) {
+                               my $i = 0;
+
+                               while (my $fields = WebPAC::Normalize::_get_marc_fields( fetch_next => 1 ) ) {
+                                       $marc->add(
+                                               id => $mfn . ( $i ? "/$i" : '' ),
+                                               fields => $fields,
+                                               leader => WebPAC::Normalize::_get_marc_leader(),
+                                               row => $row,
+                                       );
+                                       $i++;
                                }
 
-                               $total_rows++;
+                               $log->info("Created $i instances of MFN $mfn\n") if ($i > 1);
                        }
 
-                       if ($validate) {
-                               my $errors = $validate->report;
-                               if ($errors) {
-                                       $log->info("validation errors:\n$errors\n" );
-                                       print $report_fh "$errors\n" if ($report_fh);
-                               }
+                       foreach my $out ( @output_modules ) {
+                               $out->add( $mfn, $ds ) if $out->can('add');
                        }
 
-                       if ($stats) {
-                               my $s = $input_db->stats;
-                               $log->info("statistics of fields usage:\n$s");
-                               print $report_fh "Statistics of fields usage:\n$s" if ($report_fh);
+               }
+
+               if ($validate) {
+                       my $errors = $validate->report;
+                       if ($errors) {
+                               $log->info("validation errors:\n$errors\n" );
+                               print $report_fh "$errors\n" if ($report_fh);
                        }
 
-                       # close MARC file
-                       $marc->finish if ($marc);
+                       print $report_fh "\nAll possible subfields/delimiter templates:\n", $validate->delimiters_templates( report => 1, current_input => 1 ), "\n\n";
+
+                       # must be last thing that touches $validate for this input
+                       $validate->reset;
+               }
 
-                       # close report
-                       close($report_fh) if ($report_fh)
+               if ($stats) {
+                       my $s = $input_db->stats;
+                       $log->info("statistics of fields usage:\n$s");
+                       print $report_fh "Statistics of fields usage:\n$s" if ($report_fh);
                }
 
+               # close MARC file
+               $marc->finish if ($marc);
+
+               # close report
+               close($report_fh) if ($report_fh);
        }
 
        eval { $indexer->finish } if ($indexer && $indexer->can('finish'));
 
+       foreach my $out ( @output_modules ) {
+               $out->finish if $out->can('finish');
+       }
+
        my $dt = time() - $start_t;
        $log->info("$total_rows records ", $indexer ? "indexed " : "",
                sprintf("in %.2f sec [%.2f rec/sec]",
@@ -549,6 +651,9 @@ if ($parallel) {
        $log->info("all parallel processes finished");
 }
 
+# save new delimiters if needed
+$validate->save_delimiters_templates if ( $validate_delimiters_path );
+
 #
 # handle links or merge after indexing
 #