Application refactored

Author: kwn

activities.py

@@ -1,6 +1,7 @@
 import yaml
 import pymongo
-from flask import Flask
+from flask import Flask, Response
+from bson.json_util import dumps
 
 app = Flask(__name__)
 app.debug = True
@@ -17,14 +18,23 @@ def get_collection(collection_name):
     db = client[mongo_config.get('database')]
     return db[collection_name]
 
-
+# date               : '',
+# performer_id       : '',
+# performer_username : '',
+# performer_avatar   : '',
+# type               : '',
+# related_id         : '',
+# related_image      : '',
+# info               : '',
+# clicked            : ''
 @app.route("/activities")
 def get_activities():
-    activities = get_collection('activities').find({"user_id": 12}).sort([
+    cursor = get_collection('activities').find({"user_id": 12}, {"_id": 0, "user_id": 0}).sort([
         ("created_at", pymongo.DESCENDING)
     ])
 
-    return "Hello World 123!"
+    return Response(dumps(cursor), status=200, mimetype='application/json')
+
 
 if __name__ == "__main__":
     app.run()

activities/__init__.py

@@ -0,0 +1,22 @@
+import yaml
+from flask import Flask
+from flask_mongoengine import MongoEngine
+from flask_mongorest import MongoRest
+
+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'),
+)
+
+db = MongoEngine(app)
+api = MongoRest(app)
+
+import activities.views

activities/documents.py

@@ -0,0 +1,13 @@
+from mongoengine import *
+
+
+class Activity(Document):
+    type = StringField()
+    performed_by = StringField()
+    performer_avatar = StringField()
+    created_at = DateTimeField()
+    post_id = IntField()
+    post_image = StringField()
+    badge_name = StringField()
+    badge_image = StringField()
+    clicked = BooleanField()

activities/resources.py

@@ -0,0 +1,6 @@
+from flask_mongorest.resources import Resource
+from documents import Activity
+
+
+class ActivityResource(Resource):
+    document = Activity

activities/views.py

@@ -0,0 +1,10 @@
+from activities import api
+from flask_mongorest.methods import *
+from flask_mongorest.views import ResourceView
+from resources import ActivityResource
+
+
+@api.register(name='activities', url='/activities/')
+class ActivityView(ResourceView):
+    resource = ActivityResource
+    methods = [List]

config.dist.yml

@@ -1,3 +1,5 @@
+app:
+  debug: false
 mysql:
   host: localhost
   database: db

makefile

@@ -1,2 +1,5 @@
+run:
+	python run.py
+
 db-data-copy:
-	python db-data-copy.py
+	python db-data-copy.py

run.py

@@ -0,0 +1,4 @@
+from activities import app
+
+if __name__ == '__main__':
+    app.run()