aboutsummaryrefslogtreecommitdiffstats
path: root/becommands/depend.py
blob: 977edae5d52e3a37f2f55df068392eed61eb7607 (plain) (blame)
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
# Copyright (C) 2009 W. Trevor King <wking@drexel.edu>
#
# This program 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.
#
# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
"""Add/remove bug dependencies"""
from libbe import cmdutil, bugdir
import os, copy
__desc__ = __doc__

def execute(args, manipulate_encodings=True):
    """
    >>> from libbe import utility
    >>> bd = bugdir.simple_bug_dir()
    >>> bd.save()
    >>> os.chdir(bd.root)
    >>> execute(["a", "b"], manipulate_encodings=False)
    Blocks on a:
    b
    >>> execute(["a"], manipulate_encodings=False)
    Blocks on a:
    b
    >>> execute(["--show-status", "a"], manipulate_encodings=False) # doctest: +NORMALIZE_WHITESPACE
    Blocks on a:
    b closed
    >>> execute(["-r", "a", "b"], manipulate_encodings=False)
    """
    parser = get_parser()
    options, args = parser.parse_args(args)
    cmdutil.default_complete(options, args, parser,
                             bugid_args={0: lambda bug : bug.active==True,
                                         1: lambda bug : bug.active==True})

    if len(args) < 1:
        raise cmdutil.UsageError("Please a bug id.")
    if len(args) > 2:
        help()
        raise cmdutil.UsageError("Too many arguments.")
    
    bd = bugdir.BugDir(from_disk=True,
                       manipulate_encodings=manipulate_encodings)
    bugA = bd.bug_from_shortname(args[0])
    if len(args) == 2:
        bugB = bd.bug_from_shortname(args[1])
        estrs = bugA.extra_strings
        depend_string = "BLOCKED-BY:%s" % bugB.uuid
        if options.remove == True:
            estrs.remove(depend_string)
        else: # add the dependency
            estrs.append(depend_string)
        bugA.extra_strings = estrs # reassign to notice change
        bugA.save()

    depends = []
    for estr in bugA.extra_strings:
        if estr.startswith("BLOCKED-BY:"):
            uuid = estr[11:]
            if options.show_status == True:
                blocker = bd.bug_from_uuid(uuid)
                block_string = "%s\t%s" % (uuid, blocker.status)
            else:
                block_string = uuid
            depends.append(block_string)
    if len(depends) > 0:
        print "Blocks on %s:" % bugA.uuid
        print '\n'.join(depends)

def get_parser():
    parser = cmdutil.CmdOptionParser("be depend BUG-ID [BUG-ID]")
    parser.add_option("-r", "--remove", action="store_true", dest="remove",
                      help="Remove dependency (instead of adding it)")
    parser.add_option("-s", "--show-status", action="store_true",
                      dest="show_status",
                      help="Show status of blocking bugs")
    return parser

longhelp="""
Set a dependency with the second bug (B) blocking the first bug (A).
If bug B is not specified, just print a list of bugs blocking (A).

To search for bugs blocked by a particular bug, try
  $ be list --extra-strings BLOCKED-BY:<your-bug-uuid>
"""

def help():
    return get_parser().help_str() + longhelp