diff options
| -rwxr-xr-x | profile/views.py | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/profile/views.py b/profile/views.py index 832ccf6..8dfed4a 100755 --- a/profile/views.py +++ b/profile/views.py @@ -10,6 +10,8 @@ from pytask.profile.utils import get_notification @login_required def view_profile(request): + """ Display the profile information. + """ user = request.user profile = user.get_profile() @@ -21,6 +23,8 @@ def view_profile(request): @login_required def edit_profile(request): + """ Make only a few fields editable. + """ user = request.user profile = user.get_profile() @@ -86,3 +90,57 @@ def view_notification(request, nid): } return render_to_response('profile/view_notification.html', context) + +@login_required +def delete_notification(request, nid): + """ check if the user owns the notification and delete it. + """ + + user = request.user + newest, newer, notification, older, oldest = get_notification(nid, user) + + if not notification: + raise Http404 + + notification.is_deleted = True + notification.save() + + context = {'user':user, + 'notification':notification, + 'newest':newest, + 'newer':newer, + 'older':older, + 'oldest':oldest, + } + redirect_url = "/profile/notf/" + \ + "view/nid=%s"%older.uniq_key if older else "browse" + + return redirect(redirect_url) + +@login_required +def unread_notification(request, nid) + + """ check if the user owns the notification and delete it. + """ + + user = request.user + newest, newer, notification, older, oldest = get_notification(nid, user) + + if not notification: + raise Http404 + + notification.is_read = False + notification.save() + + context = {'user':user, + 'notification':notification, + 'newest':newest, + 'newer':newer, + 'older':older, + 'oldest':oldest, + } + + redirect_url = "/profile/notf/" + \ + "view/nid=%s"%older.uniq_key if older else "browse" + + return redirect(redirect_url) |