Schedules automod task

bookwyrm/views/__init__.py

@@ -2,7 +2,7 @@
 # site admin
 from .admin.announcements import Announcements, Announcement
 from .admin.announcements import EditAnnouncement, delete_announcement
-from .admin.automod import AutoMod, automod_delete, run_automod
+from .admin.automod import AutoMod, automod_delete, run_automod, schedule_automod_task
 from .admin.dashboard import Dashboard
 from .admin.federation import Federation, FederatedServer
 from .admin.federation import AddFederatedServer, ImportServerBlocklist

bookwyrm/views/admin/automod.py

@@ -1,10 +1,12 @@
 """ moderation via flagged posts and users """
 from django.contrib.auth.decorators import login_required, permission_required
+from django.db import transaction
 from django.shortcuts import get_object_or_404, redirect
 from django.template.response import TemplateResponse
 from django.utils.decorators import method_decorator
 from django.views import View
 from django.views.decorators.http import require_POST
+from django_celery_beat.models import PeriodicTask
 
 from bookwyrm import forms, models
 
@@ -24,8 +26,9 @@ class AutoMod(View):
 
     def get(self, request):
         """view rules"""
-        data = {"rules": models.AutoMod.objects.all(), "form": forms.AutoModRuleForm()}
-        return TemplateResponse(request, "settings/automod/rules.html", data)
+        return TemplateResponse(
+            request, "settings/automod/rules.html", automod_view_data()
+        )
 
     def post(self, request):
         """add rule"""
@@ -35,14 +38,32 @@ class AutoMod(View):
             form.save()
             form = forms.AutoModRuleForm()
 
-        data = {
-            "rules": models.AutoMod.objects.all(),
-            "form": form,
-            "success": success,
-        }
+        data = automod_view_data()
+        data["form"] = form
         return TemplateResponse(request, "settings/automod/rules.html", data)
 
 
+@require_POST
+@permission_required("bookwyrm.moderate_user", raise_exception=True)
+@permission_required("bookwyrm.moderate_post", raise_exception=True)
+def schedule_automod_task(request):
+    """scheduler"""
+    form = forms.IntervalScheduleForm(request.POST)
+    if not form.is_valid():
+        data = automod_view_data()
+        data["task_form"] = form
+        return TemplateResponse(request, "settings/automod/rules.html", data)
+
+    with transaction.atomic():
+        schedule = form.save()
+        PeriodicTask.objects.get_or_create(
+            interval=schedule,
+            name="automod-task",
+            task="bookwyrm.models.antispam.automod_task",
+        )
+    return redirect("settings-automod")
+
+
 @require_POST
 @permission_required("bookwyrm.moderate_user", raise_exception=True)
 @permission_required("bookwyrm.moderate_post", raise_exception=True)
@@ -62,3 +83,18 @@ def run_automod(request):
     """run scan"""
     models.automod_task.delay()
     return redirect("settings-automod")
+
+
+def automod_view_data():
+    """helper to get data used in the template"""
+    try:
+        task = PeriodicTask.objects.get(name="automod-task")
+    except PeriodicTask.DoesNotExist:
+        task = None
+
+    return {
+        "task": task,
+        "task_form": forms.IntervalScheduleForm(),
+        "rules": models.AutoMod.objects.all(),
+        "form": forms.AutoModRuleForm(),
+    }