Merge pull request #134 from spaceshipearth/mission_serializer

mission: add basic serializer

Author: igorkofman

requirements.txt

@@ -51,10 +51,11 @@ redis==2.10.6
 requests==2.21.0
 rsa==4.0
 sendgrid==6.0.4
+serpy==0.3.1
 six==1.12.0
 SQLAlchemy==1.3.4
 structlog==19.1.0
--e git+https://github.com/spaceshipearth/tasktiger.git@exceptions_as_context#egg=tasktiger
+-e git+https://github.com/spaceshipearth/tasktiger.git@2f159cec740acc8035da136dfa4253723a40886d#egg=tasktiger
 typed-ast==1.3.1
 urllib3==1.24.2
 webassets==0.12.1

spaceship/goals.py

@@ -1,4 +1,5 @@
 
+import serpy
 
 GOALS_BY_CATEGORY = {
   "diet": {
@@ -153,6 +154,14 @@
     }
 }
 
+class GoalSerializer(serpy.DictSerializer):
+  name = serpy.StrField()
+  icon = serpy.StrField()
+  alt = serpy.StrField()
+  short_description = serpy.StrField()
+  description = serpy.StrField()
+  category = serpy.StrField()
+
 # let goals know their category
 for category_name, category_info in GOALS_BY_CATEGORY.items():
   category_info["name"] = category_name

spaceship/models/base.py

@@ -1,11 +1,18 @@
 
+import serpy
 from sqlalchemy.exc import DatabaseError
 
 from flask_sqlalchemy.model import Model
 
 from spaceship.db import db
 
 class Base(Model):
+  class Serializer(serpy.Serializer):
+    id = serpy.IntField()
+
+  def serialize(self):
+    return self.Serializer(self).data
+
   @property
   def session(self):
     return db.session

spaceship/models/goal.py

@@ -1,7 +1,9 @@
 
 import pendulum
+import serpy
 
 from spaceship.db import db
+from spaceship.goals import GOALS_BY_CATEGORY, GoalSerializer
 from spaceship.models.custom_fields import PendulumDateTimeField
 from sqlalchemy.ext.hybrid import hybrid_property
 
@@ -16,6 +18,16 @@ class Goal(db.Model):
   visible_after = db.Column(PendulumDateTimeField(), nullable=True)
   hidden_after = db.Column(PendulumDateTimeField(), nullable=True)
 
+  class Serializer(db.Model.Serializer):
+    short_description = serpy.StrField()
+    category = serpy.StrField()
+    actual_goal = GoalSerializer()
+
+  @property
+  def actual_goal(self):
+    cat_goals = GOALS_BY_CATEGORY[self.category]
+    return [g for g in cat_goals['goals'] if g['name'] == self.short_description][0]
+
   @hybrid_property
   def is_visible(self):
     now = pendulum.now('UTC')

spaceship/models/mission.py

@@ -1,10 +1,11 @@
 import pendulum
+import serpy
 from sqlalchemy.ext.hybrid import hybrid_property
 from sqlalchemy.ext.associationproxy import association_proxy
 import uuid
 
 from spaceship.db import db
-from spaceship.models import MissionGoal
+from spaceship.models import MissionGoal, Goal
 from spaceship.models.custom_fields import PendulumDateTimeField
 
 class Mission(db.Model):
@@ -27,6 +28,14 @@ class Mission(db.Model):
   created_at = db.Column(PendulumDateTimeField(), default=lambda: pendulum.now('UTC'))
   deleted_at = db.Column(PendulumDateTimeField(), nullable=True)
 
+  class Serializer(db.Model.Serializer):
+    uuid = serpy.StrField()
+    title = serpy.StrField()
+    short_description = serpy.StrField()
+    duration_in_weeks = serpy.IntField()
+
+    goals = Goal.Serializer(many=True)
+
   @property
   def is_deleted(self):
     return self.deleted_at != None