mirror of
https://github.com/ae-utbm/sith.git
synced 2026-11-05 00:39:32 +00:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c81c679f74 | ||
|
|
37786d2475 | ||
|
|
e980c4d727 | ||
|
|
4f989d82bb | ||
|
|
7dddb36254 | ||
|
|
85c5f7b8dd |
No files matched your search
+2
-2
@@ -2,7 +2,7 @@ from django import forms
|
||||
from django.core.validators import EmailValidator
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
|
||||
from antispam.models import ToxicDomain
|
||||
from antispam.models.models import ToxicDomain
|
||||
|
||||
|
||||
class AntiSpamEmailValidator(EmailValidator):
|
||||
@@ -17,6 +17,6 @@ validate_antispam_email = AntiSpamEmailValidator()
|
||||
|
||||
|
||||
class AntiSpamEmailField(forms.EmailField):
|
||||
"""An email field that email addresses with a known toxic domain."""
|
||||
"""An email field that forbids email addresses with a known toxic domain."""
|
||||
|
||||
default_validators = [validate_antispam_email]
|
||||
@@ -38,22 +38,31 @@ class Command(BaseCommand):
|
||||
return domains
|
||||
|
||||
def _update_domains(self, domains: set[str]):
|
||||
# Cleanup database
|
||||
ToxicDomain.objects.filter(is_externally_managed=True).delete()
|
||||
|
||||
# Create database
|
||||
ToxicDomain.objects.bulk_create(
|
||||
[
|
||||
ToxicDomain(domain=domain, is_externally_managed=True)
|
||||
for domain in domains
|
||||
],
|
||||
ignore_conflicts=True,
|
||||
)
|
||||
existing = set(ToxicDomain.objects.values_list("domain", flat=True))
|
||||
if existing == domains:
|
||||
self.stdout.write("Domain database was up to date")
|
||||
return
|
||||
if to_remove := (existing - domains):
|
||||
# Cleanup database
|
||||
ToxicDomain.objects.filter(
|
||||
domain__in=to_remove, is_externally_managed=True
|
||||
).delete()
|
||||
if to_create := (domains - existing):
|
||||
# Create database
|
||||
ToxicDomain.objects.bulk_create(
|
||||
[
|
||||
ToxicDomain(domain=domain, is_externally_managed=True)
|
||||
for domain in to_create
|
||||
],
|
||||
ignore_conflicts=True,
|
||||
)
|
||||
self.stdout.write("Domain database updated")
|
||||
|
||||
def handle(self, *args, **options):
|
||||
if not self._should_update(force=options["force"]):
|
||||
self.stdout.write("Domain database is up to date")
|
||||
self.stdout.write(
|
||||
"No update required. Use --force if you want to do it anyway."
|
||||
)
|
||||
return
|
||||
self.stdout.write("Updating domain database")
|
||||
|
||||
|
||||
@@ -0,0 +1,4 @@
|
||||
from antispam.models.fields import AntiSpamEmailField
|
||||
from antispam.models.models import ToxicDomain
|
||||
|
||||
__all__ = ["AntiSpamEmailField", "ToxicDomain"]
|
||||
@@ -0,0 +1,21 @@
|
||||
from django.conf import settings
|
||||
from django.db import models
|
||||
|
||||
from antispam import forms
|
||||
|
||||
|
||||
class AntiSpamEmailField(models.EmailField):
|
||||
"""An email field that forbids email addresses with a known toxic domain."""
|
||||
|
||||
default_validators = [forms.validate_antispam_email]
|
||||
|
||||
|
||||
if settings.TESTING:
|
||||
from model_bakery import baker
|
||||
|
||||
baker.generators.add(
|
||||
AntiSpamEmailField, lambda: baker.random_gen.gen_email(max_length=100)
|
||||
)
|
||||
else: # pragma: no cover
|
||||
# baker is only used in tests, so we don't need coverage for this part
|
||||
pass
|
||||
File renamed without changes.
Whitespace-only changes.
+3
-2
@@ -55,6 +55,7 @@ from django.utils.translation import gettext_lazy as _
|
||||
from phonenumber_field.modelfields import PhoneNumberField
|
||||
from PIL import Image, ImageOps
|
||||
|
||||
from antispam.models import AntiSpamEmailField
|
||||
from core.utils import get_last_promo
|
||||
|
||||
if TYPE_CHECKING:
|
||||
@@ -158,7 +159,7 @@ class User(AbstractUser):
|
||||
|
||||
first_name = models.CharField(_("first name"), max_length=64)
|
||||
last_name = models.CharField(_("last name"), max_length=64)
|
||||
email = models.EmailField(_("email address"), unique=True)
|
||||
email = AntiSpamEmailField(_("email address"), unique=True)
|
||||
date_of_birth = models.DateField(_("date of birth"), blank=True, null=True)
|
||||
nick_name = models.CharField(_("nick name"), max_length=64, null=True, blank=True)
|
||||
last_update = models.DateTimeField(_("last update"), auto_now=True)
|
||||
@@ -267,7 +268,7 @@ class User(AbstractUser):
|
||||
forum_signature = models.TextField(
|
||||
_("forum signature"), max_length=256, blank=True, default=""
|
||||
)
|
||||
second_email = models.EmailField(_("second email address"), null=True, blank=True)
|
||||
second_email = AntiSpamEmailField(_("second email address"), null=True, blank=True)
|
||||
phone = PhoneNumberField(_("phone"), null=True, blank=True)
|
||||
parent_phone = PhoneNumberField(_("parent phone"), null=True, blank=True)
|
||||
address = models.CharField(_("address"), max_length=128, blank=True, default="")
|
||||
|
||||
@@ -472,6 +472,36 @@ def test_user_preferences(client: Client):
|
||||
assert user.preferences.notify_on_click is True
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
@pytest.mark.parametrize("can_edit", [True, False])
|
||||
def test_user_visibility(client: Client, monkeypatch, can_edit):
|
||||
# The view is a CanView, so just patch User.can_be_edited_by for the test
|
||||
monkeypatch.setattr("core.models.User.can_be_edited_by", lambda *args: can_edit)
|
||||
client.force_login(subscriber_user.make())
|
||||
|
||||
user, other_user = subscriber_user.make(_quantity=2)
|
||||
url = reverse("core:user_visibility_fragment", kwargs={"user_id": user.id})
|
||||
response = client.get(url)
|
||||
assert response.status_code == 200 if can_edit else 403
|
||||
response = client.post(
|
||||
url,
|
||||
{
|
||||
"show_my_stats": False,
|
||||
"is_viewable": False,
|
||||
"whitelisted_users": [other_user.id],
|
||||
},
|
||||
)
|
||||
if not can_edit:
|
||||
assert response.status_code == 403
|
||||
return
|
||||
assertRedirects(response, url)
|
||||
user.refresh_from_db()
|
||||
user.preferences.refresh_from_db()
|
||||
assert user.is_viewable is False
|
||||
assert list(user.whitelisted_users.all()) == [other_user]
|
||||
assert user.preferences.show_my_stats is False
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_user_stats(client: Client):
|
||||
user = subscriber_user.make()
|
||||
|
||||
+1
-5
@@ -48,7 +48,6 @@ from django.utils.translation import gettext_lazy as _
|
||||
from phonenumber_field.widgets import RegionalPhoneNumberWidget
|
||||
from PIL import Image
|
||||
|
||||
from antispam.forms import AntiSpamEmailField
|
||||
from core.models import Gift, Group, Page, PageRev, Preferences, SithFile, User
|
||||
from core.utils import resize_image
|
||||
from core.views.widgets.ajax_select import (
|
||||
@@ -185,10 +184,7 @@ class RegisteringForm(UserCreationForm):
|
||||
class Meta:
|
||||
model = User
|
||||
fields = ("first_name", "last_name", "email", "cgu_approved_at")
|
||||
field_classes = {
|
||||
"email": AntiSpamEmailField,
|
||||
"cgu_approved_at": CGUApprovalField,
|
||||
}
|
||||
field_classes = {"cgu_approved_at": CGUApprovalField}
|
||||
|
||||
|
||||
class CGUApprovalForm(forms.ModelForm):
|
||||
|
||||
@@ -94,8 +94,8 @@ class FragmentMixin(TemplateResponseMixin, AllowFragment, ContextMixin):
|
||||
return render(
|
||||
request,
|
||||
"app/template.jinja",
|
||||
context={"fragment": fragment(request)
|
||||
}
|
||||
context={"fragment": fragment(request)}
|
||||
)
|
||||
|
||||
# in urls.py
|
||||
urlpatterns = [
|
||||
|
||||
+3
-1
@@ -502,7 +502,9 @@ class UserClubView(UserTabsMixin, CanViewMixin, DetailView):
|
||||
current_tab = "clubs"
|
||||
|
||||
|
||||
class UserVisibilityFormFragment(FragmentMixin, SuccessMessageMixin, UpdateView):
|
||||
class UserVisibilityFormFragment(
|
||||
FragmentMixin, SuccessMessageMixin, CanEditMixin, UpdateView
|
||||
):
|
||||
model = User
|
||||
form_class = UserVisibilityForm
|
||||
template_name = "core/fragment/user_visibility.jinja"
|
||||
|
||||
+1
-1
@@ -73,7 +73,7 @@ dev = [
|
||||
]
|
||||
tests = [
|
||||
"freezegun>=1.5.5,<2.0.0",
|
||||
"pytest>=9.1.1,<10.0.0",
|
||||
"pytest>=9.2.0,<10.0.0",
|
||||
"pytest-cov>=7.1.0,<8.0.0",
|
||||
"pytest-django>=4.14.0,<5.0.0",
|
||||
"model-bakery>=1.24.2,<2.0.0",
|
||||
|
||||
Reference in new issue
Block a user