~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: Robert Collins
  • Date: 2006-06-26 16:23:10 UTC
  • mfrom: (1780.2.1 misc-fixen)
  • mto: This revision was merged to the branch mainline in revision 1815.
  • Revision ID: robertc@robertcollins.net-20060626162310-98f5b55b8cc19d46
(robertc) Misc minor typos and the like.

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.revisionspec import RevisionSpec
 
18
from bzrlib.trace import note
 
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
        - Bundle for the last commit
 
70
    bzr bundle-revisions BASE
 
71
        - Bundle to apply the current tree into BASE
 
72
    bzr bundle-revisions --revision A
 
73
        - Bundle for revision A
 
74
    bzr bundle-revisions --revision A..B
 
75
        - Bundle to transform A into B
 
76
    bzr bundle-revisions --revision A..B BASE
 
77
        - Bundle to transform revision A of BASE into revision B
 
78
          of the local tree
 
79
    """
 
80
    takes_options = ['verbose', 'revision', 'remember',
 
81
                     Option("output", help="write bundle to specified file",
 
82
                            type=unicode)]
 
83
    takes_args = ['base?']
 
84
    aliases = ['bundle']
 
85
 
 
86
    def run(self, base=None, revision=None, output=None, remember=False):
 
87
        from bzrlib import user_encoding
 
88
        from bzrlib.bundle.serializer import write_bundle
 
89
 
 
90
        target_branch = Branch.open_containing(u'.')[0]
 
91
        submit_branch = target_branch.get_submit_branch()
 
92
 
 
93
        if base is None:
 
94
            base_specified = False
 
95
            base = submit_branch
 
96
            note('Using saved location: %s' % base)
 
97
        else:
 
98
            base_specified = True
 
99
 
 
100
        if base is None:
 
101
            base = target_branch.get_parent()
 
102
            if base is None:
 
103
                raise errors.BzrCommandError("No base branch known or"
 
104
                                             " specified.")
 
105
            else:
 
106
                note('Using saved location: %s' % base)
 
107
        base_branch = Branch.open(base)
 
108
 
 
109
        # We don't want to lock the same branch across
 
110
        # 2 different branches
 
111
        if target_branch.base == base_branch.base:
 
112
            base_branch = target_branch 
 
113
        if submit_branch is None or remember:
 
114
            if base_specified:
 
115
                target_branch.set_submit_branch(base_branch.base)
 
116
            elif remember:
 
117
                raise errors.BzrCommandError('--remember requires a branch to'
 
118
                                             ' be specified.')
 
119
 
 
120
        base_revision = None
 
121
        if revision is None:
 
122
            target_revision = target_branch.last_revision()
 
123
        elif len(revision) == 1:
 
124
            target_revision = revision[0].in_history(target_branch).rev_id
 
125
            if base_branch is not None:
 
126
                base_revision = base_branch.last_revision()
 
127
        elif len(revision) == 2:
 
128
            target_revision = revision[1].in_history(target_branch).rev_id
 
129
            if base_branch is not None:
 
130
                base_revision = revision[0].in_history(base_branch).rev_id
 
131
            else:
 
132
                base_revision = revision[0].in_history(target_branch).rev_id
 
133
        else:
 
134
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
 
135
 
 
136
        if revision is None or len(revision) == 1:
 
137
            if base_branch is not None:
 
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
                if base_revision is None:
 
144
                    base_revision = NULL_REVISION
 
145
 
 
146
        if base_revision is None:
 
147
            rev = target_branch.repository.get_revision(target_revision)
 
148
            if rev.parent_ids:
 
149
                base_revision = rev.parent_ids[0]
 
150
            else:
 
151
                base_revision = NULL_REVISION
 
152
 
 
153
        if base_branch is not None:
 
154
            target_branch.repository.fetch(base_branch.repository, 
 
155
                                           revision_id=base_revision)
 
156
            del base_branch
 
157
 
 
158
        if output is not None:
 
159
            fileobj = file(output, 'wb')
 
160
        else:
 
161
            fileobj = sys.stdout
 
162
        write_bundle(target_branch.repository, target_revision, base_revision,
 
163
                     fileobj)
 
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)