--- trunk/run.pl 2006/09/29 20:25:34 732 +++ trunk/run.pl 2009/05/18 18:51:40 1184 @@ -7,12 +7,12 @@ use lib './lib'; use WebPAC::Common 0.02; -use WebPAC::Parser 0.04; -use WebPAC::Input 0.13; -use WebPAC::Store 0.11; -use WebPAC::Normalize 0.21; -use WebPAC::Output::TT; -use WebPAC::Validate 0.06; +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.11; use WebPAC::Output::MARC; use WebPAC::Config; use Getopt::Long; @@ -21,8 +21,9 @@ use File::Slurp; use Data::Dump qw/dump/; use Storable qw/dclone/; +use Pod::Usage qw/pod2usage/; +use LWP::Simple; -use Proc::Queue size => 1; use POSIX ":sys_wait_h"; # imports WNOHANG =head1 NAME @@ -31,7 +32,7 @@ B -Options: +=head1 OPTIONS =over 4 @@ -67,17 +68,22 @@ turn on extra validation of imput records, see L -=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 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 directives. You can disable lint messages with C<--no-marc-lint>. =item --marc-dump @@ -97,6 +103,10 @@ Create merged index of databases which have links +=item --mirror http://www.example.com + +Tries to download input path files from mirror URI + =back =cut @@ -110,12 +120,15 @@ 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 $mirror; +my $help; my $log = _new WebPAC::Common()->_get_logger(); @@ -125,27 +138,41 @@ "clean" => \$clean, "one=s" => \$only_filter, "only=s" => \$only_filter, - "config" => \$config_path, + "config=s" => \$config_path, "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, + "mirror=s" => \$mirror, + "help" => \$help, ); +$marc_generate = 0 if ( $validate_delimiters_path ); + +pod2usage(-verbose => 2) if ($help); + my $config = new WebPAC::Config( path => $config_path ); +WebPAC::Normalize::_debug( $debug - 1 ) if $debug > 1; + #print "config = ",dump($config) if ($debug); 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,27 +183,29 @@ $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 ); +my ($only_database,$only_input) = split(m#/#, $only_filter) if $only_filter; + +my $parser = new WebPAC::Parser( + config => $config, + only_database => $only_database, + only_input => $only_input, +); my $total_rows = 0; my $start_t = time(); @@ -184,6 +213,8 @@ my @links; if ($parallel) { + eval 'use Proc::Queue size => 1;'; + die $@ if $@; $log->info("Using $parallel processes for speedup"); Proc::Queue::size($parallel); } @@ -197,9 +228,9 @@ 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); if ($parallel) { @@ -212,7 +243,7 @@ } my $indexer; - if ($use_indexer) { + if ($use_indexer && $parser->have_rules( 'search', $database )) { my $cfg_name = $use_indexer; $cfg_name =~ s/\-.*$//; @@ -228,27 +259,24 @@ 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,52 +310,97 @@ # # now WebPAC::Store # - my $abs_path = abs_path($0); - $abs_path =~ s#/[^/]*$#/#; + my $store = new WebPAC::Store({ + debug => $debug, + }); - my $db_path = $config->webpac('db_path'); - 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"); - } + # + # prepare output + # + my @outputs = force_array( $db_config->{output}, sub { + $log->error("Database $database doesn't have any outputs defined. Do you want to remove it from configuration?" ); + } ); - my $store = new WebPAC::Store( - path => $db_path, - debug => $debug, - ); + my @output_modules; + + 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"; + + # add database to arugemnts for output filter + $output->{database} = $database; + $output->{clean} = $clean; + + $log->debug("calling $module->new(",dump( $output ),")"); + my $out = new $module->new( $output ); + if ( $out->init ) { + push @output_modules, $out; + } else { + $log->warn("SKIPPED $module"); + } + } # # 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 = force_array( $db_config->{input}, sub { $log->info("database $database doesn't have inputs defined"); - } + } ); + + if ( -e 'out/debug' ) { # FIXME flag? + my $out; + foreach my $i ( @inputs ) { + warn dump( $i ); + next unless defined $i->{normalize}; + warn dump( $i->{normalize} ); + foreach my $normalize ( @{ $i->{normalize} } ) { + my $path = $normalize->{path}; + $out .= qq/\n##\n## $path\n##\n\n/; + $out .= read_file( $path ); + } + } + my $all = "out/debug/all-normalize.pl"; + write_file( $all, $out ); + warn "### all normalize for this input saved to: $all"; + }; foreach my $input (@inputs) { my $input_name = $input->{name} || $log->logdie("input without a name isn't valid: ",dump($input)); + if ( $input->{skip} ) { + $log->info("skip $input_name"); + next; + } + next if ($only_input && ($input_name !~ m#$only_input#i && $input->{type} !~ m#$only_input#i)); my $type = lc($input->{type}); - die "I know only how to handle input types ", join(",", $config->webpac('inputs') ), " not '$type'!\n" unless (grep(/$type/, $config->webpac('inputs'))); + # FIXME check if input module exists + my $input_module = $input->{module}; - my $input_module = $config->webpac('inputs')->{$type}; + if ( ! $input_module ) { + if ( grep(/$type/, $config->webpac('inputs')) ) { + $input_module = $config->webpac('inputs')->{$type}; + } else { + $log->logdie("I know only how to handle input types ", join(",", $config->webpac('inputs') ), " not '$type'!" ); + } + } my @lookups = $parser->have_lookup_create($database, $input); - $log->info("working on input '$input_name' in $input->{path} [type: $input->{type}] using $input_module", + $log->info("working on $database/$input_name with $input_module on $input->{path}", @lookups ? " creating lookups: ".join(", ", @lookups) : "" ); @@ -337,15 +410,31 @@ delete($input->{modify_file}); } + if ( $mirror ) { + my $path = $input->{path} || die "no input path in ",dump( $input ); + + my $base = $path; + $base =~ s{/[^/]+$}{}; + mkpath $base unless -e $base; + + my $rc = mirror( "$mirror/$path", $path ); + if (is_error( $rc )) { + die "can't mirror $mirror/$path -> $path [$rc]"; + } else { + $log->info( "mirror ", $path, " [$rc] ", -s $path, " bytes" ); + } + + } + my $input_db = new WebPAC::Input( module => $input_module, - encoding => $config->webpac('webpac_encoding'), limit => $limit || $input->{limit}, offset => $offset, recode => $input->{recode}, 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); @@ -383,16 +472,34 @@ my $maxmfn = $input_db->open( path => $input->{path}, - code_page => $input->{encoding}, # database encoding + input_encoding => $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( @@ -406,24 +513,37 @@ my $report_fh; if ($stats || $validate) { - my $path = "out/report/${database}-${input_name}.txt"; + my $out_report = 'out/report'; # FIXME move to config + mkpath $out_report unless -e $out_report; + my $path = "$out_report/${database}-${input_name}.txt"; open($report_fh, '>', $path) || $log->logdie("can't open $path: $!"); 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"); + + 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 ); + } } my $marc; - if ($marc_normalize) { + if ($marc_generate && $parser->have_rules( 'marc', $database, $input_name )) { + + my $out_marc = 'out/marc'; # FIXME move to config + mkpath $out_marc unless -e $out_marc; + $marc = new WebPAC::Output::MARC( - path => $marc_output || "out/marc/${database}-${input_name}.marc", + path => "$out_marc/${database}-${input_name}.marc", lint => $marc_lint, dump => $marc_dump, ); } - my $rules = $parser->normalize_rules($database,$input_name) || $log->logdie("no normalize rules found for $database/$input_name"); + my $rules = $parser->normalize_rules($database,$input_name); + $log->logwarn("no normalize rules for $database/$input_name") unless $rules; + $log->debug("parsed normalize rules:\n$rules"); # reset position in database @@ -443,17 +563,29 @@ 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)); } } } - $log->debug("lookup_hash = ", dump( $lookup_hash )); + $log->debug("lookup_hash = ", sub { dump( $lookup_hash ) }); + } + + + # setup input name for all output filters + foreach my $out ( @output_modules ) { + if ( $out->can('input') ) { + $out->input( $input_name ); + } else { + $log->warn("output filter ",ref($out)," doesn't support input name"); + } } @@ -461,9 +593,11 @@ my $row = $input_db->fetch || next; + $total_rows++; + my $mfn = $row->{'000'}->[0]; - if (! $mfn || $mfn !~ m#^\d+$#) { + if (! $mfn || $mfn !~ m{^\d+$}) { $log->warn("record $pos doesn't have valid MFN but '$mfn', using $pos"); $mfn = $pos; push @{ $row->{'000'} }, $pos; @@ -471,50 +605,67 @@ if ($validate) { - if ( my $errors = $validate->validate_errors( $row, $input_db->dump ) ) { + 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 } - 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_ds_coderef => sub { - my ($database,$input,$mfn) = @_; - return $store->load_ds( + if ($rules) { + + 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) = @_; +#warn "### load_row($database,$input,$mfn) from data_structure\n"; + 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, + input => $input_name, id => $mfn, - ); - }, - ); + ds => $ds, + ) if !$stats; - $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); + $indexer->add( + id => "${input_name}/${mfn}", + ds => $ds, + type => $config->get($indexer_config)->{type}, + ) if $indexer; + + foreach my $out ( @output_modules ) { + $out->add( $mfn, $ds ) if $out->can('add'); + } + + } else { + $log->warn("record $pos didn't produce any output after normalization rules!") unless $marc; + } + } if ($marc) { my $i = 0; - while (my $fields = WebPAC::Normalize::_get_marc_fields( fetch_next => 1 ) ) { + while (my $fields = WebPAC::Normalize::MARC::_get_marc_fields( fetch_next => 1 ) ) { $marc->add( id => $mfn . ( $i ? "/$i" : '' ), fields => $fields, - leader => WebPAC::Normalize::marc_leader(), + leader => WebPAC::Normalize::MARC::_get_marc_leader(), row => $row, ); $i++; @@ -523,7 +674,6 @@ $log->info("Created $i instances of MFN $mfn\n") if ($i > 1); } - $total_rows++; } if ($validate) { @@ -532,6 +682,11 @@ $log->info("validation errors:\n$errors\n" ); print $report_fh "$errors\n" if ($report_fh); } + + 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; } if ($stats) { @@ -544,11 +699,14 @@ $marc->finish if ($marc); # close report - close($report_fh) if ($report_fh) - + close($report_fh) if ($report_fh); } - eval { $indexer->finish } if ($indexer && $indexer->can('finish')); + $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 " : "", @@ -572,6 +730,9 @@ $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 #