#!/usr/bin/perl
+
+# Copyright 2009-2012, Equinox Software, Inc.
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+
use strict;
use warnings;
use Getopt::Long;
-use Equinox::Migration::MapDrivenMARCXMLProc;
+use Equinox::Migration::MapDrivenMARCXMLProc 1.005;
use Equinox::Migration::MARCXMLSampler;
my $VERSION = '1.001';
# run samples if we've been asked for them
run_samples($c) if ($c->{sample} or $c->{samplemap} or $c->{samplestr});
extract_holdings($c) if ($c->{map});
+print "\n";
#--------------------------
sub extract_holdings {
my ($c) = @_;
- print "Parsing records for extraction... ";
+ print "Parsing records for extraction:\n";
my $m = Equinox::Migration::MapDrivenMARCXMLProc->new( marcfile => $c->{marcfile},
mapfile => $c->{map},
verbose => 1,
);
+
print "Writing holdings to output file(s)...\n";
# open main holdings file
open HOLDINGS, '>', ($c->{prefix} . "-HOLDINGS.pg");
- # open the files for multi mappings
- # FIXME make this actually key off multi fields
- open X, '>', ($c->{prefix} . "-HOLDINGS-privnotes.pg");
- open Z, '>', ($c->{prefix} . "-HOLDINGS-pubnotes.pg");
- select HOLDINGS;
-
- my $i = 0;
- for my $rec ( @{$m->{data}{recs}} ) {
- # for each holdings tag in the record...
- for my $holdidx ( @{$rec->{tmap}{ $c->{holdings} }} ) {
+ # create multi files
+ my $multis = $m->get_multis;
+ my %MULTIFILE = ();
+ for my $t ( keys %{$multis} ) {
+ for my $s ( keys %{$multis->{$t}})
+ { open my $fh, ">", ($c->{prefix} . "-HOLDINGS-MULT-$t$s.pg"); $MULTIFILE{"$t$s"} = $fh }
+ }
+
+ my $parallel_fields = $m->get_parallel_fields;
+
+ my $i = 0; # record counter
+ my $j = 0; # holdings counter
+
+ while ( $m->{data}{recs}[$i] ) {
+ print HOLDINGS "BEGIN;\n\negid, hseq, " unless $j;
+ my $rec = $m->{data}{recs}[$i];
+ my $k = 0; # holding-within-record pointer
+
+ for my $holdidx ( @{ $rec->{tmap}{ $c->{holdings} } } ) {
+ # for each holdings tag in the record...
my $tagid = $rec->{tags}[$holdidx]{tag};
+ $k++;
- print STDOUT "\r$i";
my @out = (); # clear the output buffer
push @out, $rec->{egid}; # slug in the egid first thing
- print "BEGIN;\n\n" unless $i;
+ push @out, $j; # holding seq goes next
# grab the unary mappings and slug 'em in
for my $sub ( sort keys %{$rec->{tags}[$holdidx]{uni}} ) {
push @out, $rec->{tags}[$holdidx]{uni}{$sub};
- #print "l_", $m->name($tagid, $sub),"\t"
- # unless $i;
- # FIXME column names should be made workable again
+ print HOLDINGS "l_", $m->name($tagid, $sub),", " unless $j;
}
- for my $x (@{$rec->{tags}[$holdidx]{multi}{x}} ) {
- print X $rec->{egid}, "\t",
- $rec->{tags}[$holdidx]{uni}{ $c->{copyid} },
- "\t$x\n";
- }
- for my $z (@{$rec->{tags}[$holdidx]{multi}{z}} ) {
- print Z $rec->{egid}, "\t",
- $rec->{tags}[$holdidx]{uni}{ $c->{copyid} },
- "\t$z\n";
+ # handle holdings multis
+ for my $sub ( sort keys %{$multis->{$tagid}} ) {
+ for my $value ( @{$rec->{tags}[$holdidx]{multi}{$sub}} ) {
+ my $fh = $MULTIFILE{"$tagid$sub"};
+ my $clean_value = $value;
+ $clean_value =~ s/[\r\n\t]//g;
+ print $fh join("\t", $rec->{egid}, $j, $clean_value), "\n";
+ }
}
# now get everything else in the mapping
for my $othertag ( sort keys %{$rec->{tmap}} ) {
next if $othertag eq $c->{holdings}; # ignoring the holdings, o'course
- my $idx = $rec->{tmap}{$othertag}[0]; # get index into tags struct
- for my $sub ( sort keys %{$rec->{tags}[$idx]{uni}} ) {
- push @out, $rec->{tags}[$idx]{uni}{$sub};
- print "l_", $m->name($rec->{tags}[$idx]{tag}, $sub), "\t"
- unless $i;
+ my $test_idx = $rec->{tmap}{$othertag}[0]; # get index into tags struct
+ unless (defined $test_idx) {
+ push @out, '';
+ next;
+ }
+
+ # handle parallel fields
+ if (exists($parallel_fields->{$othertag})) {
+ my $num_fields = $#{ $rec->{tmap}{$othertag} };
+ my $tag_idx;
+ if ($holdidx > $num_fields) {
+ $tag_idx = -1;
+ } else {
+ $tag_idx = $rec->{tmap}{$othertag}[$holdidx];
+ }
+ for my $sub ( sort keys %{ $parallel_fields->{$othertag } } ) {
+ push @out, $tag_idx > -1 ? $rec->{tags}[$tag_idx]{parallel}{$sub}->[0] : '';
+ print HOLDINGS "l_", $m->name($rec->{tags}[$tag_idx]{tag}, $sub), ", " unless $j;
+ }
+ }
+
+ # handle only first other tag unless it is known to be multi
+ my $limit = 0;
+ if (exists($multis->{$othertag})) {
+ $limit = $#{ $rec->{tmap}{$othertag} };
+ }
+ foreach my $idx (0..$limit) {
+ my $tag_idx = $rec->{tmap}{$othertag}[$idx];
+ for my $sub ( sort keys %{$rec->{tags}[$tag_idx]{uni}} ) {
+ if ($m->first_only($rec->{tags}[$tag_idx]{tag}, $sub)) {
+ push @out, ($k == 1) ? $rec->{tags}[$tag_idx]{uni}{$sub} : '';
+ } else {
+ push @out, $rec->{tags}[$tag_idx]{uni}{$sub};
+ }
+ print HOLDINGS "l_", $m->name($rec->{tags}[$tag_idx]{tag}, $sub), ", " unless $j;
+ }
+ next unless exists($multis->{$othertag});
+ for my $sub ( sort keys %{$multis->{$othertag}} ) {
+ next if $m->first_only($rec->{tags}[$tag_idx]{tag}, $sub) and ($k > 1);
+ for my $value ( @{$rec->{tags}[$tag_idx]{multi}{$sub}} ) {
+ my $fh = $MULTIFILE{"$othertag$sub"};
+ my $clean_value = $value;
+ $clean_value =~ s/[\r\n\t]//g;
+ print $fh normalize_output(join("\t", $rec->{egid}, $j, $clean_value)), "\n";
+ }
+ }
}
}
# and dump it
- print "\n" if ($i == 1);
- print join("\t", @out);
- print "\n";
+ print HOLDINGS "\n" unless $j;
+ print HOLDINGS normalize_output(join("\t", @out));
+ print HOLDINGS "\n";
+ $j++;
}
$i++;
+ print "\r$i $j";
}
- select STDOUT;
print "\n";
}
sub run_samples {
my ($c) = @_;
my $s;
+ print "Parsing records for sampling... ";
if ($c->{samplemap}) {
$s = Equinox::Migration::MARCXMLSampler->new( marcfile => $c->{marcfile},
mapfile => $c->{samplemap});
} else {
$s = Equinox::Migration::MARCXMLSampler->new( marcfile => $c->{marcfile} );
}
- print "Parsing records for sampling... ";
- $s->parse_records;
dump_sample_overview($c, $s) if $c->{sample};
dump_sample_detail($c, $s) if ($c->{samplemap} or $c->{samplestr});
for my $subkey (sort keys %{ $tags->{$tag} }) {
my $sub = $tags->{$tag}{$subkey};
print "|| $subkey | ", $sub->{value}, " | ",
- $sub->{count}, "/", $sub->{tcnt}, " | ||\n";
+ $sub->{count}, "/", $sub->{tcnt}, " | ", ($sub->{count} > $sub->{tcnt}) ? "MULTI" : "", " ||\n";
}
print "\n";
}
print "\n";
}
select STDOUT;
+ print "Saved results as ", ($c->{prefix} . "-HOLDINGS-DETAIL.txt"), " and ",
+ ($c->{prefix} . "-HOLDINGS-SUBCOUNTS.txt"), "\n";
}
sub dump_sample_overview {
print " (", sprintf("%03d", int($tags->{ $tagsbycount[$i] } / $count * 100)), "%)\n";
}
select STDOUT;
- print "\n";
+ print "Saved results as ", ($c->{prefix} . "-HOLDINGS-OVERVIEW.txt"), "\n";
close SAMPLE;
}
'sample|s',
'samplemap|sm=s',
'samplestr|ss=s',
- 'marcfile|m=s',
- 'map=s',
+ 'marcfile=s',
+ 'map|m=s',
'holdings|h=i',
'copyid|c=s',
'prefix|p=s',
+ 'disable-pg-normalization',
'version|v',
'help',
);
if ($c->{map} and !($c->{holdings} and $c->{copyid}));
show_version() if $c->{version};
+ if ($c->{prefix} and !$c->{marcfile}) {
+ $c->{marcfile} = $c->{prefix} . ".clean.marc.xml";
+ }
+
my @keys = keys %{$c};
for my $key ('prefix', 'marcfile')
{ push @missing, $key unless $c->{$key} }
return $c;
}
+sub normalize_output {
+ my $str = shift;
+ $str =~ s!\\!\\\\!g unless $c->{'disable-pg-normalization'};
+ return $str;
+}
+
sub show_help {
my ($msg) = @_;
print "\nERROR - $msg\n" if $msg;
print <<HELP;
-Usage is: extract_holdings -p PREFIX -m MARCFILE [ARGUMENTS]
+Usage is: extract_holdings --prefix PREFIX --marcfile MARCFILE [ARGUMENTS]
REQUIRED ARGUMENTS
--prefix -p Prefix string for output filenames
- --marcfile -m MARCXML to use as source data
+ --marcfile MARCXML to use as source data
+ Defaults to 'PREFIX.clean.marc.xml'
SAMPLING ARGUMENTS
--sample -s Generate a report of all tags in the MARC data
- --samplemap -sm Specify a E::M::STL map file which will be used to generate
- subfield breakdown reports about specific tags in the MARC
- data
+ --samplemap -sm Specify a Equinox::Migration::SimpleTagList map file which
+ will be used to generate subfield breakdown reports about
+ specific tags in the MARC data
--samplestr -ss As above, but with a one-liner map specified on the command
line as a string (e.g. '-ss "852 999"')
If --samplemap and --samplestr are both specified, --samplemap wins.
HOLDINGS EXTRACTION ARGUMENTS
- --map E::M::SM map file which will be used to extract holdings data
- from the input MARC file
+ --map -m Equinox::Migration::SubfieldMapper map file which will be
+ used to extract holdings data from the input MARC file
--holdings -h Specifies actual holdings tag
--copyid -c Specifies subfield of holdings with unique copy identifier
+ --disable-pg-normalization By default, output is normalized so that a Postgres
+ copy or \\copy can import the data without choking on
+ backslashes; use this command-line option if
+ output is not meant to be consumed by psql.
- Both these must be given together.
+ All three of these must be given together.
HELP
exit;
}
projects / migration-tools.git / blobdiff