blob: e26a43f0262fa4f56d35be8e8abcdfc52acf05fb (
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
37
38
39
40
41
42
43
44
45
46
|
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, external_file_access):
self._base = base
self._external_file_access = external_file_access
def open(self, uri):
if not self._external_file_access:
raise ExternalFileAccessIsDisabledError(
"could not open external image '{0}', external file access is disabled".format(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
class ExternalFileAccessIsDisabledError(InvalidFileReferenceError):
pass
|
