def follow(self): db.following.save(vars(self)) # Notification for the Followee notification = Notification(uid=self.followee) notification.make_follow(follower_id=self.follower) notification.save() return self
def add_like_to_sheet(sheet_id, uid):
"""
Add uid as a liker of sheet_id.
"""
db.sheets.update({"id": sheet_id}, {"$addToSet": {"likes": uid}})
sheet = get_sheet(sheet_id)
notification = Notification(uid=sheet["owner"])
notification.make_sheet_like(liker_id=uid, sheet_id=sheet_id)
notification.save()
