List activities working example

Author: kwn

activities/__init__.py

@@ -2,36 +2,19 @@
 from flask import Flask
 from mongoalchemy.session import Session
 from documents import Activity
-from flask_mongoengine import MongoEngine
-from flask_mongorest import MongoRest
+from flask_cors import CORS
 
 with open('config.yml', 'r') as f:
     config_file = f.read()
 
 config = yaml.load(config_file)
 
 app = Flask(__name__)
-app.debug = config.get('app', {}).get('debug', False)
-# app.config.update(
-#     MONGODB_HOST=config.get('mongo', {}).get('hostname'),
-#     MONGODB_PORT=config.get('mongo', {}).get('port'),
-#     MONGODB_DB=config.get('mongo', {}).get('database'),
-# )
+cors = CORS(app)
 
-# db = MongoEngine(app)
-# api = MongoRest(app)
+app.debug = config.get('app', {}).get('debug', False)
+app.config['CORS_HEADERS'] = 'Content-Type'
 
 session = Session.connect(config.get('mongo', {}).get('database'))
 
-
-@app.route('/activities')
-def list_activities():
-    activities = session.query(Activity).filter(Activity.user_id == 12)
-
-    # print activities.one()
-
-    for activity in activities:
-        print activity
-
-    return 'test'
-
+import activities

activities/activities.py

@@ -0,0 +1,31 @@
+from flask import jsonify, request
+from . import app, session
+from documents import Activity
+from pymongo import DESCENDING
+from flask_cors import cross_origin
+
+PAGE_SIZE = 15
+
+
+@app.route('/activities')
+@cross_origin()
+def list_activities():
+    try:
+        page = int(request.args.get('page', 0))
+        assert page >= 0
+    except (ValueError, AssertionError):
+        page = 0
+
+    activities = session.query(Activity).filter(Activity.user_id == 12).sort((Activity.created_at, DESCENDING)) \
+        .limit(PAGE_SIZE).skip(page * PAGE_SIZE)
+
+    meta = {
+        'total': activities.count(),
+        'offset': page * PAGE_SIZE,
+        'count': activities.count(with_limit_and_skip=True)
+    }
+
+    # res = [activity.to_dict() for activity in activities]
+    res = map(lambda activity: activity.to_dict(), activities)
+
+    return jsonify(results=res, meta=meta)

activities/documents.py

@@ -6,7 +6,6 @@ class Activity(Document):
     config_collection_name = 'activity'
     config_polymorphic = 'type'
     config_polymorphic_collection = True
-    config_extra_fields = 'ignore'
 
     type = StringField()
     performed_by = StringField()
@@ -22,8 +21,19 @@ def __str__(self):
 class Like(Activity):
     config_polymorphic_identity = 'like'
 
-    def __str__(self):
-        return '%s: like by %s' % (self.created_at, self.performed_by)
+    post_id = IntField()
+    post_image = StringField()
+
+    def to_dict(self):
+        return {
+            'type': self.type,
+            'performed_by': self.performed_by,
+            'performer_avatar': self.performer_avatar,
+            'post_id': self.post_id,
+            'post_image': self.post_image,
+            'created_at': self.created_at.isoformat(),
+            'clicked': self.clicked
+        }
 
 
 class Comment(Activity):
@@ -32,8 +42,16 @@ class Comment(Activity):
     post_id = IntField()
     post_image = StringField()
 
-    def __str__(self):
-        return '%s: comment by %s' % (self.created_at, self.performed_by)
+    def to_dict(self):
+        return {
+            'type': self.type,
+            'performed_by': self.performed_by,
+            'performer_avatar': self.performer_avatar,
+            'post_id': self.post_id,
+            'post_image': self.post_image,
+            'created_at': self.created_at.isoformat(),
+            'clicked': self.clicked
+        }
 
 
 class Repost(Activity):
@@ -42,15 +60,29 @@ class Repost(Activity):
     post_id = IntField()
     post_image = StringField()
 
-    def __str__(self):
-        return '%s: repost by %s' % (self.created_at, self.performed_by)
+    def to_dict(self):
+        return {
+            'type': self.type,
+            'performed_by': self.performed_by,
+            'performer_avatar': self.performer_avatar,
+            'post_id': self.post_id,
+            'post_image': self.post_image,
+            'created_at': self.created_at.isoformat(),
+            'clicked': self.clicked
+        }
 
 
 class Follow(Activity):
     config_polymorphic_identity = 'follow'
 
-    def __str__(self):
-        return '%s: follow by %s' % (self.created_at, self.performed_by)
+    def to_dict(self):
+        return {
+            'type': self.type,
+            'performed_by': self.performed_by,
+            'performer_avatar': self.performer_avatar,
+            'created_at': self.created_at.isoformat(),
+            'clicked': self.clicked
+        }
 
 
 class Mention(Activity):
@@ -59,8 +91,16 @@ class Mention(Activity):
     post_id = IntField()
     post_image = StringField()
 
-    def __str__(self):
-        return '%s: mention by %s' % (self.created_at, self.performed_by)
+    def to_dict(self):
+        return {
+            'type': self.type,
+            'performed_by': self.performed_by,
+            'performer_avatar': self.performer_avatar,
+            'post_id': self.post_id,
+            'post_image': self.post_image,
+            'created_at': self.created_at.isoformat(),
+            'clicked': self.clicked
+        }
 
 
 class Badge(Activity):
@@ -69,5 +109,13 @@ class Badge(Activity):
     badge_name = StringField()
     badge_image = StringField()
 
-    def __str__(self):
-        return '%s: badge by %s' % (self.created_at, self.performed_by)
+    def to_dict(self):
+        return {
+            'type': self.type,
+            'performed_by': self.performed_by,
+            'performer_avatar': self.performer_avatar,
+            'badge_name': self.badge_name,
+            'badge_image': self.badge_image,
+            'created_at': self.created_at.isoformat(),
+            'clicked': self.clicked
+        }