def login(self, args=None):
"""Authenticate a session.
"""
# prompt for username if we were not supplied with it
if not self.user:
log_info('No username given.')
self.user = self.get_input('Username: '******'No password given.')
self.password = getpass.getpass()
else:
process = subprocess.Popen(self.passwordcmd.split(), shell=False,
stdout=subprocess.PIPE)
self.password, _ = process.communicate()
# perform login
params = {}
params['login'] = self.user
params['password'] = self.password
if args is not None:
params['remember'] = True
log_info('Logging in')
try:
self.bz.User.login(params)
except xmlrpclib.Fault as fault:
raise BugzError("Can't login: " + fault.faultString)
if args is not None:
self.cookiejar.save()
os.chmod(self.cookiejar.filename, 0600)
def modify(self, args):
"""Modify an existing bug (eg. adding a comment or changing resolution.)"""
if args.comment_from:
try:
if args.comment_from == '-':
args.comment = sys.stdin.read()
else:
args.comment = open(args.comment_from, 'r').read()
except IOError, e:
raise BugzError('unable to read file: %s: %s' % \
(args.comment_from, e))
def get(self, args):
""" Fetch bug details given the bug id """
log_info('Getting bug %s ..' % args.bugid)
try:
result = self.bzcall(self.bz.Bug.get, {'ids':[args.bugid]})
except xmlrpclib.Fault as fault:
raise BugzError("Can't get bug #" + str(args.bugid) + ": " \
+ fault.faultString)
for bug in result['bugs']:
self.showbuginfo(bug, args.attachments, args.comments)
def post(self, args):
"""Post a new bug"""
# load description from file if possible
if args.description_from is not None:
try:
if args.description_from == '-':
args.description = sys.stdin.read()
else:
args.description = open( args.description_from, 'r').read()
except IOError, e:
raise BugzError('Unable to read from file: %s: %s' %
(args.description_from, e))
def search(self, args):
"""Performs a search on the bugzilla database with the keywords given on the title (or the body if specified).
"""
valid_keys = ['alias', 'assigned_to', 'component', 'creator',
'limit', 'offset', 'op_sys', 'platform',
'priority', 'product', 'resolution',
'severity', 'status', 'version', 'whiteboard']
search_opts = sorted([(opt, val) for opt, val in args.__dict__.items()
if val is not None and opt in valid_keys])
params = {}
for key in args.__dict__.keys():
if key in valid_keys and getattr(args, key) is not None:
params[key] = getattr(args, key)
if getattr(args, 'terms'):
params['summary'] = args.terms
search_term = ' '.join(args.terms).strip()
if not (params or search_term):
raise BugzError('Please give search terms or options.')
if search_term:
log_msg = 'Searching for \'%s\' ' % search_term
else:
log_msg = 'Searching for bugs '
if search_opts:
log_info(log_msg + 'with the following options:')
for opt, val in search_opts:
log_info(' %-20s = %s' % (opt, val))
else:
log_info(log_msg)
if not 'status' in params.keys():
params['status'] = ['CONFIRMED', 'IN_PROGRESS', 'UNCONFIRMED']
elif 'ALL' in params['status']:
del params['status']
result = self.bzcall(self.bz.Bug.search, params)['bugs']
if not len(result):
log_info('No bugs found.')
else:
self.listbugs(result, args.show_status)
def attach(self, args):
""" Attach a file to a bug given a filename. """
filename = args.filename
content_type = args.content_type
bugid = args.bugid
summary = args.summary
is_patch = args.is_patch
comment = args.comment
if not os.path.exists(filename):
raise BugzError('File not found: %s' % filename)
if content_type is None:
content_type = get_content_type(filename)
if comment is None:
comment = block_edit('Enter optional long description of attachment')
if summary is None:
summary = os.path.basename(filename)
params = {}
params['ids'] = [bugid]
fd = open(filename, 'rb')
params['data'] = xmlrpclib.Binary(fd.read())
fd.close()
params['file_name'] = os.path.basename(filename)
params['summary'] = summary
if not is_patch:
params['content_type'] = content_type;
params['comment'] = comment
params['is_patch'] = is_patch
result = self.bzcall(self.bz.Bug.add_attachment, params)
log_info("'%s' has been attached to bug %s" % (filename, bugid))
def logout(self, args):
log_info('logging out')
try:
self.bz.User.logout()
except xmlrpclib.Fault as fault:
raise BugzError("Failed to logout: " + fault.faultString)
params['url'] = args.url
if args.version is not None:
params['version'] = args.version
if args.whiteboard is not None:
params['whiteboard'] = args.whiteboard
if args.fixed:
params['status'] = 'RESOLVED'
params['resolution'] = 'FIXED'
if args.invalid:
params['status'] = 'RESOLVED'
params['resolution'] = 'INVALID'
if len(params) < 2:
raise BugzError('No changes were specified')
result = self.bzcall(self.bz.Bug.update, params)
for bug in result['bugs']:
changes = bug['changes']
if not len(changes):
log_info('Added comment to bug %s' % bug['id'])
else:
log_info('Modified the following fields in bug %s' % bug['id'])
for key in changes.keys():
log_info('%-12s: removed %s' %(key, changes[key]['removed']))
log_info('%-12s: added %s' %(key, changes[key]['added']))
def attachment(self, args):
""" Download or view an attachment given the id."""
log_info('Getting attachment %s' % args.attachid)