~bzr-pqm/bzr/bzr.dev

5177.1.1 by Vincent Ladeuil
Manually assign docstrings to command objects, so that they work with python -OO
1
# Copyright (C) 2006, 2007, 2009, 2010 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 (
23
    config,
24
    gpg,
25
    )
26
from bzrlib.bzrdir import BzrDir
27
""")
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
28
from bzrlib.commands import Command
29
from bzrlib.option import Option
30
31
32
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
33
    __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.
34
35
    If location is not specified the local tree is used.
36
    If committer is not specified the default committer is used.
37
38
    This does not sign commits that already have signatures.
39
    """
1732.2.3 by Martin Pool
sign-my-commits just signs revisions in the branch's ancestry.
40
    # Note that this signs everything on the branch's ancestry
41
    # (both mainline and merged), but not other revisions that may be in the
42
    # repository
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
43
2598.1.2 by Martin Pool
Also check that option help ends in a period, and fix those that don't
44
    takes_options = [
45
            Option('dry-run',
46
                   help='Don\'t actually sign anything, just print'
47
                        ' the revisions that would be signed.'),
48
            ]
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
49
    takes_args = ['location?', 'committer?']
50
51
    def run(self, location=None, committer=None, dry_run=False):
52
        if location is None:
1773.4.1 by Martin Pool
Add pyflakes makefile target; fix many warnings
53
            bzrdir = BzrDir.open_containing('.')[0]
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
54
        else:
55
            # Passed in locations should be exact
1773.4.1 by Martin Pool
Add pyflakes makefile target; fix many warnings
56
            bzrdir = BzrDir.open(location)
1732.2.3 by Martin Pool
sign-my-commits just signs revisions in the branch's ancestry.
57
        branch = bzrdir.open_branch()
58
        repo = branch.repository
1770.2.12 by Aaron Bentley
Merge bzr.dev
59
        branch_config = branch.get_config()
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
60
61
        if committer is None:
1773.4.1 by Martin Pool
Add pyflakes makefile target; fix many warnings
62
            committer = branch_config.username()
63
        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.
64
65
        count = 0
1711.2.35 by John Arbash Meinel
sign-my-commits should take out a write lock.
66
        repo.lock_write()
67
        try:
3010.1.15 by Robert Collins
Manage write groups in sign_my_commits, for efficiency.
68
            repo.start_write_group()
69
            try:
70
                for rev_id in repo.get_ancestry(branch.last_revision())[1:]:
71
                    if repo.has_signature_for_revision_id(rev_id):
72
                        continue
73
                    rev = repo.get_revision(rev_id)
74
                    if rev.committer != committer:
75
                        continue
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
76
                    # 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.
77
                    # matching committer, start signing
78
                    print rev_id
79
                    count += 1
80
                    if not dry_run:
81
                        repo.sign_revision(rev_id, gpg_strategy)
82
            except:
83
                repo.abort_write_group()
84
                raise
85
            else:
86
                repo.commit_write_group()
1711.2.35 by John Arbash Meinel
sign-my-commits should take out a write lock.
87
        finally:
88
            repo.unlock()
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
89
        print 'Signed %d revisions' % (count,)
90
91