/
import.pm
87 lines (66 loc) · 1.91 KB
/
import.pm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package Catmandu::Cmd::import;
use Catmandu::Sane;
our $VERSION = '1.10_03';
use parent 'Catmandu::Cmd';
use Catmandu;
use namespace::clean;
sub command_opt_spec {
(
["verbose|v", ""],
["fix=s@", ""],
["var=s%", ""],
["preprocess|pp", ""],
["start=i", ""],
["total=i", ""],
["delete", "delete existing items first"],
["transaction|tx", "wrap in a transaction"],
);
}
sub command {
my ($self, $opts, $args) = @_;
my ($from_args, $from_opts, $into_args, $into_opts)
= $self->_parse_options($args);
my $from = Catmandu->importer($from_args->[0], $from_opts);
my $into_bag = delete $into_opts->{bag};
my $into = Catmandu->store($into_args->[0], $into_opts)->bag($into_bag);
if ($opts->start // $opts->total) {
$from = $from->slice($opts->start, $opts->total);
}
if ($opts->fix) {
$from = $self->_build_fixer($opts)->fix($from);
}
if ($opts->verbose) {
$from = $from->benchmark;
}
my $tx = sub {
if ($opts->delete) {
$into->delete_all;
$into->commit;
}
my $n = $into->add_many($from);
$into->commit;
if ($opts->verbose) {
say STDERR $n == 1 ? "imported 1 item" : "imported $n items";
say STDERR "done";
}
};
if ($opts->transaction) {
$self->usage_error("Bag isn't transactional")
if !$into->store->does('Catmandu::Transactional');
$into->store->transaction($tx);
}
else {
$tx->();
}
}
1;
__END__
=pod
=head1 NAME
Catmandu::Cmd::import - import items into a store
=head1 EXAMPLES
catmandu import <IMPORTER> <OPTIONS> to <STORE> <OPTIONS>
catmandu import YAML to MongoDB --database-name items --bag book < books.yml
catmandu help importer YAML
catmandu help importer MongoDB
=cut