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

  • Committer: John Arbash Meinel
  • Date: 2011-04-20 14:27:19 UTC
  • mto: This revision was merged to the branch mainline in revision 5837.
  • Revision ID: john@arbash-meinel.com-20110420142719-advs1k5vztqzbrgv
https://launchpad.net/bugs/767177
Fix bug #767177. Be more agressive with file.close() calls.

Our test suite gets a number of thread leaks and failures because it happens to get async
SFTPFile.close() calls. (if an SFTPFile closes due to __del__ it is done as an async request,
while if you call SFTPFile.close() it is done as a synchronous request.)
We have a couple other cases, probably. Namely SFTPTransport.get() also does an async
prefetch of the content, so if you don't .read() you'll also leak threads that think they
are doing work that you want.

The biggest change here, though, is using a try/finally in a generator, which is not 
python2.4 compatible.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/testament.py
76
76
    contains_linebreaks,
77
77
    sha,
78
78
    )
 
79
from bzrlib.tree import Tree
79
80
 
80
81
 
81
82
class Testament(object):
91
92
 
92
93
    long_header = 'bazaar-ng testament version 1\n'
93
94
    short_header = 'bazaar-ng testament short form 1\n'
 
95
    include_root = False
94
96
 
95
97
    @classmethod
96
98
    def from_revision(cls, repository, revision_id):
97
 
        """Produce a new testament from a historical revision"""
 
99
        """Produce a new testament from a historical revision."""
98
100
        rev = repository.get_revision(revision_id)
99
 
        inventory = repository.get_inventory(revision_id)
100
 
        return cls(rev, inventory)
101
 
 
102
 
    def __init__(self, rev, inventory):
103
 
        """Create a new testament for rev using inventory."""
 
101
        tree = repository.revision_tree(revision_id)
 
102
        return cls(rev, tree)
 
103
 
 
104
    @classmethod
 
105
    def from_revision_tree(cls, tree):
 
106
        """Produce a new testament from a revision tree."""
 
107
        rev = tree._repository.get_revision(tree.get_revision_id())
 
108
        return cls(rev, tree)
 
109
 
 
110
    def __init__(self, rev, tree):
 
111
        """Create a new testament for rev using tree."""
104
112
        self.revision_id = rev.revision_id
105
113
        self.committer = rev.committer
106
114
        self.timezone = rev.timezone or 0
107
115
        self.timestamp = rev.timestamp
108
116
        self.message = rev.message
109
117
        self.parent_ids = rev.parent_ids[:]
110
 
        self.inventory = inventory
 
118
        if not isinstance(tree, Tree):
 
119
            raise TypeError("As of bzr 2.4 Testament.__init__() takes a "
 
120
                "Revision and a Tree.")
 
121
        self.tree = tree
111
122
        self.revprops = copy(rev.properties)
112
123
        if contains_whitespace(self.revision_id):
113
124
            raise ValueError(self.revision_id)
143
154
        return [line.encode('utf-8') for line in r]
144
155
 
145
156
    def _get_entries(self):
146
 
        entries = self.inventory.iter_entries()
147
 
        entries.next()
148
 
        return entries
 
157
        return ((path, ie) for (path, versioned, kind, file_id, ie) in
 
158
                self.tree.list_files(include_root=self.include_root))
149
159
 
150
160
    def _escape_path(self, path):
151
161
        if contains_linebreaks(path):
209
219
 
210
220
    long_header = 'bazaar-ng testament version 2.1\n'
211
221
    short_header = 'bazaar-ng testament short form 2.1\n'
 
222
    include_root = False
212
223
    def _entry_to_line(self, path, ie):
213
224
        l = Testament._entry_to_line(self, path, ie)[:-1]
214
225
        l += ' ' + ie.revision
224
235
 
225
236
    long_header = 'bazaar testament version 3 strict\n'
226
237
    short_header = 'bazaar testament short form 3 strict\n'
227
 
    def _get_entries(self):
228
 
        return self.inventory.iter_entries()
 
238
    include_root = True
229
239
 
230
240
    def _escape_path(self, path):
231
241
        if contains_linebreaks(path):