from django import forms
from django.forms import ALL_FIELDS

from .models import Referral, ReferralStats


class ReferralAdminForm(forms.ModelForm):
    class Meta:
        model = Referral
        fields = ('code',)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.disable_fields(self.fields)

    def disable_fields(self, fields):
        for field_name, field in fields.items():
            field.disabled = True


class ReferralStatsAdminForm(forms.ModelForm):
    class Meta:
        model = ReferralStats
        fields = ALL_FIELDS

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.disable_fields(self.fields)

    def disable_fields(self, fields):
        for field_name, field in fields.items():
            field.disabled = True