145 lines
5.5 KiB
Python
145 lines
5.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
import codecs
|
|
import datetime
|
|
import logging
|
|
import os
|
|
from django.apps import apps
|
|
from django.contrib import messages
|
|
from django.urls import reverse_lazy
|
|
from django.utils.translation import ugettext as _
|
|
from django.views import generic
|
|
|
|
from .forms import UploadForm
|
|
|
|
app_config = apps.get_containing_app_config(__package__)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class UploadView(generic.edit.FormView):
|
|
initial = {
|
|
# 'name': u'heinzel',
|
|
# 'email_address': 'heinzel@heinzelwelt.de',
|
|
# 'title': u'Mein Beitrag',
|
|
# 'description': 'Foobar',
|
|
# 'accepted': True,
|
|
}
|
|
template_name = 'dav_submission/upload_form.html'
|
|
form_class = UploadForm
|
|
success_url = reverse_lazy('dav_submission:success')
|
|
|
|
def _sanitize_filename(self, input):
|
|
max_length = None
|
|
discard_chars = u''
|
|
allowed_chars = (u'abcdefghijklmnopqrstuvwxyz'
|
|
u'äöüß'
|
|
u'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
|
|
u'ÄÖÜ'
|
|
u'0123456789'
|
|
u'._-')
|
|
non_allowed_substitute = u'_'
|
|
space_substitute = u'_'
|
|
|
|
if space_substitute is None:
|
|
space_substitute = non_allowed_substitute
|
|
|
|
r = ''
|
|
for c in input:
|
|
if c in discard_chars:
|
|
continue
|
|
elif allowed_chars is not None and c in allowed_chars:
|
|
r += c
|
|
elif allowed_chars is None and c.isalnum():
|
|
r += c
|
|
elif c.isspace():
|
|
r += space_substitute
|
|
else:
|
|
r += non_allowed_substitute
|
|
|
|
return r[:max_length]
|
|
|
|
def form_valid(self, form):
|
|
base_path = app_config.settings.upload_path
|
|
|
|
subdir_format_str = u'{datetime}--{title}'
|
|
now = datetime.datetime.now()
|
|
subdir_format_kwargs = {'datetime': now.strftime('%Y-%m-%d--%H%M%S'),
|
|
'date': now.strftime('%Y-%m-%d'),
|
|
'time': now.strftime('%H-%M-%S'),
|
|
'title': form.cleaned_data['title']}
|
|
subdir_name = self._sanitize_filename(
|
|
subdir_format_str.format(**subdir_format_kwargs)
|
|
)
|
|
subdir_path = os.path.join(base_path, subdir_name)
|
|
|
|
if os.path.isdir(subdir_path):
|
|
message = _(u'Es gibt bereits einen Beitrag mit dem Titel "%(title)s".') % subdir_format_kwargs
|
|
messages.error(self.request, message)
|
|
form.add_error('title', message)
|
|
return self.render_to_response(self.get_context_data(form=form))
|
|
|
|
os.makedirs(subdir_path)
|
|
|
|
try:
|
|
metadata_format_str = u"""Absender: {name} <{email_address}>
|
|
Datum: {date} {time}
|
|
Titel: {title}
|
|
Beschreibung:
|
|
{description}
|
|
"""
|
|
metadata_format_kwargs = {
|
|
'date': now.strftime('%d.%m.%Y'),
|
|
'time': now.strftime('%H:%M:%S'),
|
|
'name': form.cleaned_data['name'],
|
|
'email_address': form.cleaned_data['email_address'],
|
|
'title': form.cleaned_data['title'],
|
|
'description': form.cleaned_data['description'],
|
|
}
|
|
|
|
metadata = metadata_format_str.format(**metadata_format_kwargs)
|
|
metadata_file_name = app_config.settings.metadata_file_name
|
|
metadata_file_path = os.path.join(subdir_path, metadata_file_name)
|
|
|
|
with codecs.open(metadata_file_path, 'w', encoding='utf-8') as metadata_file:
|
|
metadata_file.write(metadata)
|
|
except Exception as e:
|
|
message = _(u'Jetzt ist irgendwas schief gelaufen.')
|
|
messages.error(self.request, message)
|
|
logger.error('dav_submission.views.UploadView.form_valid(): Catched Exception #2: %s', str(e))
|
|
return super(UploadView, self).form_valid(form)
|
|
|
|
try:
|
|
for input_file in form.files.getlist('files'):
|
|
file_name = self._sanitize_filename(input_file.name)
|
|
file_path = os.path.join(subdir_path, file_name)
|
|
|
|
if os.path.exists(file_path):
|
|
message = _(u'Die Datei %(name)s haben wir bereits.') % {'name': input_file.name}
|
|
messages.error(self.request, message)
|
|
continue
|
|
|
|
with open(file_path, 'wb+') as local_file:
|
|
for chunk in input_file.chunks():
|
|
local_file.write(chunk)
|
|
|
|
size = os.path.getsize(file_path)
|
|
if size > (1024 * 1024):
|
|
size_str = u'%s MiB' % ('%.3f' % (size / 1024.0 / 1024.0)).rstrip('0').rstrip('.')
|
|
elif size > 1024:
|
|
size_str = u'%s KiB' % ('%.3f' % (size / 1024.0)).rstrip('0').rstrip('.')
|
|
else:
|
|
size_str = u'%d Byte' % size
|
|
message = _(u'Datei erfolgreich hochgeladen: %(name)s (%(size)s)') % {'name': input_file.name,
|
|
'size': size_str}
|
|
messages.success(self.request, message)
|
|
except Exception as e:
|
|
message = _(u'Jetzt ist irgendwas schief gelaufen.')
|
|
messages.error(self.request, message)
|
|
logger.error('dav_submission.views.UploadView.form_valid(): Catched Exception #3: %s', str(e))
|
|
return super(UploadView, self).form_valid(form)
|
|
|
|
return super(UploadView, self).form_valid(form)
|
|
|
|
|
|
class SuccessView(generic.TemplateView):
|
|
template_name = 'dav_submission/success.html'
|