~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: Michael Ellerman
  • Date: 2005-12-10 22:11:13 UTC
  • mto: This revision was merged to the branch mainline in revision 1528.
  • Revision ID: michael@ellerman.id.au-20051210221113-99ca561aaab4661e
Simplify handling of DivergedBranches in cmd_pull()

Show diffs side-by-side

added added

removed removed

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