blob: bde488dd5e4eac391d842f0914908145a3a87506 (
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
|
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 verify(self, uri):
with self.open(uri):
pass
def open(self, uri):
if _is_absolute(uri):
return contextlib.closing(urlopen(uri))
elif self._base is not None:
try:
return open(os.path.join(self._base, uri), "rb")
except IOError as error:
message = "could not open external image '{0}' from document directory '{1}'\n{2}".format(
uri, self._base, str(error))
raise InvalidFileReferenceError(message)
else:
raise InvalidFileReferenceError("could not find external image '{0}', fileobj has no name".format(uri))
def _is_absolute(url):
return urlparse(url).scheme != ""
class InvalidFileReferenceError(ValueError):
pass
|
