diff options
| -rw-r--r-- | mediagoblin/db/mongo/migrations.py | 7 | ||||
| -rw-r--r-- | mediagoblin/db/mongo/models.py | 4 | ||||
| -rw-r--r-- | mediagoblin/edit/forms.py | 5 | ||||
| -rw-r--r-- | mediagoblin/edit/views.py | 48 | ||||
| -rw-r--r-- | mediagoblin/templates/mediagoblin/user_pages/comment_email.txt | 26 | ||||
| -rw-r--r-- | mediagoblin/tests/test_edit.py | 1 | ||||
| -rw-r--r-- | mediagoblin/user_pages/lib.py | 54 | ||||
| -rw-r--r-- | mediagoblin/user_pages/views.py | 7 |
8 files changed, 133 insertions, 19 deletions
diff --git a/mediagoblin/db/mongo/migrations.py b/mediagoblin/db/mongo/migrations.py index c5766b0d..3de3694e 100644 --- a/mediagoblin/db/mongo/migrations.py +++ b/mediagoblin/db/mongo/migrations.py @@ -153,3 +153,10 @@ def convert_video_media_data(database): assert len(document['media_data']) == 1 document['media_data'] = document['media_data']['video'] collection.save(document) + +@RegisterMigration(11) +def user_add_wants_comment_notification(database): + """ + Add wants_comment_notification to user model + """ + add_table_field(database, 'users', 'wants_comment_notification', True) diff --git a/mediagoblin/db/mongo/models.py b/mediagoblin/db/mongo/models.py index 2e35a2b8..3f1363d5 100644 --- a/mediagoblin/db/mongo/models.py +++ b/mediagoblin/db/mongo/models.py @@ -62,6 +62,8 @@ class User(Document, UserMixin): we'll change this to a boolean with a key of 'active' and have a separate field for a reason the user's been disabled if that's appropriate... email_verified is already separate, after all.) + - wants_comment_notification: The user has selected that they want to be + notified when comments are posted on their media. - verification_key: If the user is awaiting email verification, the user will have to provide this key (which will be encoded in the presented URL) in order to confirm their email as active. @@ -80,6 +82,7 @@ class User(Document, UserMixin): 'pw_hash': unicode, 'email_verified': bool, 'status': unicode, + 'wants_comment_notification': bool, 'verification_key': unicode, 'is_admin': bool, 'url': unicode, @@ -93,6 +96,7 @@ class User(Document, UserMixin): default_values = { 'created': datetime.datetime.utcnow, 'email_verified': False, + 'wants_comment_notification': True, 'status': u'needs_email_verification', 'is_admin': False} diff --git a/mediagoblin/edit/forms.py b/mediagoblin/edit/forms.py index 46ee02e2..83600e4d 100644 --- a/mediagoblin/edit/forms.py +++ b/mediagoblin/edit/forms.py @@ -70,6 +70,11 @@ class EditAccountForm(wtforms.Form): wtforms.validators.Length(min=6, max=30)], id="password") + wants_comment_notification = wtforms.BooleanField( + _('Comment notification?'), + [wtforms.validators.Required()], + description=_("Check this box to be emailed when someone else comments on your media.")) + class EditAttachmentsForm(wtforms.Form): attachment_name = wtforms.TextField( diff --git a/mediagoblin/edit/views.py b/mediagoblin/edit/views.py index 2bcb5694..2704a20c 100644 --- a/mediagoblin/edit/views.py +++ b/mediagoblin/edit/views.py @@ -188,29 +188,39 @@ def edit_account(request): edit_username = request.GET.get('username') user = request.user - form = forms.EditAccountForm(request.POST) + form = forms.EditAccountForm(request.POST, + wants_comment_notification=user.wants_comment_notification) - if request.method == 'POST' and form.validate(): - password_matches = auth_lib.bcrypt_check_password( - request.POST['old_password'], - user.pw_hash) - - if (request.POST['old_password'] or request.POST['new_password']) and not \ - password_matches: - form.old_password.errors.append(_('Wrong password')) - - return render_to_response( - request, - 'mediagoblin/edit/edit_account.html', - {'user': user, - 'form': form}) - - if password_matches: - user.pw_hash = auth_lib.bcrypt_gen_password_hash( - request.POST['new_password']) + if request.method == 'POST': + #save status of comment checkbox first, so user does not need to + #change their password as well. + user.wants_comment_notification = request.POST.get( + 'wants_comment_notification', False) == u'y' user.save() + #check remaining fields for validation + if form.validate(): + password_matches = auth_lib.bcrypt_check_password( + request.POST['old_password'], + user.pw_hash) + + if (request.POST['old_password'] or \ + request.POST['new_password']) and not \ + password_matches: + form.old_password.errors.append(_('Wrong password')) + + return render_to_response( + request, + 'mediagoblin/edit/edit_account.html', + {'user': user, + 'form': form}) + + if password_matches: + user.pw_hash = auth_lib.bcrypt_gen_password_hash( + request.POST['new_password']) + user.save() + messages.add_message(request, messages.SUCCESS, _("Account settings saved")) diff --git a/mediagoblin/templates/mediagoblin/user_pages/comment_email.txt b/mediagoblin/templates/mediagoblin/user_pages/comment_email.txt new file mode 100644 index 00000000..7bf649cf --- /dev/null +++ b/mediagoblin/templates/mediagoblin/user_pages/comment_email.txt @@ -0,0 +1,26 @@ +{# +# GNU MediaGoblin -- federated, autonomous media hosting +# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. +-#} + +{% trans username=username, comment_author=comment_author -%} + +Hi {{ username }}, +{{ comment_author }} commented on your post ({{ comment_url }}) at GNU MediaGoblin: +{% endtrans %} +{{ comment_content }} + +GNU MediaGoblin diff --git a/mediagoblin/tests/test_edit.py b/mediagoblin/tests/test_edit.py index 398a5d25..6c4020da 100644 --- a/mediagoblin/tests/test_edit.py +++ b/mediagoblin/tests/test_edit.py @@ -37,6 +37,7 @@ def test_change_password(test_app): '/edit/account/', { 'old_password': 'toast', 'new_password': '123456', + 'wants_comment_notification': 'y' }) # test_user has to be fetched again in order to have the current values diff --git a/mediagoblin/user_pages/lib.py b/mediagoblin/user_pages/lib.py new file mode 100644 index 00000000..00091d7a --- /dev/null +++ b/mediagoblin/user_pages/lib.py @@ -0,0 +1,54 @@ +# GNU MediaGoblin -- federated, autonomous media hosting +# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +from mediagoblin.tools.mail import send_email +from mediagoblin.tools.template import render_template +from mediagoblin import mg_globals + +def send_comment_email(user, comment, media, request): + """ + Sends comment email to user when a comment is made on their media. + + Args: + - user: the user object to whom the email is sent + - comment: the comment object referencing user's media + - media: the media object the comment is about + - request: the request + """ + + comment_url = u'http://{host}{comment_uri}'.format( + host=request.host, + comment_uri=request.urlgen( + 'mediagoblin.user_pages.media_home.view_comment', + comment = comment._id, + user = media.get_uploader.username, + media = media.slug_or_id) + '#comment') + + comment_author = comment.get_author['username'] + + rendered_email = render_template( + request, 'mediagoblin/user_pages/comment_email.txt', + {'username':user.username, + 'comment_author':comment_author, + 'comment_content':comment.content, + 'comment_url':comment_url}) + + send_email( + mg_globals.app_config['email_sender_address'], + [user.email], + 'GNU MediaGoblin - {comment_author} commented on your post'.format( + comment_author=comment_author), + rendered_email) diff --git a/mediagoblin/user_pages/views.py b/mediagoblin/user_pages/views.py index 530dea64..e93faf86 100644 --- a/mediagoblin/user_pages/views.py +++ b/mediagoblin/user_pages/views.py @@ -23,6 +23,7 @@ from mediagoblin.tools.translate import pass_to_ugettext as _ from mediagoblin.tools.pagination import Pagination from mediagoblin.tools.files import delete_media_files from mediagoblin.user_pages import forms as user_forms +from mediagoblin.user_pages.lib import send_comment_email from mediagoblin.decorators import (uses_pagination, get_user_media_entry, require_active_login, user_may_delete_media) @@ -158,6 +159,12 @@ def media_post_comment(request, media): request, messages.SUCCESS, _('Your comment has been posted!')) + media_uploader = media.get_uploader + #don't send email if you comment on your own post + if (comment.author != media_uploader and + media_uploader['wants_comment_notification']): + send_comment_email(media_uploader, comment, media, request) + return exc.HTTPFound( location=media.url_for_self(request.urlgen)) |