如何使用django-graphql-auth自定义寄存器突变

Question

django-graphql-auth是一个很棒的包，但文档中没有太多关于模型与自定义用户模型的关联。

我正在寻找一个客场使用这个包与自定义注册突变两种用户类型申请人和雇主

class User(AbstractUser):
email = models.EmailField(blank=False,
max_length=255,
verbose_name="email address")
is_applicant = models.BooleanField(default=False)
is_employer = models.BooleanField(default=False)
USERNAME_FIELD = "username"
EMAIL_FIELD = "email"

class Applicant(models.Model):
user = models.OneToOneField(
User,
on_delete=models.CASCADE,
primary_key=True,
)
fullname = models.CharField(max_length=255, null=True)

class Employer(models.Model):
user = models.OneToOneField(
User,
on_delete=models.CASCADE,
primary_key=True,
)
company_name = models.CharField(max_length=225, null=True)

要实现的突变目标：

mutation {
registerApplicant(
input: {
email: "[e-mail]",
username: "new_applicant",
password1: "createsuperpassword",
password2: "createsuperpassword",
isApplicant: true,
applicant {
fullname: "New Applicant"
}
}
) {
success,
errors,
token,
refreshToken
}
}
mutation {
registerEmployer(
input: {
email: "[e-mail]",
username: "new_employer",
password1: "createsuperpassword",
password2: "createsuperpassword",
isEmployer: true,
employer {
fullname: "New Employer"
}
}
) {
success,
errors,
token,
refreshToken
}
}

Answer 1

您可以只使用graphql_jwt完成整个过程，首先您需要定义输入；像这样：

class ApplicantInfoInput(graphene.InputObjectType):
email = graphene.String(required=True)
username = graphene.String(required=True)
password1 = graphene.String(required=True)
password2 = graphene.String(required=True)
is_applicant = graphene.Boolean(required=True)
fullname = graphene.String(required=True)

那么突变类应该是这样的：

from graphql_jwt.shortcuts import get_token, create_refresh_token
from ..models import User, Applicant
class UserCreateMutation(graphene.Mutation):
user = graphene.Field(UserType)
success = graphene.Boolean()
token = graphene.String()
refresh_token = graphene.String()
class Arguments:
input = ApplicantInfoInput(required=True)
@staticmethod
def mutate(root, info, input=None):
try:
user = User.objects.get(username=input.username)
except User.DoesNotExist as e:
user = None
if user:
raise Exception('User Already Exists!')
else:
user_instance = Applicant(fullname=input.fullname)
user_instance.user.email = input.email
user_instance.user.set_password(input.password1)
user_instance.user.is_applicant = True
user_instance.save()
token = get_token(user_instance)
refresh_token = create_refresh_token(user_instance)
return UserCreateMutation(user=user_instance,
token=token,
refresh_token=refresh_token,
success=True)