1 files changed, 60 insertions, 1 deletions
diff --git a/account/views.py b/account/views.py
index 2419fe6..e6b90b1 100644
--- a/account/views.py
+++ b/account/views.py
@@ -5,5 +5,64 @@ views.py of app 'account'
 """
 
 from django.shortcuts import render
+from django.views.generic.base import TemplateView
+from django.views.generic.edit import UpdateView
+from django.core.urlresolvers import reverse_lazy
+from django.http import HttpResponseRedirect
+
+from account.models import UserProfile
+from account.forms import UpdateProfileForm
+
+
+###### Class-based views ######
+class ProfileView(TemplateView):
+    """
+    class view to show profile page
+    """
+    template_name = 'account/profile.html'
+
+    def get_context_data(self, **kwargs):
+        context = super(ProfileView, self).get_context_data(**kwargs)
+        user = self.request.user
+        profile = user.userprofile_set.get(user=user)
+        context['user'] = user
+        context['profile'] = profile
+        return context
+
+class UpdateProfileView(UpdateView):
+    form_class = UpdateProfileForm
+    model = UserProfile
+    template_name = 'account/profile_update.html'
+    success_url = reverse_lazy('profile_update_done')
+
+    # get profile object
+    def get_object(self, queryset=None):
+        user = self.request.user
+        profile = user.userprofile_set.get(user=user)
+        return profile
+
+    def get(self, request, *args, **kwargs):
+        """
+        Returns the keyword arguments for instantiating the form.
+        modify this method to add 'email' data
+        """
+        self.object = self.get_object()
+        form_class = self.get_form_class()
+        form = self.get_form(form_class)
+        # initialize form 'email' field
+        user = self.request.user
+        form.fields['email'].initial = user.email
+        return self.render_to_response(self.get_context_data(form=form))
+
+    def form_valid(self, form):
+        """
+        modify 'form_valid' to update email field
+        """
+        form_data = form.cleaned_data
+        # update email and save
+        user = self.request.user
+        user.email = form_data.get('email', user.email)
+        user.save()
+        return super(UpdateProfileView, self).form_valid(form)
+
 
-# Create your views here.