These are changes for issue #405, add email comment notification.

2 files changed, 11 insertions, 0 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}