← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: John Arbash Meinel
  • Date: 2006-09-13 02:44:23 UTC
  • mto: This revision was merged to the branch mainline in revision 2071.
  • Revision ID: john@arbash-meinel.com-20060913024423-2f0729076ddd4e31
lazy_import osutils and sign_my_commits
Move doc tests into test_osutils, since lazy_import doesn't play nicely
with DocTest

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/commands.py
1
 
# Copyright (C) 2005-2010 Canonical Ltd
2
 
#
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.
7
 
#
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.
12
 
#
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
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
1
#!/usr/bin/env python
16
2
"""\
17
3
This is an attempt to take the internal delta object, and represent
18
4
it as a single-file text-only changeset.
21
7
"""
22
8
 
23
9
import sys
24
 
from cStringIO import StringIO
25
10
 
26
11
from bzrlib.lazy_import import lazy_import
27
12
lazy_import(globals(), """
28
13
from bzrlib import (
29
14
    branch,
30
15
    errors,
31
 
    merge_directive,
32
 
    revision as _mod_revision,
33
16
    urlutils,
34
 
    transport,
35
17
    )
 
18
from bzrlib.revision import common_ancestor
36
19
""")
37
20
 
38
21
from bzrlib.commands import Command
40
23
from bzrlib.trace import note
41
24
 
42
25
 
43
 
class cmd_bundle_info(Command):
44
 
    __doc__ = """Output interesting stats about a bundle"""
45
 
 
46
 
    hidden = True
47
 
    takes_args = ['location']
48
 
    takes_options = ['verbose']
49
 
    encoding_type = 'exact'
50
 
 
51
 
    def run(self, location, verbose=False):
52
 
        from bzrlib.bundle.serializer import read_bundle
53
 
        from bzrlib.bundle import read_mergeable_from_url
54
 
        from bzrlib import osutils
55
 
        term_encoding = osutils.get_terminal_encoding()
56
 
        bundle_info = read_mergeable_from_url(location)
57
 
        if isinstance(bundle_info, merge_directive.BaseMergeDirective):
58
 
            bundle_file = StringIO(bundle_info.get_raw_bundle())
59
 
            bundle_info = read_bundle(bundle_file)
60
 
        else:
61
 
            if verbose:
62
 
                raise errors.BzrCommandError('--verbose requires a merge'
63
 
                    ' directive')
64
 
        reader_method = getattr(bundle_info, 'get_bundle_reader', None)
65
 
        if reader_method is None:
66
 
            raise errors.BzrCommandError('Bundle format not supported')
67
 
 
68
 
        by_kind = {}
69
 
        file_ids = set()
70
 
        for bytes, parents, repo_kind, revision_id, file_id\
71
 
            in reader_method().iter_records():
72
 
            by_kind.setdefault(repo_kind, []).append(
73
 
                (bytes, parents, repo_kind, revision_id, file_id))
74
 
            if file_id is not None:
75
 
                file_ids.add(file_id)
76
 
        self.outf.write('Records\n')
77
 
        for kind, records in sorted(by_kind.iteritems()):
78
 
            multiparent = sum(1 for b, m, k, r, f in records if
79
 
                              len(m.get('parents', [])) > 1)
80
 
            self.outf.write('%s: %d (%d multiparent)\n' % \
81
 
                (kind, len(records), multiparent))
82
 
        self.outf.write('unique files: %d\n' % len(file_ids))
83
 
        self.outf.write('\n')
84
 
        nicks = set()
85
 
        committers = set()
86
 
        for revision in bundle_info.real_revisions:
87
 
            if 'branch-nick' in revision.properties:
88
 
                nicks.add(revision.properties['branch-nick'])
89
 
            committers.add(revision.committer)
90
 
 
91
 
        self.outf.write('Revisions\n')
92
 
        self.outf.write(('nicks: %s\n'
93
 
            % ', '.join(sorted(nicks))).encode(term_encoding, 'replace'))
94
 
        self.outf.write(('committers: \n%s\n' %
95
 
        '\n'.join(sorted(committers)).encode(term_encoding, 'replace')))
96
 
        if verbose:
97
 
            self.outf.write('\n')
98
 
            bundle_file.seek(0)
99
 
            line = bundle_file.readline()
100
 
            line = bundle_file.readline()
101
 
            content = bundle_file.read().decode('bz2')
102
 
            self.outf.write("Decoded contents\n")
103
 
            self.outf.write(content)
104
 
            self.outf.write('\n')
 
26
class cmd_send_changeset(Command):
 
27
    """Send a bundled up changset via mail.
 
28
 
 
29
    If no revision has been specified, the last commited change will
 
30
    be sent.
 
31
 
 
32
    Subject of the mail can be specified by the --message option,
 
33
    otherwise information from the changeset log will be used.
 
34
 
 
35
    A editor will be spawned where the user may enter a description
 
36
    of the changeset.  The description can be read from a file with
 
37
    the --file FILE option.
 
38
    """
 
39
    takes_options = ['revision', 'message', 'file']
 
40
    takes_args = ['to?']
 
41
 
 
42
    def run(self, to=None, message=None, revision=None, file=None):
 
43
        from bzrlib.errors import BzrCommandError
 
44
        from send_changeset import send_changeset
 
45
        
 
46
        if isinstance(revision, (list, tuple)):
 
47
            if len(revision) > 1:
 
48
                raise BzrCommandError('We do not support rollup-changesets yet.')
 
49
            revision = revision[0]
 
50
 
 
51
        b = branch.Branch.open_containing('.')
 
52
 
 
53
        if not to:
 
54
            try:
 
55
                to = b.controlfile('x-send-address', 'rb').read().strip('\n')
 
56
            except:
 
57
                raise BzrCommandError('destination address is not known')
 
58
 
 
59
        if not isinstance(revision, (list, tuple)):
 
60
            revision = [revision]
 
61
 
 
62
        send_changeset(b, revision, to, message, file)
 
63
 
 
64
 
 
65
class cmd_bundle_revisions(Command):
 
66
    """Generate a revision bundle.
 
67
 
 
68
    This bundle contains all of the meta-information of a
 
69
    diff, rather than just containing the patch information.
 
70
 
 
71
    You can apply it to another tree using 'bzr merge'.
 
72
 
 
73
    bzr bundle-revisions
 
74
        - Generate a bundle relative to a remembered location
 
75
    bzr bundle-revisions BASE
 
76
        - Bundle to apply the current tree into BASE
 
77
    bzr bundle-revisions --revision A
 
78
        - Bundle to apply revision A to remembered location 
 
79
    bzr bundle-revisions --revision A..B
 
80
        - Bundle to transform A into B
 
81
    """
 
82
    takes_options = ['verbose', 'revision', 'remember',
 
83
                     Option("output", help="write bundle to specified file",
 
84
                            type=unicode)]
 
85
    takes_args = ['base?']
 
86
    aliases = ['bundle']
 
87
 
 
88
    def run(self, base=None, revision=None, output=None, remember=False):
 
89
        from bzrlib import user_encoding
 
90
        from bzrlib.bundle.serializer import write_bundle
 
91
 
 
92
        target_branch = branch.Branch.open_containing(u'.')[0]
 
93
 
 
94
        if base is None:
 
95
            base_specified = False
 
96
        else:
 
97
            base_specified = True
 
98
 
 
99
        if revision is None:
 
100
            target_revision = target_branch.last_revision()
 
101
        elif len(revision) < 3:
 
102
            target_revision = revision[-1].in_history(target_branch).rev_id
 
103
            if len(revision) == 2:
 
104
                if base_specified:
 
105
                    raise errors.BzrCommandError('Cannot specify base as well'
 
106
                                                 ' as two revision arguments.')
 
107
                base_revision = revision[0].in_history(target_branch).rev_id
 
108
        else:
 
109
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
 
110
 
 
111
        if revision is None or len(revision) < 2:
 
112
            submit_branch = target_branch.get_submit_branch()
 
113
            if base is None:
 
114
                base = submit_branch
 
115
            if base is None:
 
116
                base = target_branch.get_parent()
 
117
            if base is None:
 
118
                raise errors.BzrCommandError("No base branch known or"
 
119
                                             " specified.")
 
120
            elif not base_specified:
 
121
                # FIXME:
 
122
                # note() doesn't pay attention to terminal_encoding() so
 
123
                # we must format with 'ascii' to be safe
 
124
                note('Using saved location: %s',
 
125
                     urlutils.unescape_for_display(base, 'ascii'))
 
126
            base_branch = branch.Branch.open(base)
 
127
 
 
128
            # We don't want to lock the same branch across
 
129
            # 2 different branches
 
130
            if target_branch.base == base_branch.base:
 
131
                base_branch = target_branch 
 
132
            if submit_branch is None or remember:
 
133
                if base_specified:
 
134
                    target_branch.set_submit_branch(base_branch.base)
 
135
                elif remember:
 
136
                    raise errors.BzrCommandError('--remember requires a branch'
 
137
                                                 ' to be specified.')
 
138
            target_branch.repository.fetch(base_branch.repository, 
 
139
                                           base_branch.last_revision())
 
140
            base_revision = common_ancestor(base_branch.last_revision(),
 
141
                                            target_revision,
 
142
                                            target_branch.repository)
 
143
 
 
144
 
 
145
        if output is not None:
 
146
            fileobj = file(output, 'wb')
 
147
        else:
 
148
            fileobj = sys.stdout
 
149
        target_branch.repository.lock_read()
 
150
        try:
 
151
            write_bundle(target_branch.repository, target_revision,
 
152
                         base_revision, fileobj)
 
153
        finally:
 
154
            target_branch.repository.unlock()
 
155
 
 
156
 
 
157
class cmd_verify_changeset(Command):
 
158
    """Read a written changeset, and make sure it is valid.
 
159
 
 
160
    """
 
161
    takes_args = ['filename?']
 
162
 
 
163
    def run(self, filename=None):
 
164
        from read_changeset import read_changeset
 
165
        #from bzrlib.xml import serializer_v4
 
166
 
 
167
        b, relpath = branch.Branch.open_containing('.')
 
168
 
 
169
        if filename is None or filename == '-':
 
170
            f = sys.stdin
 
171
        else:
 
172
            f = open(filename, 'U')
 
173
 
 
174
        cset_info, cset_tree = read_changeset(f, b)
 
175
        # print cset_info
 
176
        # print cset_tree
 
177
        #serializer_v4.write(cset_tree.inventory, sys.stdout)