~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/weave_commands.py

  • Committer: mbp at sourcefrog
  • Date: 2005-04-11 02:53:57 UTC
  • Revision ID: mbp@sourcefrog.net-20050411025357-af577721308648ae
- remove profiler temporary file when done

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
"""builtin bzr commands relating to individual weave files
18
 
 
19
 
These should never normally need to be used by end users, but might be
20
 
of interest in debugging or data recovery.
21
 
"""
22
 
 
23
 
import sys
24
 
 
25
 
from bzrlib.commands import Command
26
 
from bzrlib.trace import warning
27
 
 
28
 
class cmd_versionedfile_list(Command):
29
 
    """List the revision ids present in a versionedfile, alphabetically"""
30
 
 
31
 
    hidden = True
32
 
    takes_args = ['filename']
33
 
    aliases = ['weave-list']
34
 
 
35
 
    def run(self, filename):
36
 
        from bzrlib.weavefile import read_weave
37
 
        from bzrlib.transport import get_transport
38
 
        from bzrlib import osutils
39
 
        from bzrlib.knit import KnitVersionedFile
40
 
        if filename.endswith('.knit'):
41
 
            transport = get_transport(osutils.dirname(filename))
42
 
            relpath = osutils.basename(filename)[:-len('.knit')]
43
 
            vf = KnitVersionedFile(relpath, transport)
44
 
        else:
45
 
            vf = read_weave(file(filename, 'rb'))
46
 
        names = vf.versions()
47
 
        names.sort()
48
 
        print '\n'.join(names)
49
 
 
50
 
 
51
 
class cmd_weave_join(Command):
52
 
    """Join the contents of two weave files.
53
 
 
54
 
    The resulting weave is sent to stdout.
55
 
 
56
 
    This command is only intended for bzr developer use.
57
 
    """
58
 
 
59
 
    hidden = True
60
 
    takes_args = ['weave1', 'weave2']
61
 
 
62
 
    def run(self, weave1, weave2):
63
 
        from bzrlib.weavefile import read_weave, write_weave
64
 
        w1 = read_weave(file(weave1, 'rb'))
65
 
        w2 = read_weave(file(weave2, 'rb'))
66
 
        w1.join(w2)
67
 
        write_weave(w1, sys.stdout)
68
 
 
69
 
 
70
 
class cmd_weave_plan_merge(Command):
71
 
    """Show the plan for merging two versions within a weave"""
72
 
    hidden = True
73
 
    takes_args = ['weave_file', 'revision_a', 'revision_b']
74
 
 
75
 
    def run(self, weave_file, revision_a, revision_b):
76
 
        from bzrlib.weavefile import read_weave
77
 
        w = read_weave(file(weave_file, 'rb'))
78
 
        for state, line in w.plan_merge(revision_a, revision_b):
79
 
            # make sure to print every line with a newline, even if it doesn't
80
 
            # really have one
81
 
            if not line:
82
 
                continue
83
 
            if line[-1] != '\n':
84
 
                state += '!eol'
85
 
                line += '\n'
86
 
            if '\n' in line[:-1]:
87
 
                warning("line in weave contains embedded newline: %r" % line)
88
 
            print '%15s | %s' % (state, line),
89
 
 
90
 
class cmd_weave_merge_text(Command):
91
 
    """Debugging command to merge two texts of a weave"""
92
 
    hidden = True
93
 
    takes_args = ['weave_file', 'revision_a', 'revision_b']
94
 
 
95
 
    def run(self, weave_file, revision_a, revision_b):
96
 
        from bzrlib.weavefile import read_weave
97
 
        w = read_weave(file(weave_file, 'rb'))
98
 
        p = w.plan_merge(revision_a, revision_b)
99
 
        sys.stdout.writelines(w.weave_merge(p))