1 # Copyright (c) 2011 Google Inc. All rights reserved. 2 # 3 # Redistribution and use in source and binary forms, with or without 4 # modification, are permitted provided that the following conditions are 5 # met: 6 # 7 # * Redistributions of source code must retain the above copyright 8 # notice, this list of conditions and the following disclaimer. 9 # * Redistributions in binary form must reproduce the above 10 # copyright notice, this list of conditions and the following disclaimer 11 # in the documentation and/or other materials provided with the 12 # distribution. 13 # * Neither the name of Google Inc. nor the names of its 14 # contributors may be used to endorse or promote products derived from 15 # this software without specific prior written permission. 16 # 17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29 from webkitpy.tool.multicommandtool import AbstractDeclarativeCommand 30 31 32 class AddUsersToGroups(AbstractDeclarativeCommand): 33 name = "add-users-to-groups" 34 help_text = "Add users matching subtring to specified groups" 35 36 # This probably belongs in bugzilla.py 37 known_groups = ['canconfirm', 'editbugs'] 38 39 def execute(self, options, args, tool): 40 search_string = args[0] 41 # FIXME: We could allow users to specify groups on the command line. 42 list_title = 'Add users matching "%s" which groups?' % search_string 43 # FIXME: Need a way to specify that "none" is not allowed. 44 # FIXME: We could lookup what groups the current user is able to grant from bugzilla. 45 groups = tool.user.prompt_with_list(list_title, self.known_groups, can_choose_multiple=True) 46 if not groups: 47 print "No groups specified." 48 return 49 50 login_userid_pairs = tool.bugs.queries.fetch_login_userid_pairs_matching_substring(search_string) 51 if not login_userid_pairs: 52 print "No users found matching '%s'" % search_string 53 return 54 55 print "Found %s users matching %s:" % (len(login_userid_pairs), search_string) 56 for (login, user_id) in login_userid_pairs: 57 print "%s (%s)" % (login, user_id) 58 59 confirm_message = "Are you sure you want add %s users to groups %s? (This action cannot be undone using webkit-patch.)" % (len(login_userid_pairs), groups) 60 if not tool.user.confirm(confirm_message): 61 return 62 63 for (login, user_id) in login_userid_pairs: 64 print "Adding %s to %s" % (login, groups) 65 tool.bugs.add_user_to_groups(user_id, groups) 66