# Copyright (C) 2009-2012 Chris Ball <cjb@laptop.org>
# W. Trevor King <wking@tremily.us>
#
# This file is part of Bugs Everywhere.
#
# Bugs Everywhere is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 2 of the License, or (at your option) any
# later version.
#
# Bugs Everywhere is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# Bugs Everywhere. If not, see <http://www.gnu.org/licenses/>.
import glob
import os.path
import libbe
import libbe.command
class Completer (object):
def __init__(self, options):
self.options = options
def __call__(self, bugdirs, fragment=None):
return [fragment]
def complete_command(command, argument, fragment=None):
"""
List possible command completions for fragment.
command argument is not used.
"""
return list(libbe.command.commands(command_names=True))
def comp_path(fragment=None):
"""List possible path completions for fragment."""
if fragment == None:
fragment = '.'
comps = glob.glob(fragment+'*') + glob.glob(fragment+'/*')
if len(comps) == 1 and os.path.isdir(comps[0]):
comps.extend(glob.glob(comps[0]+'/*'))
return comps
def complete_path(command, argument, fragment=None):
"""List possible path completions for fragment."""
return comp_path(fragment)
def complete_status(command, argument, fragment=None):
bd = sorted(command._get_bugdirs().items())[1]
import libbe.bug
return libbe.bug.status_values
def complete_severity(command, argument, fragment=None):
bd = sorted(command._get_bugdirs().items())[1]
import libbe.bug
return libbe.bug.severity_values
def assignees(bugdirs):
ret = set()
for bugdir in bugdirs.values():
bugdir.load_all_bugs()
ret.update(set([bug.assigned for bug in bugdir
if bug.assigned != None]))
return list(ret)
def complete_assigned(command, argument, fragment=None):
return assignees(command._get_bugdirs())
def complete_extra_strings(command, argument, fragment=None):
if fragment == None:
return []
return [fragment]
def complete_bugdir_id(command, argument, fragment=None):
bugdirs = command._get_bugdirs()
return bugdirs.keys()
def complete_bug_id(command, argument, fragment=None):
return complete_bug_comment_id(command, argument, fragment,
comments=False)
def complete_bug_comment_id(command, argument, fragment=None,
active_only=True, comments=True):
import libbe.bugdir
import libbe.util.id
bugdirs = command._get_bugdirs()
if fragment == None or len(fragment) == 0:
fragment = '/'
try:
p = libbe.util.id.parse_user(bugdirs, fragment)
matches = None
root,residual = (fragment, None)
if not root.endswith('/'):
root += '/'
except libbe.util.id.InvalidIDStructure, e:
return []
except libbe.util.id.NoIDMatches:
return []
except libbe.util.id.MultipleIDMatches, e:
if e.common == None:
# choose among bugdirs
return e.matches
common = e.common
matches = e.matches
root,residual = libbe.util.id.residual(common, fragment)
p = libbe.util.id.parse_user(bugdirs, e.common)
bug = None
if matches == None: # fragment was complete, get a list of children uuids
if p['type'] == 'bugdir':
bugdir = bugdirs[p['bugdir']]
matches = bugdir.uuids()
common = bugdir.id.user()
elif p['type'] == 'bug':
if comments == False:
return [fragment]
bugdir = bugdirs[p['bugdir']]
bug = bugdir.bug_from_uuid(p['bug'])
matches = bug.uuids()
common = bug.id.user()
else:
assert p['type'] == 'comment', p
return [fragment]
if p['type'] == 'bugdir':
bugdir = bugdirs[p['bugdir']]
child_fn = bugdir.bug_from_uuid
elif p['type'] == 'bug':
if comments == False:
return[fragment]
bugdir = bugdirs[p['bugdir']]
if bug == None:
bug = bugdir.bug_from_uuid(p['bug'])
child_fn = bug.comment_from_uuid
elif p['type'] == 'comment':
assert matches == None, matches
return [fragment]
possible = []
common += '/'
for m in matches:
child = child_fn(m)
id = child.id.user()
possible.append(id.replace(common, root))
return possible
def select_values(string, possible_values, name="unkown"):
"""
This function allows the user to select values from a list of
possible values. The default is to select all the values:
>>> select_values(None, ['abc', 'def', 'hij'])
['abc', 'def', 'hij']
The user selects values with a comma-separated limit_string.
Prepending a minus sign to such a list denotes blacklist mode:
>>> select_values('-abc,hij', ['abc', 'def', 'hij'])
['def']
Without the leading -, the selection is in whitelist mode:
>>> select_values('abc,hij', ['abc', 'def', 'hij'])
['abc', 'hij']
In either case, appropriate errors are raised if on of the
user-values is not in the list of possible values. The name
parameter lets you make the error message more clear:
>>> select_values('-xyz,hij', ['abc', 'def', 'hij'], name="foobar")
Traceback (most recent call last):
...
UserError: Invalid foobar xyz
['abc', 'def', 'hij']
>>> select_values('xyz,hij', ['abc', 'def', 'hij'], name="foobar")
Traceback (most recent call last):
...
UserError: Invalid foobar xyz
['abc', 'def', 'hij']
"""
possible_values = list(possible_values) # don't alter the original
if string == None:
pass
elif string.startswith('-'):
blacklisted_values = set(string[1:].split(','))
for value in blacklisted_values:
if value not in possible_values:
raise libbe.command.UserError('Invalid %s %s\n %s'
% (name, value, possible_values))
possible_values.remove(value)
else:
whitelisted_values = string.split(',')
for value in whitelisted_values:
if value not in possible_values:
raise libbe.command.UserError(
'Invalid %s %s\n %s'
% (name, value, possible_values))
possible_values = whitelisted_values
return possible_values
def bugdir_bug_comment_from_user_id(bugdirs, id):
p = libbe.util.id.parse_user(bugdirs, id)
if not p['type'] in ['bugdir', 'bug', 'comment']:
raise libbe.command.UserError(
'{} is a {} id, not a bugdir, bug, or comment id'.format(
id, p['type']))
if p['bugdir'] not in bugdirs:
raise libbe.command.UserError(
"{} doesn't belong to any bugdirs in {}".format(
id, sorted(bugdirs.keys())))
bugdir = bugdirs[p['bugdir']]
if p['bugdir'] != bugdir.uuid:
raise libbe.command.UserError(
"%s doesn't belong to this bugdir (%s)"
% (id, bugdir.uuid))
if 'bug' in p:
bug = bugdir.bug_from_uuid(p['bug'])
if 'comment' in p:
comment = bug.comment_from_uuid(p['comment'])
else:
comment = bug.comment_root
else:
bug = comment = None
return (bugdir, bug, comment)
def bug_from_uuid(bugdirs, uuid):
error = None
for bugdir in bugdirs.values():
try:
bug = bugdir.bug_from_uuid(uuid)
except libbe.bugdir.NoBugMatches as e:
error = e
else:
return bug
raise error
