from django import forms
from django.core.exceptions import ValidationError
from django.template.loader import render_to_string

from postgresqleu.confsponsor.backendforms import BackendSponsorshipLevelBenefitForm
from postgresqleu.confsponsor.models import SponsorScanner, ScannedAttendee

from .base import BaseBenefit, BaseBenefitForm


class BadgeScanningForm(BaseBenefitForm):
    confirm = forms.ChoiceField(label="Claim benefit", choices=((0, '* Choose'), (1, 'Claim this benefit'), (2, 'Decline this benefit')))

    def clean_confirm(self):
        if not int(self.cleaned_data['confirm']) in (1, 2):
            raise ValidationError('You must decide if you want to claim this benefit')
        return self.cleaned_data['confirm']


class BadgeScanning(BaseBenefit):
    can_multiclaim = False

    @classmethod
    def get_backend_form(self):
        return BackendSponsorshipLevelBenefitForm

    def generate_form(self):
        return BadgeScanningForm

    def save_form(self, form, claim, request):
        if int(form.cleaned_data['confirm']) == 2:
            return False
        return True

    def can_unclaim(self, claimedbenefit):
        if ScannedAttendee.objects.filter(sponsor=claimedbenefit.sponsor).exists():
            # If the sponsor has scanned something, we don't allow unclaiming anymore
            return False
        return True

    def render_claimdata(self, claimedbenefit, isadmin):
        if claimedbenefit.declined:
            return ""
        if claimedbenefit.confirmed:
            return 'To manage your badge scanning and results, please click <a href="/events/sponsor/{0}/scanning/">here</a>.'.format(
                claimedbenefit.sponsor.id,
            )
        else:
            return ""

    def render_reportinfo(self, claimedbenefit):
        if claimedbenefit.declined or not claimedbenefit.confirmed:
            return ''

        return '{} badges scanned'.format(ScannedAttendee.objects.filter(sponsor=claimedbenefit.sponsor).count())

    def inject_summary_section(self, claimedbenefit):
        if claimedbenefit.declined or not claimedbenefit.confirmed:
            return None

        scanners = [s.scanner.fullname for s in SponsorScanner.objects.select_related('scanner').filter(sponsor=claimedbenefit.sponsor)]
        return (
            "Badge Scanning",
            render_to_string('confsponsor/section_badgescanning.html', {
                'sponsor': claimedbenefit.sponsor,
                'scanners': scanners,
                'scancount': ScannedAttendee.objects.filter(sponsor=claimedbenefit.sponsor).count(),
            })
        )