basic pluggable like model

Author: killpanda

firefly/models/like.py

@@ -0,0 +1,40 @@
+# coding: utf-8
+
+from __future__ import absolute_import
+from datetime import datetime
+
+from firefly.ext import db
+from .user import User
+
+
+class Likes(object):
+    def __init__(self, product_type):
+        self.product_type = product_type
+        self._instance = None
+
+    def __get__(self, instance, owner):
+        self._instance = instance
+        return self
+
+    def productidgetter(self, func):
+        return func
+
+    @property
+    def product_id(self):
+        return self.productidgetter(self._instance)
+
+    def add(self, user_id):
+        user = User.objects.filter(id=user_id).first()
+        return Like.objects.create(
+            product_id=self.product_id,
+            product_type=self.product_type,
+            user=user
+        )
+
+
+class Like(db.Document):
+    id = db.SequenceField(primary_key=True)
+    created_at = db.DateTimeField(default=datetime.utcnow, required=True)
+    product_type = db.IntField()
+    product_id = db.IntField()
+    user = db.ReferenceField(User)

firefly/models/topic.py

@@ -10,6 +10,7 @@
 from firefly.views.utils import timesince
 from firefly.models.consts import CATEGORY_COLORS
 from .user import User
+from .like import Likes
 
 __all__ = ["Category", "Post", "Video", "Image", "Comment"]
 
@@ -58,6 +59,12 @@ class Post(db.Document):
     comments = db.ListField(db.ReferenceField('Comment'))
     category = db.ReferenceField(Category)
 
+    likes = Likes('Post')
+
+    @likes.productidgetter
+    def product_id(self):
+        return str(self.id)
+
     def url(self):
         return url_for('post.detail', id=self.id)
 
@@ -84,14 +91,17 @@ def recent_activity_time(self):
 
 
 class Video(Post):
+    likes = Likes('Video')
     embed_code = db.StringField(required=True)
 
 
 class Image(Post):
+    likes = Likes('Image')
     image_url = db.StringField(required=True, max_length=255)
 
 
 class Quote(Post):
+    likes = Likes('Quote')
     content = db.StringField(required=True)
     author = db.ReferenceField(User)
 
@@ -103,6 +113,12 @@ class Comment(db.Document):
     author = db.ReferenceField(User)
     ref_id = db.IntField(default=0)
 
+    likes = Likes('Comment')
+
+    @likes.productidgetter
+    def product_id(self):
+        return str(self.id)
+
     @property
     def post_type(self):
         return self.__class__.__name__