From e837da37db79a3f5957e9a29d39784bfd37ba6d4 Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Thu, 24 Feb 2022 12:15:08 -0800
Subject: [PATCH] Adds task

---
 bookwyrm/models/__init__.py |  2 +-
 bookwyrm/models/antispam.py | 69 ++++++++++++++++++++++++++++++++++++-
 2 files changed, 69 insertions(+), 2 deletions(-)

diff --git a/bookwyrm/models/__init__.py b/bookwyrm/models/__init__.py
index 1dc6825e..440d18d9 100644
--- a/bookwyrm/models/__init__.py
+++ b/bookwyrm/models/__init__.py
@@ -29,7 +29,7 @@ from .import_job import ImportJob, ImportItem
 from .site import SiteSettings, SiteInvite
 from .site import PasswordReset, InviteRequest
 from .announcement import Announcement
-from .antispam import EmailBlocklist, IPBlocklist, AutoMod
+from .antispam import EmailBlocklist, IPBlocklist, AutoMod, automod_task
 
 from .notification import Notification
 
diff --git a/bookwyrm/models/antispam.py b/bookwyrm/models/antispam.py
index be1050b3..43b524f6 100644
--- a/bookwyrm/models/antispam.py
+++ b/bookwyrm/models/antispam.py
@@ -1,6 +1,13 @@
 """ Lets try NOT to sell viagra """
-from django.db import models
+from functools import reduce
+import operator
 
+from django.apps import apps
+from django.db import models
+from django.db.models import Q
+from django.utils.translation import gettext_lazy as _
+
+from bookwyrm.tasks import app
 from .user import User
 
 
@@ -37,7 +44,67 @@ class IPBlocklist(models.Model):
 
 class AutoMod(models.Model):
     """rules to automatically flag suspicious activity"""
+
     string_match = models.CharField(max_length=200, unique=True)
     flag_users = models.BooleanField(default=True)
     flag_statuses = models.BooleanField(default=True)
     created_by = models.ForeignKey("User", on_delete=models.PROTECT)
+
+
+@app.task(queue="low_priority")
+def automod_task():
+    """Create reports"""
+    if not AutoMod.objects.exists():
+        return
+    reporter = AutoMod.objects.first().created_by
+    automod_users(reporter)
+    automod_statuses(reporter)
+
+
+def automod_users(reporter):
+    """check users for moderation flags"""
+    user_rules = AutoMod.objects.filter(flag_users=True).values_list(
+        "string_match", flat=True
+    )
+
+    filters = [{"username__icontains": r} for r in user_rules]
+    users = User.objects.filter(
+        reduce(operator.or_, (Q(**f) for f in filters)),
+        is_active=True,
+        report__isnull=True, # don't flag users that already have reports
+    ).values_list("id", flat=True)
+
+    report_model = apps.get_model("bookwyrm", "Report", require_ready=True)
+
+    report_model.objects.bulk_create([
+        report_model(
+            reporter=reporter,
+            note=_("Automatically generated report"),
+            user=u,
+        ) for u in users
+    ])
+
+
+def automod_statuses(reporter):
+    """ check statues for moderation flags """
+    status_rules = AutoMod.objects.filter(flag_statuses=True).values_list(
+        "string_match", flat=True
+    )
+
+    filters = [{"content__icontains": r} for r in status_rules]
+    status_model = apps.get_model("bookwyrm", "Status", require_ready=True)
+    statuses = status_model.objects.filter(
+        reduce(operator.or_, (Q(**f) for f in filters)),
+        deleted=False,
+        report__isnull=True, # don't flag statuses that already have reports
+    )
+
+    report_model = apps.get_model("bookwyrm", "Report", require_ready=True)
+    report_model.objects.bulk_create([
+        report_model(
+            reporter=reporter,
+            note=_("Automatically generated report"),
+            user=s.user,
+            statuses=[s.id],
+        ) for s in statuses
+    ])