def removeAnsVote(self, ansID):
USERS_COLLECTION.find_one_and_update(
{'_id': self.username}, {'$pull': {
'voted_ans': {
'ansID': ansID
}
}})
def updateAnsVote(self, ansID, vote):
USERS_COLLECTION.find_one_and_update(
{
'_id': self.username,
'voted_ans.ansID': ansID
}, {'$set': {
'voted_ans.$.vote': vote
}})
def removeQuesVote(self, quesID):
USERS_COLLECTION.find_one_and_update(
{'_id': self.username},
{'$pull': {
'voted_ques': {
'quesID': quesID
}
}})
def updateQuesVote(self, quesID, vote):
USERS_COLLECTION.find_one_and_update(
{
'_id': self.username,
'voted_ques.quesID': quesID
}, {'$set': {
'voted_ques.$.vote': vote
}})
def addQuesVote(self, quesID, vote):
USERS_COLLECTION.find_one_and_update(
{'_id': self.username},
{'$addToSet': {
'voted_ques': {
'quesID': quesID,
'vote': vote
}
}})
def addAnsVote(self, ansID, vote):
USERS_COLLECTION.find_one_and_update(
{'_id': self.username},
{'$addToSet': {
'voted_ans': {
'ansID': ansID,
'vote': vote
}
}})
def setBookmark(self, quesID):
bookmarked = (USERS_COLLECTION.find_one({'_id':
self.username}))['bookmarks']
if quesID in bookmarked:
USERS_COLLECTION.update_one({'_id': self.username},
{'$pull': {
'bookmarks': quesID
}})
return False
else:
USERS_COLLECTION.find_one_and_update(
{'_id': self.username}, {'$addToSet': {
'bookmarks': quesID
}})
return True
def update_answers(self, ansID):
USERS_COLLECTION.find_one_and_update({'_id': self.username}, {'$addToSet': {'ansPosted': ansID}})
def update_questions(self, quesID):
USERS_COLLECTION.find_one_and_update({'_id': self.username}, {'$addToSet': {'quesPosted': quesID}})
def update_karma(self, karma):
USERS_COLLECTION.find_one_and_update({'_id': self.username},
{'$inc': {
'karma': karma
}})
def update_comments(self, commentID):
USERS_COLLECTION.find_one_and_update(
{'_id': self.username},
{'$addToSet': {
'commentPosted': commentID
}})
