summaryrefslogtreecommitdiffstats
path: root/scripts/migrate-attachments.pl
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/migrate-attachments.pl')
-rwxr-xr-xscripts/migrate-attachments.pl165
1 files changed, 88 insertions, 77 deletions
diff --git a/scripts/migrate-attachments.pl b/scripts/migrate-attachments.pl
index dee1e7a3b..2de7c602c 100755
--- a/scripts/migrate-attachments.pl
+++ b/scripts/migrate-attachments.pl
@@ -13,8 +13,6 @@ use lib qw(. lib local/lib/perl5);
$| = 1;
-
-
use Bugzilla;
use Bugzilla::Attachment;
use Bugzilla::Install::Util qw(indicate_progress);
@@ -25,7 +23,7 @@ my @storage_names = Bugzilla::Attachment->get_storage_names();
my %options;
GetOptions(\%options, 'mirror=s@{2}', 'copy=s@{2}', 'delete=s') or exit(1);
unless ($options{mirror} || $options{copy} || $options{delete}) {
- die <<EOF;
+ die <<EOF;
Syntax:
migrate-attachments.pl --mirror source destination
migrate-attachments.pl --copy source destination
@@ -62,94 +60,107 @@ EOF
my $dbh = Bugzilla->dbh;
if ($options{mirror}) {
- if ($options{mirror}->[0] eq $options{mirror}->[1]) {
- die "Source and destination must be different\n";
+ if ($options{mirror}->[0] eq $options{mirror}->[1]) {
+ die "Source and destination must be different\n";
+ }
+ my ($source, $dest) = map { storage($_) } @{$options{mirror}};
+
+ my ($total) = $dbh->selectrow_array("SELECT COUNT(*) FROM attachments");
+ confirm(sprintf(
+ 'Mirror %s attachments from %s to %s?', $total, @{$options{mirror}}));
+
+ my $sth = $dbh->prepare(
+ "SELECT attach_id, attach_size FROM attachments ORDER BY attach_id DESC");
+ $sth->execute();
+ my ($count, $deleted, $stored) = (0, 0, 0);
+ while (my ($attach_id, $attach_size) = $sth->fetchrow_array()) {
+ indicate_progress({total => $total, current => ++$count});
+
+ # remove deleted attachments
+ if ($attach_size == 0 && $dest->exists($attach_id)) {
+ $dest->remove($attach_id);
+ $deleted++;
}
- my ($source, $dest) = map { storage($_) } @{ $options{mirror} };
-
- my ($total) = $dbh->selectrow_array("SELECT COUNT(*) FROM attachments");
- confirm(sprintf('Mirror %s attachments from %s to %s?', $total, @{ $options{mirror} }));
-
- my $sth = $dbh->prepare("SELECT attach_id, attach_size FROM attachments ORDER BY attach_id DESC");
- $sth->execute();
- my ($count, $deleted, $stored) = (0, 0, 0);
- while (my ($attach_id, $attach_size) = $sth->fetchrow_array()) {
- indicate_progress({ total => $total, current => ++$count });
-
- # remove deleted attachments
- if ($attach_size == 0 && $dest->exists($attach_id)) {
- $dest->remove($attach_id);
- $deleted++;
- }
-
- # store attachments that don't already exist
- elsif ($attach_size != 0 && !$dest->exists($attach_id)) {
- if (my $data = $source->retrieve($attach_id)) {
- $dest->store($attach_id, $data);
- $stored++;
- }
- }
+
+ # store attachments that don't already exist
+ elsif ($attach_size != 0 && !$dest->exists($attach_id)) {
+ if (my $data = $source->retrieve($attach_id)) {
+ $dest->store($attach_id, $data);
+ $stored++;
+ }
}
- print "\n";
- print "Attachments stored: $stored\n";
- print "Attachments deleted: $deleted\n" if $deleted;
+ }
+ print "\n";
+ print "Attachments stored: $stored\n";
+ print "Attachments deleted: $deleted\n" if $deleted;
}
elsif ($options{copy}) {
- if ($options{copy}->[0] eq $options{copy}->[1]) {
- die "Source and destination must be different\n";
- }
- my ($source, $dest) = map { storage($_) } @{ $options{copy} };
-
- my ($total) = $dbh->selectrow_array("SELECT COUNT(*) FROM attachments WHERE attach_size != 0");
- confirm(sprintf('Copy %s attachments from %s to %s?', $total, @{ $options{copy} }));
-
- my $sth = $dbh->prepare("SELECT attach_id, attach_size FROM attachments WHERE attach_size != 0 ORDER BY attach_id DESC");
- $sth->execute();
- my ($count, $stored) = (0, 0);
- while (my ($attach_id, $attach_size) = $sth->fetchrow_array()) {
- indicate_progress({ total => $total, current => ++$count });
-
- # store attachments that don't already exist
- if (!$dest->exists($attach_id)) {
- if (my $data = $source->retrieve($attach_id)) {
- $dest->store($attach_id, $data);
- $stored++;
- }
- }
+ if ($options{copy}->[0] eq $options{copy}->[1]) {
+ die "Source and destination must be different\n";
+ }
+ my ($source, $dest) = map { storage($_) } @{$options{copy}};
+
+ my ($total)
+ = $dbh->selectrow_array(
+ "SELECT COUNT(*) FROM attachments WHERE attach_size != 0");
+ confirm(sprintf(
+ 'Copy %s attachments from %s to %s?', $total, @{$options{copy}}));
+
+ my $sth
+ = $dbh->prepare(
+ "SELECT attach_id, attach_size FROM attachments WHERE attach_size != 0 ORDER BY attach_id DESC"
+ );
+ $sth->execute();
+ my ($count, $stored) = (0, 0);
+ while (my ($attach_id, $attach_size) = $sth->fetchrow_array()) {
+ indicate_progress({total => $total, current => ++$count});
+
+ # store attachments that don't already exist
+ if (!$dest->exists($attach_id)) {
+ if (my $data = $source->retrieve($attach_id)) {
+ $dest->store($attach_id, $data);
+ $stored++;
+ }
}
- print "\n";
- print "Attachments stored: $stored\n";
+ }
+ print "\n";
+ print "Attachments stored: $stored\n";
}
elsif ($options{delete}) {
- my $storage = storage($options{delete});
- my ($total) = $dbh->selectrow_array("SELECT COUNT(*) FROM attachments WHERE attach_size != 0");
- confirm(sprintf('DELETE %s attachments from %s?', $total, $options{delete}));
-
- my $sth = $dbh->prepare("SELECT attach_id FROM attachments WHERE attach_size != 0 ORDER BY attach_id DESC");
- $sth->execute();
- my ($count, $deleted) = (0, 0);
- while (my ($attach_id) = $sth->fetchrow_array()) {
- indicate_progress({ total => $total, current => ++$count });
- if ($storage->exists($attach_id)) {
- $storage->remove($attach_id);
- $deleted++;
- }
+ my $storage = storage($options{delete});
+ my ($total)
+ = $dbh->selectrow_array(
+ "SELECT COUNT(*) FROM attachments WHERE attach_size != 0");
+ confirm(sprintf('DELETE %s attachments from %s?', $total, $options{delete}));
+
+ my $sth
+ = $dbh->prepare(
+ "SELECT attach_id FROM attachments WHERE attach_size != 0 ORDER BY attach_id DESC"
+ );
+ $sth->execute();
+ my ($count, $deleted) = (0, 0);
+ while (my ($attach_id) = $sth->fetchrow_array()) {
+ indicate_progress({total => $total, current => ++$count});
+ if ($storage->exists($attach_id)) {
+ $storage->remove($attach_id);
+ $deleted++;
}
- print "\n";
- print "Attachments deleted: $deleted\n";
+ }
+ print "\n";
+ print "Attachments deleted: $deleted\n";
}
sub storage {
- my ($name) = @_;
- my $storage = Bugzilla::Attachment::get_storage_by_name($name)
- or die "Invalid attachment location: $name\n";
- return $storage;
+ my ($name) = @_;
+ my $storage = Bugzilla::Attachment::get_storage_by_name($name)
+ or die "Invalid attachment location: $name\n";
+ return $storage;
}
sub confirm {
- my ($prompt) = @_;
- print $prompt, "\n\nPress <Ctrl-C> to stop or <Enter> to continue..\n";
- getc();
+ my ($prompt) = @_;
+ print $prompt, "\n\nPress <Ctrl-C> to stop or <Enter> to continue..\n";
+ getc();
}
generated by cgit v1.2.3-24-g4f1b (git 2.32.0) at 2024-05-03 06:15:10 +0200