Added support for Django REST Framework.

Added serializer, modelserializer and validator

Author: michalwrona01

captcha/serializers.py

@@ -0,0 +1,32 @@
+from rest_framework import serializers
+from rest_framework.fields import empty
+
+from captcha.validators import captcha_validate
+
+
+class CaptchaSerializer(serializers.Serializer):
+    """Serializer captcha code and captcha hashkey"""
+
+    captcha_code = serializers.CharField(max_length=32, write_only=True, required=True)
+    captcha_hashkey = serializers.CharField(
+        max_length=40, write_only=True, required=True
+    )
+
+    def run_validation(self, data=empty):
+        values = super().run_validation(data=data)
+        captcha_validate(values["captcha_hashkey"], values["captcha_code"])
+        return values
+
+
+class CaptchaModelSerializer(serializers.ModelSerializer):
+    """Model serializer captcha code and captcha hashkey"""
+
+    captcha_code = serializers.CharField(max_length=32, write_only=True, required=True)
+    captcha_hashkey = serializers.CharField(
+        max_length=40, write_only=True, required=True
+    )
+
+    def run_validation(self, data=empty):
+        values = super().run_validation(data=data)
+        captcha_validate(values["captcha_hashkey"], values["captcha_code"])
+        return values

captcha/tests/tests.py

@@ -575,6 +575,53 @@ def test_empty_pool_fallback(self):
         self.assertEqual(CaptchaStore.objects.count(), 1)
         settings.CAPTCHA_GET_FROM_POOL = __current_test_get_from_pool_setting
 
+    def test_serializer(self):
+        r = self.client.post(
+            reverse("captcha-test-serializer"),
+            dict(
+                captcha_code=self.default_store.response,
+                captcha_hashkey=self.default_store.hashkey,
+            ),
+        )
+        self.assertEqual(r.status_code, 200)
+
+    def test_wrong_serializer(self):
+        r = self.client.post(
+            reverse("captcha-test-serializer"),
+            dict(
+                captcha_code="xxx",
+                captcha_hashkey="wrong hash",
+            ),
+        )
+        self.assertEqual(r.status_code, 400)
+        self.assertEqual(json.loads(r.content), {"error": "Invalid CAPTCHA"})
+
+
+    def test_model_serializer(self):
+        r = self.client.post(
+            reverse("captcha-test-model-serializer"),
+            dict(
+                captcha_code=self.default_store.response,
+                captcha_hashkey=self.default_store.hashkey,
+                subject="xxx",
+                sender="[email protected]",
+            ),
+        )
+        self.assertEqual(r.status_code, 200)
+
+    def test_wrong_model_serializer(self):
+        r = self.client.post(
+            reverse("captcha-test-model-serializer"),
+            dict(
+                captcha_code="xxx",
+                captcha_hashkey="wrong hash",
+                subject="xxx",
+                sender="[email protected]",
+            ),
+        )
+        self.assertEqual(r.status_code, 400)
+        self.assertEqual(json.loads(r.content), {"error": "Invalid CAPTCHA"})
+
 
 def trivial_challenge():
     return "trivial", "trivial"

captcha/tests/urls.py

@@ -8,12 +8,16 @@
     test_model_form,
     test_non_required,
     test_per_form_format,
+    test_serializer,
+    test_model_serializer,
 )
 
 
 urlpatterns = [
     re_path(r"test/$", test, name="captcha-test"),
     re_path(r"test-modelform/$", test_model_form, name="captcha-test-model-form"),
+    re_path(r"test-serializer/$", test_serializer, name="captcha-test-serializer"),
+    re_path(r"test-model-serializer/$", test_model_serializer, name="captcha-test-model-serializer"),
     re_path(
         r"test2/$", test_custom_error_message, name="captcha-test-custom-error-message"
     ),

captcha/tests/views.py

@@ -2,9 +2,13 @@
 from django.contrib.auth.models import User
 from django.http import HttpResponse
 from django.template import engines
+from rest_framework import status
+from rest_framework.decorators import api_view
+from rest_framework.response import Response
 
+from rest_framework import serializers
 from captcha.fields import CaptchaField
-
+from captcha.serializers import CaptchaSerializer, CaptchaModelSerializer
 
 TEST_TEMPLATE = r"""
 <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
@@ -126,3 +130,25 @@ class CaptchaTestForm(forms.Form):
         captcha2 = CaptchaField(id_prefix="form2")
 
     return _test(request, CaptchaTestForm)
+
+
+@api_view(['POST'])
+def test_serializer(request):
+    serializer = CaptchaSerializer(data=request.POST)
+    serializer.is_valid(raise_exception=True)
+    return Response(status=status.HTTP_200_OK)
+
+
+@api_view(['POST'])
+def test_model_serializer(request):
+    class UserCaptchaModelSerializer(CaptchaModelSerializer):
+        subject = serializers.CharField(max_length=100)
+        sender = serializers.EmailField()
+
+        class Meta:
+            model = User
+            fields = ("subject", "sender", "captcha_code", "captcha_hashkey")
+
+    serializer = UserCaptchaModelSerializer(data=request.POST)
+    serializer.is_valid(raise_exception=True)
+    return Response(status=status.HTTP_200_OK)

captcha/validators.py

@@ -0,0 +1,32 @@
+from rest_framework import exceptions
+
+import django
+from django.utils import timezone
+
+
+if django.VERSION >= (3, 0):
+    from django.utils.translation import gettext_lazy as gettext_lazy
+else:
+    from django.utils.translation import ugettext_lazy as gettext_lazy
+
+from captcha.conf import settings
+from captcha.models import CaptchaStore
+
+
+def captcha_validate(hashkey, response):
+    hashkey, response = hashkey.lower(), response.lower()
+
+    if not settings.CAPTCHA_GET_FROM_POOL:
+        CaptchaStore.remove_expired()
+    if settings.CAPTCHA_TEST_MODE and response.lower() == "passed":
+        try:
+            CaptchaStore.objects.get(hashkey=hashkey).delete()
+        except CaptchaStore.DoesNotExist:
+            pass
+    else:
+        try:
+            CaptchaStore.objects.get(
+                response=response, hashkey=hashkey, expiration__gt=timezone.now()
+            ).delete()
+        except CaptchaStore.DoesNotExist:
+            raise exceptions.ValidationError({"error": gettext_lazy("Invalid CAPTCHA")})

setup.py

@@ -35,6 +35,7 @@ def run_tests(self):
     "Django >= 4.2",
     "Pillow >=6.2.0",
     "django-ranged-response == 0.2.0",
+    "djangorestframework >= 3.15.0",
 ]
 EXTRAS_REQUIRE = {"test": ("testfixtures",)}
 

testproject/settings.py

@@ -29,6 +29,7 @@
     "django.contrib.sites",
     "django.forms",
     "django.contrib.messages",
+    "rest_framework",
     "captcha",
 ]
 FORM_RENDERER = "django.forms.renderers.TemplatesSetting"

tox.ini

@@ -30,6 +30,7 @@ deps =
         py{39,310,311,312}-django{42,50}: python3-memcached
         jinja2
         Pillow
+        djangorestframework
 
 extras =
         test