forked from matthiask/django-http-fallback-storage
-
Notifications
You must be signed in to change notification settings - Fork 0
/
http_fallback_storage.py
51 lines (43 loc) · 1.8 KB
/
http_fallback_storage.py
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
47
48
49
50
51
from __future__ import unicode_literals
from functools import wraps
import io
import os
import requests
from django.conf import settings
from django.core.files.storage import FileSystemStorage
from django.utils.six.moves.urllib.parse import urljoin
from django.utils.termcolors import colorize
def download_before_call(method):
@wraps(method)
def _fn(self, name, *args, **kwargs):
local = os.path.join(settings.MEDIA_ROOT, name)
if not os.path.exists(local):
remote = urljoin(settings.FALLBACK_STORAGE_URL, name)
print(colorize(
"Attempting download '%s' -> '%s'" % (remote, name),
fg='blue',
))
try:
data = requests.get(remote)
except Exception as exc:
print(colorize(exc, fg='red'))
else:
if data.status_code == 200:
dirname = os.path.dirname(local)
if not os.path.exists(dirname):
os.makedirs(os.path.dirname(local))
with io.open(local, 'wb') as f:
f.write(data.content)
return method(self, name, *args, **kwargs)
return _fn
class FallbackStorage(FileSystemStorage):
open = download_before_call(FileSystemStorage.open)
path = download_before_call(FileSystemStorage.path)
exists = download_before_call(FileSystemStorage.exists)
size = download_before_call(FileSystemStorage.size)
url = download_before_call(FileSystemStorage.url)
# open = download_before_call(FileSystemStorage.open)
# open = download_before_call(FileSystemStorage.open)
# open = download_before_call(FileSystemStorage.open)
# open = download_before_call(FileSystemStorage.open)
# open = download_before_call(FileSystemStorage.open)