def removeFromFavorites(email, msgid): ''' Remove a message from the favorites list ''' try: user = models.Favorites.objects.get(email__exact=email) #Only attempt removal if message already exists in favorites if inFavorites(user, msgid): msg = MessageRenderer.getMessageByMessageID(msgid) user.messages.remove(msg) return "Message Removed From Favorites List" except Exception as ex: return ex
def addToFavorites(email, msgid): ''' Check whether user with given email exists in the Favorite's model i.e user need not be created ''' try: try: user = models.Favorites.objects.get(email__exact=email) except: user = models.Favorites(email=email) user.save() #Get MIMessage object if not inFavorites(user, msgid): msg = MessageRenderer.getMessageByMessageID(msgid) user.messages.add(msg) return "Message Added To Favorites List" except Exception as ex: return ex