我正在使用 Django、django-allauth 和 django-invitations。我能够成功邀请用户加入该平台,但我想将他们与邀请者的公司关联起来。
我已经阅读了 bee-keeper/django-invitations,但似乎没有有关如何执行此操作的信息。
模型.py
class Company(models.Model):
name = models.CharField(max_length=100, default=None)
class CustomUser(AbstractUser):
company = models.ForeignKey(Company, on_delete=models.CASCADE, blank=True, null=True)
objects = CustomUserManager()
views.py
@login_required
def company_users(request):
# Get users that are in the company's user database as well as users that have been invited
company_users = CustomUser.objects.filter(company=request.user.company.id)
Invitations = get_invitation_model()
# I'm afraid this is going to get all invited users, not just those that belong to the company
invited_users = Invitations.objects.filter()
if request.method == 'POST':
print(request.POST)
invitees = request.POST['invitees']
invitees = re.split(',', invitees)
for invitee in invitees:
Invitation = get_invitation_model()
try:
invite = Invitation.create(invitee, inviter=request.user)
invite.send_invitation(request)
except IntegrityError as e:
print(type(e))
print(dir(e))
return render(request, "company_users.html", {
'message': e.args,
'company_users' : company_users,
'invited_users' : invited_users,
})
return render(request, 'company_users.html', {
'company_users' : company_users,
'invited_users' : invited_users,
})
在上面的代码中,用户被成功邀请到平台,但该用户与邀请者的公司没有关联。我还担心受邀请的用户列表不限于用户的公司。