IdHub/idhub/user/forms.py

from django import forms
from idhub_auth.models import User
from idhub.models import DID, VerificableCredential, Organization


ORGANIZATION = [
    (x.id, x.name) for x in Organization.objects.filter()
]


class ProfileForm(forms.ModelForm):
    MANDATORY_FIELDS = ['first_name', 'last_name', 'email']

    class Meta:
        model = User
        fields = ('first_name', 'last_name', 'email')


class RequestCredentialForm(forms.Form):
    did = forms.ChoiceField(choices=[])
    credential = forms.ChoiceField(choices=[])

    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop('user', None)
        super().__init__(*args, **kwargs)
        self.fields['did'].choices = [
            (x.did, x.label) for x in DID.objects.filter(user=self.user)
        ]
        self.fields['credential'].choices = [
            (x.id, x.type()) for x in VerificableCredential.objects.filter(
                user=self.user,
                status=VerificableCredential.Status.ENABLED
            )
        ]

    def save(self, commit=True):
        did = DID.objects.filter(
            user=self.user,
            did=self.data['did']
        )
        cred = VerificableCredential.objects.filter(
            user=self.user,
            id=self.data['credential'],
            status=VerificableCredential.Status.ENABLED
        )
        if not all([cred.exists(), did.exists()]):
            return

        did = did[0].did
        cred = cred[0]
        cred.get_issued(did)

        if commit:
            cred.save()
            return cred
        
        return 



class CredentialPresentationForm(forms.Form):
    organization = forms.ChoiceField(choices=ORGANIZATION)
    credential = forms.ChoiceField(choices=[])

    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop('user', None)
        super().__init__(*args, **kwargs)
        self.fields['credential'].choices = [
            (x.id, x.type()) for x in VerificableCredential.objects.filter(
                user=self.user,
                status=VerificableCredential.Status.ISSUED
            )
        ]

    def save(self, commit=True):
        org = Organization.objects.filter(
            id=self.data['organization']
        )
        cred = VerificableCredential.objects.filter(
            user=self.user,
            id=self.data['credential'],
            status=VerificableCredential.Status.ISSUED
        )
        if not all([org.exists(), cred.exists()]):
            return

        org =org[0]
        cred = cred[0]

        if commit:
            org.send(cred)
            return cred
        
        return