def queryp_update(request, pk):
"""Problem query update view for user who added the query."""
queryu = get_object_or_404(QueryP, pk=pk)
form = QueryFormP(request.POST or None, instance=queryu)
if form.is_valid():
form.save()
return redirect('home')
return render(request, 'vtr/update_form.html', {'form': form})
def add_queryP(request):
"""View for protune add query."""
if request.method == 'POST':
query_form = QueryFormP(request.POST, request.FILES)
if query_form.is_valid():
query = query_form.save(commit=False)
query.user = request.user
query.save()
query_form.save_m2m()
if UserProfileF.objects.filter(user=request.user.id):
return redirect(probQF)
else:
return redirect(probQS)
else:
print query_form.errors
else:
query_form = QueryFormP()
return render(request, 'probtune/add_query.html', {'query_form': query_form},)
