Merge photologue_custom into photologue

photologue/forms.py

@@ -0,0 +1,100 @@
+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 .models import Gallery, Tag
+
+
+class UploadForm(forms.Form):
+    file_field = forms.FileField(
+        label="",
+        widget=forms.FileInput(attrs={
+            'accept': 'image/*',
+            'multiple': True,
+            'class': 'mb-3',
+        }),
+    )
+    gallery = forms.ModelChoiceField(
+        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),
+                date_start=self.cleaned_data['new_gallery_date_start'],
+                date_end=self.cleaned_data['new_gallery_date_end'],
+            )
+            for tag in self.cleaned_data['new_gallery_tags']:
+                gallery.tags.add(tag)
+        return gallery