~bzr-pqm/bzr/bzr.dev

5752.3.8 by John Arbash Meinel
Merge bzr.dev 5764 to resolve release-notes (aka NEWS) conflicts
1
# Copyright (C) 2006, 2007, 2009, 2010, 2011 Canonical Ltd
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
2
#
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
7
#
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
12
#
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1732.2.3 by Martin Pool
sign-my-commits just signs revisions in the branch's ancestry.
16
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
17
"""Command which looks for unsigned commits by the current user, and signs them.
18
"""
19
1996.3.14 by John Arbash Meinel
lazy_import osutils and sign_my_commits
20
from bzrlib.lazy_import import lazy_import
21
lazy_import(globals(), """
22
from bzrlib import (
5753.2.2 by Jelmer Vernooij
Remove some unnecessary imports, clean up lazy imports.
23
    bzrdir as _mod_bzrdir,
1996.3.14 by John Arbash Meinel
lazy_import osutils and sign_my_commits
24
    gpg,
25
    )
26
""")
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
27
from bzrlib.commands import Command
28
from bzrlib.option import Option
29
30
31
class cmd_sign_my_commits(Command):
5131.2.1 by Martin
Permit bzrlib to run under python -OO by explictly assigning to __doc__ for user-visible docstrings
32
    __doc__ = """Sign all commits by a given committer.
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
33
34
    If location is not specified the local tree is used.
35
    If committer is not specified the default committer is used.
36
37
    This does not sign commits that already have signatures.
38
    """
1732.2.3 by Martin Pool
sign-my-commits just signs revisions in the branch's ancestry.
39
    # Note that this signs everything on the branch's ancestry
40
    # (both mainline and merged), but not other revisions that may be in the
41
    # repository
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
42
2598.1.2 by Martin Pool
Also check that option help ends in a period, and fix those that don't
43
    takes_options = [
44
            Option('dry-run',
45
                   help='Don\'t actually sign anything, just print'
46
                        ' the revisions that would be signed.'),
47
            ]
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
48
    takes_args = ['location?', 'committer?']
49
50
    def run(self, location=None, committer=None, dry_run=False):
51
        if location is None:
5753.2.2 by Jelmer Vernooij
Remove some unnecessary imports, clean up lazy imports.
52
            bzrdir = _mod_bzrdir.BzrDir.open_containing('.')[0]
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
53
        else:
54
            # Passed in locations should be exact
5753.2.2 by Jelmer Vernooij
Remove some unnecessary imports, clean up lazy imports.
55
            bzrdir = _mod_bzrdir.BzrDir.open(location)
1732.2.3 by Martin Pool
sign-my-commits just signs revisions in the branch's ancestry.
56
        branch = bzrdir.open_branch()
57
        repo = branch.repository
1770.2.12 by Aaron Bentley
Merge bzr.dev
58
        branch_config = branch.get_config()
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
59
60
        if committer is None:
1773.4.1 by Martin Pool
Add pyflakes makefile target; fix many warnings
61
            committer = branch_config.username()
62
        gpg_strategy = gpg.GPGStrategy(branch_config)
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
63
64
        count = 0
1711.2.35 by John Arbash Meinel
sign-my-commits should take out a write lock.
65
        repo.lock_write()
66
        try:
3010.1.15 by Robert Collins
Manage write groups in sign_my_commits, for efficiency.
67
            repo.start_write_group()
68
            try:
69
                for rev_id in repo.get_ancestry(branch.last_revision())[1:]:
70
                    if repo.has_signature_for_revision_id(rev_id):
71
                        continue
72
                    rev = repo.get_revision(rev_id)
73
                    if rev.committer != committer:
74
                        continue
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
75
                    # We have a revision without a signature who has a
3010.1.15 by Robert Collins
Manage write groups in sign_my_commits, for efficiency.
76
                    # matching committer, start signing
77
                    print rev_id
78
                    count += 1
79
                    if not dry_run:
80
                        repo.sign_revision(rev_id, gpg_strategy)
81
            except:
82
                repo.abort_write_group()
83
                raise
84
            else:
85
                repo.commit_write_group()
1711.2.35 by John Arbash Meinel
sign-my-commits should take out a write lock.
86
        finally:
87
            repo.unlock()
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
88
        print 'Signed %d revisions' % (count,)
89
90