From c11f21ab3c64c6ef9d9cde8bfdcddf12f6663932 Mon Sep 17 00:00:00 2001
From: Joar Wandborg <git@wandborg.com>
Date: Mon, 27 Jun 2011 23:39:40 +0200
Subject: [PATCH] Issue 362 - Add simple comments

*   Added MediaComment database model
    Holds `media_entry` (`ObjectId`), `author` (`ObjectId`), `created`, `content` and `content_html`.
---
 mediagoblin/db/models.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/mediagoblin/db/models.py b/mediagoblin/db/models.py
index 600b79ff..5d00aa34 100644
--- a/mediagoblin/db/models.py
+++ b/mediagoblin/db/models.py
@@ -147,8 +147,32 @@ class MediaEntry(Document):
     def uploader(self):
         return self.db.User.find_one({'_id': self['uploader']})
 
+class MediaComment(Document):
+    __collection__ = 'media_comments'
 
-REGISTER_MODELS = [MediaEntry, User]
+    structure = {
+        'media_entry': ObjectId,
+        'author': ObjectId,
+        'created': datetime.datetime,
+        'content': unicode,
+        'content_html': unicode}
+
+    required_fields = [
+        'author', 'created', 'content']
+
+    default_values = {
+        'created': datetime.datetime.utcnow}
+
+    def media_entry(self):
+        pass
+
+    def author(self):
+        return self.db.User.find_one({'_id': self['author']})
+
+REGISTER_MODELS = [
+    MediaEntry,
+    User,
+    MediaComment]
 
 
 def register_models(connection):