/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: John Arbash Meinel
  • Date: 2006-10-05 05:37:25 UTC
  • mto: This revision was merged to the branch mainline in revision 2080.
  • Revision ID: john@arbash-meinel.com-20061005053725-3d2e757dab45eb94
Add tests to cleanup the copyright of all source files

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/commands.py
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
"""\
17
17
This is an attempt to take the internal delta object, and represent
18
18
it as a single-file text-only changeset.
21
21
"""
22
22
 
23
23
import sys
24
 
from cStringIO import StringIO
25
 
 
26
 
from bzrlib.lazy_import import lazy_import
27
 
lazy_import(globals(), """
28
 
from bzrlib import (
29
 
    branch,
30
 
    errors,
31
 
    merge_directive,
32
 
    revision as _mod_revision,
33
 
    urlutils,
34
 
    transport,
35
 
    )
36
 
""")
37
 
 
38
 
from bzrlib.commands import Command
 
24
 
 
25
from bzrlib.branch import Branch
 
26
from bzrlib.commands import Command, register_command
 
27
import bzrlib.errors as errors
39
28
from bzrlib.option import Option
 
29
from bzrlib.revision import (common_ancestor, MultipleRevisionSources,
 
30
                             NULL_REVISION)
40
31
from bzrlib.trace import note
41
 
 
42
 
 
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')
 
32
from bzrlib import urlutils
 
33
 
 
34
 
 
35
class cmd_send_changeset(Command):
 
36
    """Send a bundled up changset via mail.
 
37
 
 
38
    If no revision has been specified, the last commited change will
 
39
    be sent.
 
40
 
 
41
    Subject of the mail can be specified by the --message option,
 
42
    otherwise information from the changeset log will be used.
 
43
 
 
44
    A editor will be spawned where the user may enter a description
 
45
    of the changeset.  The description can be read from a file with
 
46
    the --file FILE option.
 
47
    """
 
48
    takes_options = ['revision', 'message', 'file']
 
49
    takes_args = ['to?']
 
50
 
 
51
    def run(self, to=None, message=None, revision=None, file=None):
 
52
        from bzrlib.errors import BzrCommandError
 
53
        from send_changeset import send_changeset
 
54
        
 
55
        if isinstance(revision, (list, tuple)):
 
56
            if len(revision) > 1:
 
57
                raise BzrCommandError('We do not support rollup-changesets yet.')
 
58
            revision = revision[0]
 
59
 
 
60
        b = Branch.open_containing('.')
 
61
 
 
62
        if not to:
 
63
            try:
 
64
                to = b.controlfile('x-send-address', 'rb').read().strip('\n')
 
65
            except:
 
66
                raise BzrCommandError('destination address is not known')
 
67
 
 
68
        if not isinstance(revision, (list, tuple)):
 
69
            revision = [revision]
 
70
 
 
71
        send_changeset(b, revision, to, message, file)
 
72
 
 
73
 
 
74
class cmd_bundle_revisions(Command):
 
75
    """Generate a revision bundle.
 
76
 
 
77
    This bundle contains all of the meta-information of a
 
78
    diff, rather than just containing the patch information.
 
79
 
 
80
    You can apply it to another tree using 'bzr merge'.
 
81
 
 
82
    bzr bundle-revisions
 
83
        - Generate a bundle relative to a remembered location
 
84
    bzr bundle-revisions BASE
 
85
        - Bundle to apply the current tree into BASE
 
86
    bzr bundle-revisions --revision A
 
87
        - Bundle to apply revision A to remembered location 
 
88
    bzr bundle-revisions --revision A..B
 
89
        - Bundle to transform A into B
 
90
    """
 
91
    takes_options = ['verbose', 'revision', 'remember',
 
92
                     Option("output", help="write bundle to specified file",
 
93
                            type=unicode)]
 
94
    takes_args = ['base?']
 
95
    aliases = ['bundle']
 
96
 
 
97
    def run(self, base=None, revision=None, output=None, remember=False):
 
98
        from bzrlib import user_encoding
 
99
        from bzrlib.bundle.serializer import write_bundle
 
100
 
 
101
        target_branch = Branch.open_containing(u'.')[0]
 
102
 
 
103
        if base is None:
 
104
            base_specified = False
 
105
        else:
 
106
            base_specified = True
 
107
 
 
108
        if revision is None:
 
109
            target_revision = target_branch.last_revision()
 
110
        elif len(revision) < 3:
 
111
            target_revision = revision[-1].in_history(target_branch).rev_id
 
112
            if len(revision) == 2:
 
113
                if base_specified:
 
114
                    raise errors.BzrCommandError('Cannot specify base as well'
 
115
                                                 ' as two revision arguments.')
 
116
                base_revision = revision[0].in_history(target_branch).rev_id
 
117
        else:
 
118
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
 
119
 
 
120
        if revision is None or len(revision) < 2:
 
121
            submit_branch = target_branch.get_submit_branch()
 
122
            if base is None:
 
123
                base = submit_branch
 
124
            if base is None:
 
125
                base = target_branch.get_parent()
 
126
            if base is None:
 
127
                raise errors.BzrCommandError("No base branch known or"
 
128
                                             " specified.")
 
129
            elif not base_specified:
 
130
                # FIXME:
 
131
                # note() doesn't pay attention to terminal_encoding() so
 
132
                # we must format with 'ascii' to be safe
 
133
                note('Using saved location: %s',
 
134
                     urlutils.unescape_for_display(base, 'ascii'))
 
135
            base_branch = Branch.open(base)
 
136
 
 
137
            # We don't want to lock the same branch across
 
138
            # 2 different branches
 
139
            if target_branch.base == base_branch.base:
 
140
                base_branch = target_branch 
 
141
            if submit_branch is None or remember:
 
142
                if base_specified:
 
143
                    target_branch.set_submit_branch(base_branch.base)
 
144
                elif remember:
 
145
                    raise errors.BzrCommandError('--remember requires a branch'
 
146
                                                 ' to be specified.')
 
147
            target_branch.repository.fetch(base_branch.repository, 
 
148
                                           base_branch.last_revision())
 
149
            base_revision = common_ancestor(base_branch.last_revision(),
 
150
                                            target_revision,
 
151
                                            target_branch.repository)
 
152
 
 
153
 
 
154
        if output is not None:
 
155
            fileobj = file(output, 'wb')
 
156
        else:
 
157
            fileobj = sys.stdout
 
158
        target_branch.repository.lock_read()
 
159
        try:
 
160
            write_bundle(target_branch.repository, target_revision,
 
161
                         base_revision, fileobj)
 
162
        finally:
 
163
            target_branch.repository.unlock()
 
164
 
 
165
 
 
166
class cmd_verify_changeset(Command):
 
167
    """Read a written changeset, and make sure it is valid.
 
168
 
 
169
    """
 
170
    takes_args = ['filename?']
 
171
 
 
172
    def run(self, filename=None):
 
173
        from read_changeset import read_changeset
 
174
        #from bzrlib.xml import serializer_v4
 
175
 
 
176
        b, relpath = Branch.open_containing('.')
 
177
 
 
178
        if filename is None or filename == '-':
 
179
            f = sys.stdin
 
180
        else:
 
181
            f = open(filename, 'U')
 
182
 
 
183
        cset_info, cset_tree = read_changeset(f, b)
 
184
        # print cset_info
 
185
        # print cset_tree
 
186
        #serializer_v4.write(cset_tree.inventory, sys.stdout)