~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: John Arbash Meinel
  • Date: 2005-09-15 21:35:53 UTC
  • mfrom: (907.1.57)
  • mto: (1393.2.1)
  • mto: This revision was merged to the branch mainline in revision 1396.
  • Revision ID: john@arbash-meinel.com-20050915213552-a6c83a5ef1e20897
(broken) Transport work is merged in. Tests do not pass yet.

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)