blob: ce557f6695e629566755d9810c7db2c26d559188 (
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
|
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:
path = os.path.join(self._base, uri)
if not os.path.exists(path):
raise InvalidFileReferenceError("could not find external image '{0}' from document directory '{1}'".format(uri, self._base))
return open(path, "rb")
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
|
