def edit_existing_schedule(request, schedule_pk):
if not request.user.has_perm(Schedule._meta.get_change_permission()):
raise PermissionDenied
schedule = get_object_or_404(Schedule, pk=schedule_pk)
spot_dicts = spot_dicts_from_schedule(schedule)
request.session["spots"] = spot_dicts
return HttpResponseRedirect(reverse("admin:edit_schedule", kwargs={"schedule_pk": schedule_pk}))
def generate_schedule(request, schedule_pk):
if not request.user.has_perm(Schedule._meta.get_change_permission()):
raise PermissionDenied
response = None
generate_form = GenerateScheduleForm()
copy_form = CopyScheduleForm()
try:
if request.method == "POST":
generate_form = GenerateScheduleForm(request.POST)
copy_form = CopyScheduleForm(request.POST)
spot_dicts = None
if generate_form.is_valid() and generate_form.cleaned_data["increment"]:
dj = generate_form.cleaned_data["default_dj"]
if dj:
dj = DJ.objects.get(pk=dj)
else:
dj = None
show = generate_form.cleaned_data["default_show"]
if show:
show = Show.objects.get(pk=show)
else:
show = None
spot_dicts = spot_generate_dicts(int(generate_form.cleaned_data["increment"]), dj, show)
elif copy_form.is_valid():
schedule = Schedule.objects.get(pk=generate_form["schedule"])
spot_dicts = spot_dicts_from_schedule(schedule, {"pk": -1})
request.session["spots"] = spot_dicts
return HttpResponseRedirect(reverse("admin:edit_schedule", kwargs={"schedule_pk": schedule_pk}))
except Schedule.DoesNotExist:
pass
except DJ.DoesNotExist:
pass
except Show.DoesNotExist:
pass
if response is None:
context = {
"schedule": get_object_or_404(Schedule, pk=schedule_pk),
"generate_form": generate_form,
"copy_form": copy_form,
}
response = render_to_response("station/admin/generate.html", context, context_instance=RequestContext(request))
return response
