def edit(request, id):
digest = get_object(request, Digest, id)
if request.method == 'POST':
form = DigestForm(request.POST, instance=digest)
if form.is_valid():
form.save()
return HttpResponseRedirect('/control/')
else:
form = DigestForm(instance=digest,initial={'template': digest.template.id})
form.fields['template'].choices = templates_as_choices(request)
return render_to_response('control/digest-edit.html', {'form': form},
context_instance=RequestContext(request))
def new(request):
message = ''
groupinfo = GroupInfo(request.user)
form = DigestForm()
form.fields['template'].choices = templates_as_choices(request)
if groupinfo.get_digests_limit() == Digest.objects.filter(user=request.user).count():
message = 'You have reached your Digest limit of %i. \
Please remove a Digest or <a href="/pricing">upgrade your package</a>\
to add more.' % (groupinfo.get_digests_limit())
elif request.method == 'POST':
form = DigestForm(request.POST)
if form.is_valid():
digest = form.save(commit=False)
digest.user = request.user
digest.save()
return HttpResponseRedirect(digest.get_feeds_url())
return render_to_response('control/digest.html', {'form': form,
'message': message},
context_instance=RequestContext(request))
