from django import forms

from pingcap_sso.blog.models import (
    User as BlogUser,
    Role as BlogRole,
)


class BlogUserRoleForm(forms.Form):

    user_id = forms.IntegerField(required=True)
    role_id = forms.IntegerField(required=True)

    def clean_user_id(self):
        user_id = self.cleaned_data.get('user_id')
        try:
            self.user = BlogUser.objects.get(id=user_id)
        except BlogUser.DoesNotExist:
            raise forms.ValidationError('user not found')
        return user_id

    def clean_role_id(self):
        role_id = self.cleaned_data.get('role_id')
        try:
            self.role = BlogRole.objects.get(id=role_id)
        except BlogRole.DoesNotExist:
            raise forms.ValidationError('role not found')
        return role_id