se-hub/models/Message.py

import json

__author__ = 'Aran'
from google.appengine.ext import db
from models.User import User
from models.Project import Project
from models.Course import Course

class Message(db.Model):
    groupId = db.IntegerProperty(required=True)
    message = db.StringProperty(required=True, multiline=True)
    msgDate = db.DateTimeProperty(required=True)
    master_id = db.IntegerProperty(required=True)
    isProject = db.BooleanProperty(default=False)

    def to_JSON(self):
        user = User.get_by_id(self.master_id)
        if self.isProject:
            group = Project.get_by_id(self.groupId)
        else:
            group = Course.get_by_id(self.groupId)
        data = {
                'groupId' : self.groupId,
                'message' : self.message,
                'date' : {
                    'year': self.msgDate.year,
                    'month': self.msgDate.month,
                    'day': self.msgDate.day,
                    'hour': self.msgDate.hour,
                    'minute': self.msgDate.minute
                },
                'id' : self.key().id(),
                'master_id' : self.master_id,
                'isProject' : self.isProject,
                'user': json.loads(user.to_JSON()),
                'group': json.loads(group.to_JSON())
        }
        return json.dumps(data)