forked from expo/troggle
144 lines
6.8 KiB
Python
144 lines
6.8 KiB
Python
from django.shortcuts import render, redirect
|
|
from django.http import HttpResponse, HttpResponseRedirect, Http404, JsonResponse
|
|
|
|
from django.urls import reverse, resolve
|
|
from django.template import Context, loader
|
|
import re, io
|
|
from PIL import Image
|
|
from pathlib import Path
|
|
import django.forms as forms
|
|
import troggle.settings as settings
|
|
|
|
from django.views.decorators.csrf import ensure_csrf_cookie
|
|
from .auth import login_required_if_public
|
|
|
|
from troggle.core.utils import write_and_commit, WriteAndCommitError
|
|
|
|
MAX_IMAGE_WIDTH = 1000
|
|
MAX_IMAGE_HEIGHT = 800
|
|
|
|
THUMBNAIL_WIDTH = 200
|
|
THUMBNAIL_HEIGHT = 200
|
|
|
|
def get_dir(path):
|
|
"From a path sent from urls.py, determine the directory."
|
|
if "/" in path:
|
|
return path.rsplit('/', 1)[0]
|
|
else:
|
|
return ""
|
|
|
|
def image_selector(request, path):
|
|
'''Returns available images'''
|
|
directory = get_dir(path)
|
|
thumbnailspath = Path(settings.EXPOWEB) / directory / "t"
|
|
thumbnails = []
|
|
if thumbnailspath.is_dir():
|
|
for f in thumbnailspath.iterdir():
|
|
if f.is_file():
|
|
if directory:
|
|
base = f"{directory}/"
|
|
else:
|
|
base = ""
|
|
thumbnail_url = reverse('expopage', args=["%st/%s" % (base, f.name)])
|
|
name_base = f.name.rsplit('.', 1)[0]
|
|
page_path_base = Path(settings.EXPOWEB) / directory / "l"
|
|
if ((page_path_base / ("%s.htm" % name_base)).is_file()):
|
|
page_url = reverse('expopage', args=["%sl/%s.htm" % (base, name_base)])
|
|
else:
|
|
page_url = reverse('expopage', args=["%s/l/%s.html" % (base, name_base)])
|
|
|
|
thumbnails.append({"thumbnail_url": thumbnail_url, "page_url": page_url})
|
|
|
|
return render(request, 'image_selector.html', {'thumbnails': thumbnails})
|
|
|
|
@login_required_if_public
|
|
@ensure_csrf_cookie
|
|
def new_image_form(request, path):
|
|
'''Manages a form to upload new images'''
|
|
directory = get_dir(path)
|
|
if request.method == 'POST':
|
|
form = NewWebImageForm(request.POST, request.FILES, directory = directory)
|
|
if form.is_valid():
|
|
f = request.FILES["file_"]
|
|
binary_data = io.BytesIO()
|
|
for chunk in f.chunks():
|
|
binary_data.write(chunk)
|
|
i = Image.open(binary_data)
|
|
width, height = i.size
|
|
if width > MAX_IMAGE_WIDTH or height > MAX_IMAGE_HEIGHT:
|
|
scale = max(width / MAX_IMAGE_WIDTH, height / MAX_IMAGE_HEIGHT)
|
|
i = i.resize((int(width / scale), int(height / scale)), Image.ANTIALIAS)
|
|
tscale = max(width / THUMBNAIL_WIDTH, height / THUMBNAIL_HEIGHT)
|
|
thumbnail = i.resize((int(width / tscale), int(height / tscale)), Image.ANTIALIAS)
|
|
ib = io.BytesIO()
|
|
i.save(ib, format="png")
|
|
tb = io.BytesIO()
|
|
thumbnail.save(tb, format="png")
|
|
image_rel_path, thumb_rel_path, desc_rel_path = form.get_rel_paths()
|
|
image_page_template = loader.get_template('image_page_template.html')
|
|
image_page = image_page_template.render({'header': form.cleaned_data["header"], 'description': form.cleaned_data["description"],
|
|
'photographer': form.cleaned_data["photographer"], 'year': form.cleaned_data["year"],
|
|
'filepath': f'/{image_rel_path}'
|
|
})
|
|
image_path, thumb_path, desc_path = form.get_full_paths()
|
|
# Create directories if required
|
|
for full_path in image_path, thumb_path, desc_path:
|
|
print(full_path, full_path.parent)
|
|
full_path.parent.mkdir(parents=False, exist_ok=True)
|
|
try:
|
|
change_message = form.cleaned_data["change_message"]
|
|
write_and_commit([(desc_path, image_page, "utf-8"),
|
|
(image_path, ib.getbuffer(), False),
|
|
(thumb_path, tb.getbuffer(), False)],
|
|
f'{change_message} - online adding of an image')
|
|
except WriteAndCommitError as e:
|
|
return JsonResponse({"error": e.message})
|
|
linked_image_template = loader.get_template('linked_image_template.html')
|
|
html_snippet = linked_image_template.render({'thumbnail_url': f'/{thumb_rel_path}', 'page_url': f'/{desc_rel_path}'}, request)
|
|
return JsonResponse({"html": html_snippet})
|
|
else:
|
|
form = NewWebImageForm(directory = directory)
|
|
template = loader.get_template('new_image_form.html')
|
|
htmlform = template.render({'form': form, 'path': path}, request)
|
|
return JsonResponse({"form": htmlform})
|
|
|
|
class NewWebImageForm(forms.Form):
|
|
'''The form used by the editexpopage function
|
|
'''
|
|
header = forms.CharField(widget=forms.TextInput(attrs={'size':'60', 'placeholder': "Enter title (displayed as a header and in the tab)"}))
|
|
file_ = forms.FileField()
|
|
description = forms.CharField(widget=forms.Textarea(attrs={"cols":80, "rows":20, 'placeholder': "Describe the photo (using HTML)"}))
|
|
photographer = forms.CharField(widget=forms.TextInput(attrs={'size':'60', 'placeholder': "Photographers name"}), required = False)
|
|
year = forms.CharField(widget=forms.TextInput(attrs={'size':'60', 'placeholder': "Year photo was taken"}), required = False)
|
|
change_message = forms.CharField(widget=forms.Textarea(attrs={"cols":80, "rows":3, 'placeholder': "Descibe the change made (for git)"}))
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.directory = Path(kwargs.pop('directory'))
|
|
super(forms.Form, self).__init__(*args, **kwargs)
|
|
|
|
def get_rel_paths(self):
|
|
f = self.cleaned_data['file_']
|
|
return [self.directory / "i" / (f.name.rsplit('.', 1)[0] + ".png"),
|
|
self.directory / "t" / (f.name.rsplit('.', 1)[0] + ".png"),
|
|
self.directory / "l" / (f.name.rsplit('.', 1)[0] + ".html")]
|
|
|
|
def get_full_paths(self):
|
|
return [Path(settings.EXPOWEB) / x for x in self.get_rel_paths()]
|
|
|
|
def clean_file_(self):
|
|
for rel_path, full_path in zip(self.get_rel_paths(), self.get_full_paths()):
|
|
if full_path.exists():
|
|
raise forms.ValidationError("File already exists in %s" % rel_path)
|
|
return self.cleaned_data['file_']
|
|
|
|
class HTMLarea(forms.Textarea):
|
|
template_name = "widgets/HTMLarea.html"
|
|
def __init__(self, *args, **kwargs):
|
|
self.preview = kwargs.pop('preview', False)
|
|
super(forms.Textarea, self).__init__(*args, **kwargs)
|
|
def get_context(self, name, value, attrs):
|
|
c = super(forms.Textarea, self).get_context(name, value, attrs)
|
|
c["preview"] = self.preview
|
|
return c
|
|
|