def toDictionary(self, options=[], user=None): '''For packaging JSON objects Pass in list of VisionComment.Options.* for extra information ''' objs = [] commentList = self._commentModels if self.length() > 0: from Picture import Picture idToPicture = dict() if VisionComment.Options.PICTURE in options: pictureIds = set([c.pictureId() for c in self.comments()]) pictureIds.discard(0) pictures = Picture.getByIds(pictureIds) idToPicture = dict([(picture.id(), picture) for picture in pictures]) if VisionComment.Options.AUTHOR in options: authorIds = set([comment.authorId() for comment in self.comments()]) authors = DataApi.getUsersFromIds(authorIds) idToAuthor = dict([(u.id, u) for u in authors]) # If LIKES if VisionComment.Options.LIKES in options: commentIds = [c.id() for c in self.comments()] tuples = DataApi.getVisionCommentListLikeCount(commentIds) idToLikeCount = dict([(commentId, count) for commentId, count in tuples]) if user: commentUserLikes = DataApi.getVisionCommentIdsLikedByUser( commentIds, user.id()) for comment in commentList: obj = VisionComment(comment).toDictionary() if VisionComment.Options.AUTHOR in options: from User import User author = idToAuthor[comment.authorId] obj[VisionComment.Key.AUTHOR] = User(author).toDictionary() # If LIKES if VisionComment.Options.LIKES in options: obj[VisionComment.Key.LIKE] = \ { VisionComment.Key.LIKE_COUNT : idToLikeCount[comment.id] if comment.id in idToLikeCount else 0 } if user: obj[VisionComment.Key.LIKE]\ [VisionComment.Key.USER_LIKE] =\ comment.id in commentUserLikes if VisionComment.Options.PICTURE in options: if comment.pictureId > 0 and \ comment.pictureId in idToPicture: cp = idToPicture[comment.pictureId] obj[VisionComment.Key.PICTURE] = cp.toDictionary() objs.append(obj) return objs
def toDictionary(self, options=[], user=None): '''For packaging to JSON output. options input is list of Vision.Options when we want more information fetched from DB and placed into objects This batches up queries for pictures and comments across all the visions. Whenever we want to get all this data across a list of visions, it is good to use VisionList.toDictionary() instead of the individual toDictionary() calls on objects. 'user' parameter is only used when LIKES in provided as an option. If it is passed in, we also check whether this user likes the visions or not. ''' from User import User retObj = [] # Start collecting picture ids pictureIds = set() if Vision.Options.PICTURE in options: pictureIds = set([vision.pictureId() for vision in self.visions()]) # Get parent users if we need them parentUserIds = set() if Vision.Options.PARENT_USER in options: parentVisionIds = set([vision.parentId() for vision in self.visions()]) parentVisionIds.discard(0) parentVisions = VisionList.getByIds(parentVisionIds, allowRemovedVisions=True) idToParentVisions = dict([(v.id(), v) for v in parentVisions.visions()]) parentUserIds = set([parent.userId() for parent in parentVisions.visions()]) # Get all users (including parent users if needed) userIds = set([vision.userId() for vision in self.visions()]) users = User.getByUserIds(userIds.union(parentUserIds)) idToUser = dict([(u.id(), u) for u in users]) # If COMMENTS, get comments if Vision.Options.COMMENTS in options: commentList = VisionCommentList.getEmptyList() for vision in self.visions(): # TODO: Speed up performance of this some time! commentList.extend(vision.comments(5)) idToComments = {} for comment in commentList.comments(): if not comment.visionId() in idToComments: idToComments[comment.visionId()] = [comment] else: idToComments[comment.visionId()].append(comment) if Vision.Options.COMMENT_PICTURES: if comment.hasPicture(): pictureIds.add(comment.pictureId()) authorIds = set([comment.authorId() for comment in commentList.comments()]) authors = User.getByUserIds(authorIds) idToAuthor = dict([(author.id(), author) for author in authors]) # If COMMENT_LIKES if Vision.Options.COMMENT_LIKES in options: commentIds = [c.id() for c in commentList.comments()] tuples = DataApi.getVisionCommentListLikeCount(commentIds) idToCommentLikes = dict([(commentId, count) for commentId, count in tuples]) if user: commentUserLikes = DataApi.getVisionCommentIdsLikedByUser( commentIds, user.id()) # If LIKES, get vision likes in batch if Vision.Options.LIKES in options: visionIds = [v.id() for v in self.visions()] tuples = DataApi.getVisionListLikeCount(visionIds) idToLikes = dict([(visionId, count) for visionId, count in tuples]) if user: userLikes = DataApi.getVisionIdsLikedByUser(visionIds, user.id()) # Get pictures and hash them if (Vision.Options.PICTURE in options) or \ (Vision.Options.COMMENT_PICTURES in options): pictureIds.discard(0) pictures = Picture.getByIds(pictureIds) idToPicture = dict([(picture.id(), picture) for picture in pictures]) idToPicture[0] = "" # Now start building object list to return for vision in self.visions(): obj = vision.toDictionary() if Vision.Options.USER in options: obj[Vision.Key.USER] = idToUser[vision.userId()].toDictionary() # If PICTURE if Vision.Options.PICTURE in options: if vision.pictureId != 0: picture = idToPicture[vision.pictureId()].toDictionary() obj[Vision.Key.PICTURE] = picture # If COMMENTS if Vision.Options.COMMENTS in options: obj[Vision.Key.COMMENTS] = [] if vision.id() in idToComments: for comment in idToComments[vision.id()]: commentObj = comment.toDictionary() author = idToAuthor[comment.authorId()] commentObj[VisionComment.Key.AUTHOR] = author.toDictionary() if Vision.Options.COMMENT_LIKES in options: commentObj[VisionComment.Key.LIKE] = { VisionComment.Key.LIKE_COUNT: idToCommentLikes[comment.id()] if comment.id() in idToCommentLikes else 0 } if user: commentObj[VisionComment.Key.LIKE]\ [VisionComment.Key.USER_LIKE] = \ comment.id() in commentUserLikes if Vision.Options.COMMENT_PICTURES in options: if comment.hasPicture() and\ comment.pictureId() in idToPicture: cp = idToPicture[comment.pictureId()] commentObj[VisionComment.Key.PICTURE] =\ cp.toDictionary() obj[Vision.Key.COMMENTS].append(commentObj) # If PARENT_USER if Vision.Options.PARENT_USER in options: if not vision.isOriginalVision(): if vision.parentId() in idToParentVisions: parentVision = idToParentVisions[vision.parentId()] if parentVision.userId() in idToUser: parentUser = idToUser[parentVision.userId()] obj[Vision.Key.PARENT_USER] = \ parentUser.toDictionary() # If LIKES if Vision.Options.LIKES in options: obj[Vision.Key.LIKE] = { Vision.Key.LIKE_COUNT: idToLikes[vision.id()] if vision.id() in idToLikes else 0 } if user: obj[Vision.Key.LIKE][Vision.Key.USER_LIKE] = \ vision.id() in userLikes # finally append object to list retObj.append(obj) return retObj