Activity status added

Author: kwn

activities/__init__.py

@@ -18,3 +18,4 @@
 session = Session.connect(config.get('mongo', {}).get('database'))
 
 import activities
+import status

activities/documents.py

@@ -2,24 +2,34 @@
 from mongoalchemy.fields import *
 
 
+class Status(Document):
+    new_activities = IntField()
+    user_id = IntField()
+
+    def to_dict(self):
+        return {
+            'new_activities': self.new_activities
+        }
+
+
 class Activity(Document):
     config_collection_name = 'activity'
     config_polymorphic = 'type'
     config_polymorphic_collection = True
 
-    type = StringField(on_update='ignore', )
-    created_at = DateTimeField(on_update='ignore')
-    clicked = BoolField(required=True)
-    user_id = IntField(on_update='ignore')
+    type = StringField()
+    created_at = DateTimeField()
+    clicked = BoolField()
+    user_id = IntField()
 
 
 class Like(Activity):
     config_polymorphic_identity = 'like'
 
-    performed_by = StringField(on_update='ignore')
-    performer_avatar = StringField(on_update='ignore')
-    post_id = IntField(on_update='ignore')
-    post_image = StringField(on_update='ignore')
+    performed_by = StringField()
+    performer_avatar = StringField()
+    post_id = IntField()
+    post_image = StringField()
 
     def to_dict(self):
         return {
@@ -37,10 +47,10 @@ def to_dict(self):
 class Comment(Activity):
     config_polymorphic_identity = 'comment'
 
-    performed_by = StringField(on_update='ignore')
-    performer_avatar = StringField(on_update='ignore')
-    post_id = IntField(on_update='ignore')
-    post_image = StringField(on_update='ignore')
+    performed_by = StringField()
+    performer_avatar = StringField()
+    post_id = IntField()
+    post_image = StringField()
 
     def to_dict(self):
         return {
@@ -58,10 +68,10 @@ def to_dict(self):
 class Repost(Activity):
     config_polymorphic_identity = 'repost'
 
-    performed_by = StringField(on_update='ignore')
-    performer_avatar = StringField(on_update='ignore')
-    post_id = IntField(on_update='ignore')
-    post_image = StringField(on_update='ignore')
+    performed_by = StringField()
+    performer_avatar = StringField()
+    post_id = IntField()
+    post_image = StringField()
 
     def to_dict(self):
         return {
@@ -79,8 +89,8 @@ def to_dict(self):
 class Follow(Activity):
     config_polymorphic_identity = 'follow'
 
-    performed_by = StringField(on_update='ignore')
-    performer_avatar = StringField(on_update='ignore')
+    performed_by = StringField()
+    performer_avatar = StringField()
 
     def to_dict(self):
         return {
@@ -96,10 +106,10 @@ def to_dict(self):
 class Mention(Activity):
     config_polymorphic_identity = 'mention'
 
-    performed_by = StringField(on_update='ignore')
-    performer_avatar = StringField(on_update='ignore')
-    post_id = IntField(on_update='ignore')
-    post_image = StringField(on_update='ignore')
+    performed_by = StringField()
+    performer_avatar = StringField()
+    post_id = IntField()
+    post_image = StringField()
 
     def to_dict(self):
         return {
@@ -117,10 +127,10 @@ def to_dict(self):
 class Badge(Activity):
     config_polymorphic_identity = 'badge'
 
-    badge_name = StringField(on_update='ignore')
-    badge_image = StringField(on_update='ignore')
-    post_id = IntField(on_update='ignore')
-    post_image = StringField(on_update='ignore')
+    badge_name = StringField()
+    badge_image = StringField()
+    post_id = IntField()
+    post_image = StringField()
 
     def to_dict(self):
         return {

activities/status.py

@@ -0,0 +1,15 @@
+from flask import jsonify
+from . import app, session
+from flask_cors import cross_origin
+
+
+@app.route('/status')
+@cross_origin()
+def get_status():
+    return jsonify(new_activities=3)
+
+
+@app.route('/status', methods=['PUT'])
+@cross_origin()
+def put_status():
+    return jsonify(new_activities=0)