X-Git-Url: https://git.mdrn.pl/cas.git/blobdiff_plain/37e278e43ece42375352c07d9f04991ef58d992c..7e61a123a0b4b5f6641cb791f76cda92369dca4b:/src/accounts/views.py

diff --git a/src/accounts/views.py b/src/accounts/views.py
index b00dd20..a501b97 100644
--- a/src/accounts/views.py
+++ b/src/accounts/views.py
@@ -1,16 +1,19 @@
 # Create your views here.
 from django import http
+from django.contrib.auth.decorators import login_required
+from django.contrib import messages
+from django.shortcuts import render
 from django.utils.translation import ugettext as __
-from django.views.generic.simple import direct_to_template
 from django.views.decorators.http import require_POST
-from django.contrib.auth.decorators import login_required
 from accounts.forms import UserBasicForm, UserPasswordForm
+from .models import Service
 
 @login_required
 def account_profile(request, basic_form=None, pass_form=None):
-    return direct_to_template(request, "account/profile.html", {
+    return render(request, "account/profile.html", {
         "basic_form": basic_form or UserBasicForm(instance=request.user),
         "pass_form": pass_form or UserPasswordForm(),
+        "services": Service.objects.all(),
     })
 
 
@@ -21,7 +24,7 @@ def account_change_basic_profile(request):
 
     if form.is_valid():
         form.save()
-        request.user.message_set.create(message=__("Profile has been changed."))
+        messages.add_message(request, messages.INFO, __("Profile has been changed."))
         return http.HttpResponseRedirect('/accounts/')
 
     return account_profile(request, basic_form=form)
@@ -35,7 +38,7 @@ def account_change_password(request):
         request.user.set_password(form.cleaned_data['new_password'])
         request.user.save()
 
-        request.user.message_set.create(message=__("Password has been changed."))
+        messages.add_message(request, messages.INFO, __("Password has been changed."))
         return http.HttpResponseRedirect('/accounts/')
 
     return account_profile(request, pass_form=form)