def query_update(request, pk): """View for query update to user who added the query.""" queryu = get_object_or_404(QueryS, pk=pk) form = QueryFormS(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_queryS(request): """View for student query form.""" if request.method == 'POST': query_form = QueryFormS(request.POST, request.FILES) if query_form.is_valid(): # is the form valid query = query_form.save(commit=False) if UserProfileF.objects.filter(user=request.user.id): query.branch = UserProfileF.objects.only('department').get(user=request.user).department else: query.branch = UserProfileS.objects.only('branch').get(user=request.user).branch query.user = request.user query.save() query_form.save_m2m() return redirect(home) else: print query_form.errors # no, display error to end user else: query_form = QueryFormS() return render(request, 'vtr/add_query.html', {'query_form': query_form},)