Fix attachment upload paths for DreamMedia

This commit is contained in:
Kumi 2021-01-06 13:55:32 +01:00
parent 5ea3562dd1
commit 3e6ef52be9

View file

@ -9,6 +9,7 @@ from django.http import HttpResponseRedirect
from .models import Dream, DreamTheme, DreamMedia, Theme from .models import Dream, DreamTheme, DreamMedia, Theme
from .forms import DreamForm from .forms import DreamForm
from common.helpers import get_upload_path
from msgio.models import NotificationDailySchedule, Notification from msgio.models import NotificationDailySchedule, Notification
class DreamListView(LoginRequiredMixin, ListView): class DreamListView(LoginRequiredMixin, ListView):
@ -63,7 +64,9 @@ class DreamCreateView(LoginRequiredMixin, CreateView):
DreamTheme.objects.create(theme=theme, dream=form.instance) DreamTheme.objects.create(theme=theme, dream=form.instance)
for attachment in form.cleaned_data["uploads"]: for attachment in form.cleaned_data["uploads"]:
DreamMedia.objects.create(dream=form.instance, media=attachment) dba = DreamMedia(dream=form.instance)
dba.media.save(get_upload_path(form.instance, dba.media.name), attachment)
dba.save()
return ret return ret
@ -97,8 +100,10 @@ class DreamEditView(LoginRequiredMixin, UpdateView):
if not dreamtheme.theme in form.cleaned_data["themes"]: if not dreamtheme.theme in form.cleaned_data["themes"]:
dreamtheme.delete() dreamtheme.delete()
for upload in form.cleaned_data["uploads"]: for attachment in form.cleaned_data["uploads"]:
DreamMedia.objects.create(dream=form.instance, media=upload) dba = DreamMedia(dream=form.instance)
dba.media.save(get_upload_path(form.instance, attachment.name), attachment)
dba.save()
return super().form_valid(form) return super().form_valid(form)