blob: 61cf8d58b6a365aa2b686c6dd334d2e61bedeeec (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
import os
import contextlib
try:
from urllib2 import urlopen
except ImportError:
from urllib.request import urlopen
try:
from urllib.parse import urlparse
except ImportError:
from urlparse import urlparse
class Files(object):
def __init__(self, base):
self._base = base
def open(self, uri):
try:
if _is_absolute(uri):
return contextlib.closing(urlopen(uri))
elif self._base is not None:
return open(os.path.join(self._base, uri), "rb")
else:
raise InvalidFileReferenceError("could not find external image '{0}', fileobj has no name".format(uri))
except IOError as error:
message = "could not open external image: '{0}' (document directory: '{1}')\n{2}".format(
uri, self._base, str(error))
raise InvalidFileReferenceError(message)
def _is_absolute(url):
return urlparse(url).scheme != ""
class InvalidFileReferenceError(ValueError):
pass
|
