Refactoring

Author: kwn

activities.py

@@ -1,5 +1,7 @@
 import yaml
 from flask import Flask
+from mongoalchemy.session import Session
+from documents import Activity
 from flask_mongoengine import MongoEngine
 from flask_mongorest import MongoRest
 
@@ -10,13 +12,26 @@
 
 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'),
-)
+# 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)
+# db = MongoEngine(app)
+# api = MongoRest(app)
+
+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.views

activities/__init__.py

@@ -1,14 +1,73 @@
-from mongoengine import *
+from mongoalchemy.document import Document
+from mongoalchemy.fields import *
 
 
 class Activity(Document):
+    config_collection_name = 'activity'
+    config_polymorphic = 'type'
+    config_polymorphic_collection = True
+    config_extra_fields = 'ignore'
+
     type = StringField()
     performed_by = StringField()
     performer_avatar = StringField()
     created_at = DateTimeField()
+    clicked = BoolField()
+    user_id = IntField()
+
+    def __str__(self):
+        return '%s: activity by %s' % (self.created_at, self.performed_by)
+
+
+class Like(Activity):
+    config_polymorphic_identity = 'like'
+
+    def __str__(self):
+        return '%s: like by %s' % (self.created_at, self.performed_by)
+
+
+class Comment(Activity):
+    config_polymorphic_identity = 'comment'
+
+    post_id = IntField()
+    post_image = StringField()
+
+    def __str__(self):
+        return '%s: comment by %s' % (self.created_at, self.performed_by)
+
+
+class Repost(Activity):
+    config_polymorphic_identity = 'repost'
+
     post_id = IntField()
     post_image = StringField()
+
+    def __str__(self):
+        return '%s: repost by %s' % (self.created_at, self.performed_by)
+
+
+class Follow(Activity):
+    config_polymorphic_identity = 'follow'
+
+    def __str__(self):
+        return '%s: follow by %s' % (self.created_at, self.performed_by)
+
+
+class Mention(Activity):
+    config_polymorphic_identity = 'mention'
+
+    post_id = IntField()
+    post_image = StringField()
+
+    def __str__(self):
+        return '%s: mention by %s' % (self.created_at, self.performed_by)
+
+
+class Badge(Activity):
+    config_polymorphic_identity = 'badge'
+
     badge_name = StringField()
     badge_image = StringField()
-    clicked = BooleanField()
-    user_id = IntField()
+
+    def __str__(self):
+        return '%s: badge by %s' % (self.created_at, self.performed_by)