bookwyrm/bookwyrm/models/antispam.py

153 lines
4.3 KiB
Python
Raw Normal View History

2021-09-17 18:57:28 +00:00
""" Lets try NOT to sell viagra """
2022-02-24 20:15:08 +00:00
from functools import reduce
import operator
from django.apps import apps
from django.core.exceptions import PermissionDenied
from django.db import models, transaction
2022-02-24 20:15:08 +00:00
from django.db.models import Q
from django.utils.translation import gettext_lazy as _
2021-09-17 18:57:28 +00:00
from bookwyrm.tasks import app, LOW
from .base_model import BookWyrmModel
2021-09-17 18:57:28 +00:00
from .user import User
class AdminModel(BookWyrmModel):
"""Overrides the permissions methods"""
class Meta:
"""this is just here to provide default fields for other models"""
abstract = True
def raise_not_editable(self, viewer):
if viewer.has_perm("bookwyrm.moderate_user"):
return
raise PermissionDenied()
class EmailBlocklist(AdminModel):
2021-09-17 18:57:28 +00:00
"""blocked email addresses"""
domain = models.CharField(max_length=255, unique=True)
is_active = models.BooleanField(default=True)
class Meta:
"""default sorting"""
ordering = ("-created_date",)
@property
def users(self):
"""find the users associated with this address"""
return User.objects.filter(email__endswith=f"@{self.domain}")
class IPBlocklist(AdminModel):
2021-09-17 18:57:28 +00:00
"""blocked ip addresses"""
address = models.CharField(max_length=255, unique=True)
is_active = models.BooleanField(default=True)
class Meta:
"""default sorting"""
ordering = ("-created_date",)
class AutoMod(AdminModel):
"""rules to automatically flag suspicious activity"""
2022-02-24 20:15:08 +00:00
2022-02-24 19:18:43 +00:00
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)
2022-02-24 20:15:08 +00:00
@app.task(queue=LOW)
2022-02-24 20:15:08 +00:00
def automod_task():
"""Create reports"""
if not AutoMod.objects.exists():
return
reporter = AutoMod.objects.first().created_by
2022-02-24 21:20:18 +00:00
reports = automod_users(reporter) + automod_statuses(reporter)
if not reports:
return
admins = User.admins()
notification_model = apps.get_model("bookwyrm", "Notification", require_ready=True)
with transaction.atomic():
2022-02-24 21:20:18 +00:00
for admin in admins:
notification, _ = notification_model.objects.get_or_create(
user=admin, notification_type=notification_model.REPORT, read=False
2022-02-24 21:20:18 +00:00
)
notification.related_reports.set(reports)
2022-02-24 20:15:08 +00:00
def automod_users(reporter):
"""check users for moderation flags"""
user_rules = AutoMod.objects.filter(flag_users=True).values_list(
"string_match", flat=True
)
2022-02-25 01:33:22 +00:00
if not user_rules:
return []
2022-02-24 20:15:08 +00:00
2022-02-24 21:27:26 +00:00
filters = []
for field in ["username", "summary", "name"]:
filters += [{f"{field}__icontains": r} for r in user_rules]
2022-02-25 01:33:22 +00:00
users = User.objects.filter(
reduce(operator.or_, (Q(**f) for f in filters)),
is_active=True,
2022-02-25 01:42:28 +00:00
local=True,
2022-02-25 01:33:22 +00:00
report__isnull=True, # don't flag users that already have reports
).distinct()
2022-02-24 20:15:08 +00:00
report_model = apps.get_model("bookwyrm", "Report", require_ready=True)
2022-02-24 21:20:18 +00:00
return report_model.objects.bulk_create(
2022-02-24 20:15:31 +00:00
[
report_model(
reporter=reporter,
note=_("Automatically generated report"),
user=u,
)
for u in users
]
)
2022-02-24 20:15:08 +00:00
def automod_statuses(reporter):
2022-02-24 20:15:31 +00:00
"""check statues for moderation flags"""
2022-02-24 20:15:08 +00:00
status_rules = AutoMod.objects.filter(flag_statuses=True).values_list(
"string_match", flat=True
)
2022-02-25 01:33:22 +00:00
if not status_rules:
return []
2022-02-24 21:27:26 +00:00
filters = []
for field in ["content", "content_warning", "quotation__quote", "review__name"]:
filters += [{f"{field}__icontains": r} for r in status_rules]
2022-02-24 20:15:08 +00:00
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,
2022-02-25 01:42:28 +00:00
local=True,
2022-02-24 20:15:31 +00:00
report__isnull=True, # don't flag statuses that already have reports
2022-02-24 21:27:26 +00:00
).distinct()
2022-02-24 20:15:08 +00:00
report_model = apps.get_model("bookwyrm", "Report", require_ready=True)
2022-02-24 21:20:18 +00:00
return report_model.objects.bulk_create(
2022-02-24 20:15:31 +00:00
[
report_model(
reporter=reporter,
note=_("Automatically generated report"),
user=s.user,
2022-02-24 20:48:52 +00:00
status=s,
2022-02-24 20:15:31 +00:00
)
for s in statuses
]
)