Source code for scipost.widgets

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"

import json

from django.forms.widgets import CheckboxSelectMultiple, Widget
from django.utils.safestring import mark_safe


class SelectButtonWidget(CheckboxSelectMultiple):
    template_name = 'widgets/checkbox_as_btn.html'


class ReCaptcha(Widget):
    recaptcha_response_name = 'g-recaptcha-response'
    recaptcha_challenge_name = 'g-recaptcha-response'
    template_name = 'widgets/nocaptcha.html'

    def __init__(self, public_key, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.public_key = public_key

    def value_from_datadict(self, data, files, name):
        return [
            data.get(self.recaptcha_challenge_name, None),
            data.get(self.recaptcha_response_name, None)
        ]

    def get_context(self, name, value, attrs):
        try:
            lang = attrs['lang']
        except KeyError:
            # Get the generic language code
            lang = 'en'

        try:
            context = super().get_context(name, value, attrs)
        except AttributeError:
            context = {
                "widget": {
                    "attrs": self.build_attrs(attrs)
                }
            }
        context.update({
            'public_key': self.public_key,
            'lang': lang,
            'options': mark_safe(json.dumps(self.attrs, indent=2)),
        })
        return context