def add_supervisor(request, pk):
obj = get_object_or_404(
Site, pk=int(pk))
group = Group.objects.get(name__exact="Site Supervisor")
role_obj = UserRole(site=obj, group=group)
if request.method == 'POST':
form = SetSupervisorForm(data=request.POST, instance=role_obj, request=request)
if form.is_valid():
role_obj = form.save(commit=False)
user_id = request.POST.get('user')
role_obj.user_id = int(user_id)
role_obj.save()
messages.add_message(request, messages.INFO, 'Site Supervisor Added')
return HttpResponseRedirect(reverse("fieldsight:site-dashboard", kwargs={'pk': obj.pk}))
else:
form = SetSupervisorForm(instance=role_obj, request=request)
return render(request, "fieldsight/add_supervisor.html", {'obj':obj,'form':form})
def add_project_role(request, pk):
obj = get_object_or_404(
Project, pk=pk)
role_obj = UserRole(project=obj)
scenario = 'Assign People'
form = SetProjectRoleForm(instance=role_obj, request=request)
if request.method == 'POST':
form = SetProjectRoleForm(data=request.POST, instance=role_obj, request=request)
if form.is_valid():
role_obj = form.save(commit=False)
user_id = request.POST.get('user')
role_obj.user_id = int(user_id)
role_obj.save()
messages.add_message(request, messages.INFO, '{} Added'.format(role_obj.group.name))
return HttpResponseRedirect(reverse("fieldsight:project-dashboard", kwargs={'pk': obj.pk}))
existing_staffs = obj.get_staffs
return render(request, "fieldsight/add_central_engineer.html", {'obj':obj,'form':form, 'scenario':scenario,
"existing_staffs":existing_staffs})
def add_central_engineer(request, pk):
obj = get_object_or_404(
Project, pk=pk)
group = Group.objects.get(name__exact="Reivewer")
role_obj = UserRole(project=obj, group=group)
scenario = 'Assign'
if request.method == 'POST':
form = SetProjectRoleForm(data=request.POST, instance=role_obj, request=request)
if form.is_valid():
role_obj = form.save(commit=False)
user_id = request.POST.get('user')
role_obj.user_id = int(user_id)
role_obj.save()
messages.add_message(request, messages.INFO, 'Reviewer Added')
return HttpResponseRedirect(reverse("fieldsight:project-dashboard", kwargs={'pk': obj.pk}))
else:
form = SetProjectRoleForm(instance=role_obj, request=request,)
return render(request, "fieldsight/add_central_engineer.html", {'obj':obj,'form':form, 'scenario':scenario})
def add_org_admin(request, pk=None):
organization = get_object_or_404(Organization, id=pk)
group = Group.objects.get(name__exact="Organization Admin")
role_obj = UserRole(organization=organization,group=group)
scenario = 'Assign'
if request.POST:
form = AssignOrgAdmin(data=request.POST, instance=role_obj, request=request)
if form.is_valid():
role_obj = form.save(commit=False)
user_id = request.POST.get('user')
role_obj.user_id = int(user_id)
role_obj.save()
messages.add_message(request, messages.INFO, 'Organization Admin Added')
return HttpResponseRedirect(reverse("fieldsight:organization-dashboard", kwargs={'pk': pk}))
else:
form = AssignOrgAdmin(instance=role_obj, request=request)
return render(request, 'fieldsight/add_admin_form.html',
{'form': form, 'scenario': scenario, 'obj': organization})
