110 lines
3.7 KiB
Python
110 lines
3.7 KiB
Python
import datetime
|
|
|
|
from crispy_forms.helper import FormHelper
|
|
from crispy_forms.layout import Div, Layout, Submit
|
|
from django import forms
|
|
from django.utils.text import slugify
|
|
from django.utils.translation import gettext_lazy as _
|
|
from photologue.models import Gallery
|
|
from taggit.models import Tag
|
|
|
|
from .models import GalleryExtended
|
|
|
|
|
|
class GalleryChoiceField(forms.ModelChoiceField):
|
|
def label_from_instance(self, obj):
|
|
"""Show gallery event date."""
|
|
if hasattr(obj, 'extended'):
|
|
return f"{ obj.title } ({obj.extended.date_start})"
|
|
else:
|
|
return obj.title
|
|
|
|
|
|
class UploadForm(forms.Form):
|
|
file_field = forms.FileField(
|
|
label="",
|
|
widget=forms.FileInput(attrs={
|
|
'accept': 'image/*',
|
|
'multiple': True,
|
|
'class': 'mb-3',
|
|
}),
|
|
)
|
|
gallery = GalleryChoiceField(
|
|
Gallery.objects.all(),
|
|
label=_('Gallery'),
|
|
required=False,
|
|
empty_label=_('-- Create a new gallery --'),
|
|
help_text=_('Select a gallery to add these images to. Leave this empty to '
|
|
'create a new gallery from the supplied title.')
|
|
)
|
|
new_gallery_title = forms.CharField(
|
|
label=_('New gallery title'),
|
|
max_length=250,
|
|
required=False,
|
|
)
|
|
new_gallery_date_start = forms.DateField(
|
|
label=_('New gallery event start date'),
|
|
initial=datetime.date.today,
|
|
required=False,
|
|
)
|
|
new_gallery_date_end = forms.DateField(
|
|
label=_('New gallery event end date'),
|
|
initial=datetime.date.today,
|
|
required=False,
|
|
)
|
|
new_gallery_tags = forms.ModelMultipleChoiceField(
|
|
Tag.objects.all(),
|
|
label=_('New gallery tags'),
|
|
required=False,
|
|
help_text=_('Hold down "Control", or "Command" on a Mac, to select more than one.')
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.helper = FormHelper()
|
|
self.helper.use_custom_control = False
|
|
self.helper.layout = Layout(
|
|
'file_field',
|
|
'gallery',
|
|
'new_gallery_title',
|
|
Div(
|
|
Div('new_gallery_date_start', css_class='col'),
|
|
Div('new_gallery_date_end', css_class='col'),
|
|
css_class='row'
|
|
),
|
|
'new_gallery_tags',
|
|
Submit('submit', _('Upload'), css_class='btn btn-success mt-2')
|
|
)
|
|
|
|
def clean_new_gallery_title(self):
|
|
title = self.cleaned_data['new_gallery_title']
|
|
if title and Gallery.objects.filter(title=title).exists():
|
|
raise forms.ValidationError(_('A gallery with that title already exists.'))
|
|
return title
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
|
|
# Check that either an existing gallery is chosen, or new_gallery_title is filled
|
|
if not (bool(cleaned_data['gallery']) ^ bool(cleaned_data.get('new_gallery_title', None))):
|
|
raise forms.ValidationError(
|
|
_('Select an existing gallery, or enter a title for a new gallery.'))
|
|
|
|
return cleaned_data
|
|
|
|
def get_or_create_gallery(self):
|
|
"""
|
|
Get or create gallery
|
|
"""
|
|
gallery = self.cleaned_data['gallery']
|
|
if not gallery:
|
|
# Create new gallery
|
|
title = self.cleaned_data.get('new_gallery_title')
|
|
gallery = Gallery.objects.create(title=title, slug=slugify(title))
|
|
GalleryExtended.objects.create(
|
|
gallery=gallery,
|
|
tags=self.cleaned_data['new_gallery_tags'],
|
|
date_start=self.cleaned_data['new_gallery_date_start'],
|
|
date_end=self.cleaned_data['new_gallery_date_end'],
|
|
)
|
|
return gallery
|