~bzr-pqm/bzr/bzr.dev

1711.3.2 by John Arbash Meinel
Add the read_bundle_from_url command, which handles lots of exceptions
1
# Copyright (C) 2006 by 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
1793.2.3 by Aaron Bentley
Rename read_bundle.py to bundle_data.py
17
from bzrlib.bundle.serializer import read_bundle
1711.3.2 by John Arbash Meinel
Add the read_bundle_from_url command, which handles lots of exceptions
18
import bzrlib.errors as errors
19
import bzrlib.urlutils
20
import bzrlib.transport
21
22
23
def read_bundle_from_url(url):
24
    """Read a bundle from a given URL.
25
26
    :return: A BundleReader, may raise NotABundle if the target 
27
            is not a proper bundle.
28
    """
29
    url, filename = bzrlib.urlutils.split(url, exclude_trailing_slash=False)
30
    if not filename:
31
        # A path to a directory was passed in
32
        # definitely not a bundle
33
        raise errors.NotABundle('A directory cannot be a bundle')
34
35
    # All of this must be in the try/except
36
    # Some transports cannot detect that we are trying to read a
37
    # directory until we actually issue read() on the handle.
38
    try:
39
        t = bzrlib.transport.get_transport(url)
40
        f = t.get(filename)
1793.2.3 by Aaron Bentley
Rename read_bundle.py to bundle_data.py
41
        return read_bundle(f)
1711.3.2 by John Arbash Meinel
Add the read_bundle_from_url command, which handles lots of exceptions
42
    except (errors.TransportError, errors.PathError), e:
43
        raise errors.NotABundle(str(e))
1793.4.1 by John Arbash Meinel
Handle the SFTP exception that you can't read() from the result of get() to a dir
44
    except (IOError,), e:
45
        # jam 20060707
46
        # Abstraction leakage, SFTPTransport.get('directory')
47
        # doesn't always fail at get() time. Sometimes it fails
48
        # during read. And that raises a generic IOError with
49
        # just the string 'Failure'
50
        # StubSFTPServer does fail during get() (because of prefetch) 
51
        # so it has an opportunity to translate the error.
52
        raise errors.NotABundle(str(e))
1711.4.21 by John Arbash Meinel
Workaround was not needed, already solved by using get_transport() and catching PathError
53