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
88
89
90
91
92
93
94
95
96
97
|
#!/usr/bin/perl
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# This Source Code Form is "Incompatible With Secondary Licenses", as
# defined by the Mozilla Public License, v. 2.0.
use strict;
use warnings;
$| = 1;
use FindBin qw($RealBin);
use lib ("$RealBin/..", "$RealBin/../lib", "$RealBin/../local/lib/perl5");
use Bugzilla;
use Bugzilla::CGI;
use Bugzilla::Constants;
use Bugzilla::Group;
use Bugzilla::Search;
use Bugzilla::User;
use Getopt::Long qw(GetOptions);
use URI;
use URI::QueryParam;
Bugzilla->usage_mode(USAGE_MODE_CMDLINE);
my $options = {};
GetOptions($options, 'add=s', 'remove=s') or exit(1);
my $url = URI->new(shift);
unless ($url && ($options->{add} || $options->{remove})) {
die <<EOF;
Syntax:
update-bug-groups.pl [--add group-name] [--remove group-name] buglist-url
Synopsis:
This script finds bugs matching the search URL and:
--add : adds the provided group to bugs
--remove : removes the provided group from bugs
At least --add or --remove must be specified; both options can be provided
at the same time.
EOF
}
die "Invalid buglist.cgi query\n" unless $url->path =~ m#/buglist\.cgi$#;
$url->query_param( limit => 0 );
my ($add_group, $remove_group);
$add_group = Bugzilla::Group->check({ name => $options->{add} }) if $options->{add};
$remove_group = Bugzilla::Group->check({ name => $options->{remove} }) if $options->{remove};
my $user = Bugzilla::User->check({ name => 'automation@bmo.tld' });
$user->{groups} = [ Bugzilla::Group->get_all ];
$user->{bless_groups} = [ Bugzilla::Group->get_all ];
Bugzilla->set_user($user);
# find the bugs
my $params = Bugzilla::CGI->new($url->query);
my $search = Bugzilla::Search->new(
fields => [ 'bug_id', 'short_desc' ],
params => scalar $params->Vars,
user => $user,
);
my $bugs = $search->data;
my $count = scalar @$bugs;
# update
die "No bugs found\n" unless $count;
print "Query matched $count bug(s)\nPress <Ctrl-C> to stop or <Enter> to continue..\n";
getc();
my $dbh = Bugzilla->dbh;
my $updated = 0;
foreach my $ra (@$bugs) {
$dbh->bz_start_transaction;
my ($bug_id, $summary) = @$ra;
print "$bug_id - $summary\n";
my $bug = Bugzilla::Bug->check($bug_id);
$bug->add_group($add_group) if $add_group;
$bug->remove_group($remove_group) if $remove_group;
my $changes = $bug->update();
if (scalar keys %$changes) {
$dbh->do("UPDATE bugs SET lastdiffed = delta_ts WHERE bug_id = ?", undef, $bug->id);
$updated++;
}
$dbh->bz_commit_transaction;
# drop cached user objects to avoid excessive memory usage
Bugzilla::User->object_cache_clearall();
}
print "\nUpdated $updated bugs(s)\n";
|
