"""passbook reputation request policy""" from django.db import models from django.utils.translation import gettext as _ from passbook.core.models import Policy, User from passbook.lib.utils.http import get_client_ip from passbook.policies.struct import PolicyRequest, PolicyResult class ReputationPolicy(Policy): """Return true if request IP/target username's score is below a certain threshold""" check_ip = models.BooleanField(default=True) check_username = models.BooleanField(default=True) threshold = models.IntegerField(default=-5) form = "passbook.policies.reputation.forms.ReputationPolicyForm" def passes(self, request: PolicyRequest) -> PolicyResult: remote_ip = get_client_ip(request.http_request) passing = True if self.check_ip: ip_scores = IPReputation.objects.filter( ip=remote_ip, score__lte=self.threshold ) passing = passing and ip_scores.exists() if self.check_username: user_scores = UserReputation.objects.filter( user=request.user, score__lte=self.threshold ) passing = passing and user_scores.exists() return PolicyResult(passing) class Meta: verbose_name = _("Reputation Policy") verbose_name_plural = _("Reputation Policies") class IPReputation(models.Model): """Store score coming from the same IP""" ip = models.GenericIPAddressField(unique=True) score = models.IntegerField(default=0) updated = models.DateTimeField(auto_now=True) def __str__(self): return f"IPReputation for {self.ip} @ {self.score}" class UserReputation(models.Model): """Store score attempting to log in as the same username""" user = models.OneToOneField(User, on_delete=models.CASCADE) score = models.IntegerField(default=0) updated = models.DateTimeField(auto_now=True) def __str__(self): return f"UserReputation for {self.user} @ {self.score}"