/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/revisiontree.py

  • Committer: John Arbash Meinel
  • Date: 2007-02-08 16:28:05 UTC
  • mto: This revision was merged to the branch mainline in revision 2278.
  • Revision ID: john@arbash-meinel.com-20070208162805-dcqiqrwjh9a5lo7n
``GPGStrategy.sign()`` will now raise ``BzrBadParameterUnicode`` if
you pass a Unicode string rather than an 8-bit string. It doesn't 
make sense to sign a Unicode string, and it turns out that some 
versions of python will write out the raw Unicode bytes rather than
encoding automatically. So fail and make callers do the right thing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/revisiontree.py
 
1
# Copyright (C) 2005 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
"""RevisionTree - a Tree implementation backed by repository data for a revision."""
 
18
 
 
19
from cStringIO import StringIO
 
20
 
 
21
from bzrlib.tree import Tree
 
22
 
 
23
 
 
24
class RevisionTree(Tree):
 
25
    """Tree viewing a previous revision.
 
26
 
 
27
    File text can be retrieved from the text store.
 
28
 
 
29
    TODO: Some kind of `__repr__` method, but a good one
 
30
           probably means knowing the branch and revision number,
 
31
           or at least passing a description to the constructor.
 
32
    """
 
33
    
 
34
    def __init__(self, branch, inv, revision_id):
 
35
        # for compatability the 'branch' parameter has not been renamed to 
 
36
        # repository at this point. However, we should change RevisionTree's
 
37
        # construction to always be via Repository and not via direct 
 
38
        # construction - this will mean that we can change the constructor
 
39
        # with much less chance of breaking client code.
 
40
        self._repository = branch
 
41
        self._weave_store = branch.weave_store
 
42
        self._inventory = inv
 
43
        self._revision_id = revision_id
 
44
 
 
45
    def get_parent_ids(self):
 
46
        """See Tree.get_parent_ids.
 
47
 
 
48
        A RevisionTree's parents match the revision graph.
 
49
        """
 
50
        if self._revision_id not in (None, 'null:'):
 
51
            parent_ids = self._repository.get_revision(
 
52
                self._revision_id).parent_ids
 
53
        else:
 
54
            parent_ids = []
 
55
        return parent_ids
 
56
        
 
57
    def get_revision_id(self):
 
58
        """Return the revision id associated with this tree."""
 
59
        return self._revision_id
 
60
 
 
61
    def get_weave(self, file_id):
 
62
        return self._weave_store.get_weave(file_id,
 
63
                self._repository.get_transaction())
 
64
 
 
65
    def get_file_lines(self, file_id):
 
66
        ie = self._inventory[file_id]
 
67
        weave = self.get_weave(file_id)
 
68
        return weave.get_lines(ie.revision)
 
69
 
 
70
    def get_file_text(self, file_id):
 
71
        return ''.join(self.get_file_lines(file_id))
 
72
 
 
73
    def get_file(self, file_id):
 
74
        return StringIO(self.get_file_text(file_id))
 
75
 
 
76
    def annotate_iter(self, file_id):
 
77
        """See Tree.annotate_iter"""
 
78
        w = self.get_weave(file_id)
 
79
        return w.annotate_iter(self.inventory[file_id].revision)
 
80
 
 
81
    def get_file_size(self, file_id):
 
82
        return self._inventory[file_id].text_size
 
83
 
 
84
    def get_file_sha1(self, file_id, path=None, stat_value=None):
 
85
        ie = self._inventory[file_id]
 
86
        if ie.kind == "file":
 
87
            return ie.text_sha1
 
88
        return None
 
89
 
 
90
    def get_file_mtime(self, file_id, path=None):
 
91
        ie = self._inventory[file_id]
 
92
        revision = self._repository.get_revision(ie.revision)
 
93
        return revision.timestamp
 
94
 
 
95
    def is_executable(self, file_id, path=None):
 
96
        ie = self._inventory[file_id]
 
97
        if ie.kind != "file":
 
98
            return None 
 
99
        return self._inventory[file_id].executable
 
100
 
 
101
    def has_filename(self, filename):
 
102
        return bool(self.inventory.path2id(filename))
 
103
 
 
104
    def list_files(self, include_root=False):
 
105
        # The only files returned by this are those from the version
 
106
        entries = self.inventory.iter_entries()
 
107
        # skip the root for compatability with the current apis.
 
108
        if self.inventory.root is not None and not include_root:
 
109
            # skip the root for compatability with the current apis.
 
110
            entries.next()
 
111
        for path, entry in entries:
 
112
            yield path, 'V', entry.kind, entry.file_id, entry
 
113
 
 
114
    def get_symlink_target(self, file_id):
 
115
        ie = self._inventory[file_id]
 
116
        return ie.symlink_target;
 
117
 
 
118
    def kind(self, file_id):
 
119
        return self._inventory[file_id].kind
 
120
 
 
121
    def _comparison_data(self, entry, path):
 
122
        if entry is None:
 
123
            return None, False, None
 
124
        return entry.kind, entry.executable, None
 
125
 
 
126
    def _file_size(self, entry, stat_value):
 
127
        assert entry.text_size is not None
 
128
        return entry.text_size
 
129
 
 
130
    def lock_read(self):
 
131
        self._repository.lock_read()
 
132
 
 
133
    def unlock(self):
 
134
        self._repository.unlock()
 
135
 
 
136