def edit_program(request, program_id):
program = get_object_or_404(Program.objects.filter(domain=request.user.selected_domain), pk=program_id)
if request.method == "POST":
original_ids = set([user.id for user in program.get_users()])
form = ProgramForm(request.POST, instance=program)
if form.is_valid():
edit_program = form.save(commit=False)
edit_program.domain = request.user.selected_domain
edit_program.save()
if "users" in request.POST:
updated_ids = set([int(id) for id in request.POST.getlist("users")])
else:
updated_ids = set([])
to_remove = original_ids.difference(updated_ids)
to_add = updated_ids.difference(original_ids)
for user_id in to_add:
edit_program.add_user(User.objects.get(id=user_id))
for user_id in to_remove:
edit_program.remove_user(User.objects.get(id=user_id))
return HttpResponseRedirect(reverse('list_programs'))
else:
form = ProgramForm(instance=program)
users = User.objects.filter(domain_membership__domain = request.user.selected_domain)
def add_selected(user, program):
if user in program.get_users(): user.selected = True
return user
users = map(add_selected, users, (program,) * users.count())
return render_to_response(request, "program/program_add.html",
{"form": form, "users": users})
def add_program(request):
"""Add a program to a domain"""
if request.method == "POST":
form = ProgramForm(request.POST)
if form.is_valid():
new_program = form.save(commit=False)
new_program.domain = request.user.selected_domain
new_program.save()
if "users" in request.POST:
user_ids = request.POST.getlist("users")
else:
user_ids = set([])
for user_id in user_ids:
new_program.add_user(User.objects.get(id=user_id))
return HttpResponseRedirect(reverse('list_programs'))
else:
form = ProgramForm()
users = User.objects.filter(domain_membership__domain = request.user.selected_domain)
return render_to_response(request, "program/program_add.html",
{"form": form, "users": users})
