require login to submit success stories

Author: ewdurbin

successstories/admin.py

@@ -13,7 +13,7 @@ class StoryCategoryAdmin(NameSlugAdmin):
 @admin.register(Story)
 class StoryAdmin(ContentManageableModelAdmin):
     prepopulated_fields = {'slug': ('name',)}
-    raw_id_fields = ['category']
+    raw_id_fields = ['category', 'submitted_by']
     search_fields = ['name']
 
     def get_list_filter(self, request):

successstories/migrations/0010_story_submitted_by.py

@@ -0,0 +1,21 @@
+# Generated by Django 2.2.24 on 2022-01-27 19:21
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('successstories', '0009_auto_20180705_0352'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='story',
+            name='submitted_by',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL),
+        ),
+    ]

successstories/migrations/0011_auto_20220127_1923.py

@@ -0,0 +1,20 @@
+# Generated by Django 2.2.24 on 2022-01-27 19:23
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('successstories', '0010_story_submitted_by'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='story',
+            name='submitted_by',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL),
+        ),
+    ]

successstories/models.py

@@ -58,6 +58,8 @@ class Story(NameSlugModel, ContentManageable):
     featured = models.BooleanField(default=False, help_text="Set to use story in the supernav")
     image = models.ImageField(upload_to='successstories', blank=True, null=True)
 
+    submitted_by = models.ForeignKey(settings.AUTH_USER_MODEL, blank=True, null=True, on_delete=models.SET_NULL)
+
     objects = StoryManager()
 
     class Meta:

successstories/views.py

@@ -1,4 +1,5 @@
 from django.contrib import messages
+from django.contrib.auth.mixins import LoginRequiredMixin
 from django.urls import reverse
 from django.utils.decorators import method_decorator
 from django.views.generic import CreateView, DetailView, ListView
@@ -17,7 +18,7 @@ def get_context_data(self, **kwargs):
         return context
 
 
-class StoryCreate(ContextMixin, CreateView):
+class StoryCreate(LoginRequiredMixin, ContextMixin, CreateView):
     model = Story
     form_class = StoryForm
     template_name = 'successstories/story_form.html'
@@ -34,6 +35,8 @@ def get_success_url(self):
         return reverse('success_story_create')
 
     def form_valid(self, form):
+        obj = form.save(commit=False)
+        obj.submitted_by = self.request.user
         messages.add_message(self.request, messages.SUCCESS, self.success_message)
         return super().form_valid(form)