def set_user_qu_info(userid, qu_id, info):
has_qu = __get_qu(userid, qu_id) is not None
if not has_qu:
return User.update_by_id_add(
userid, {'qus': {
'qu_id': ObjectId(qu_id),
'info': info
}})
else:
return User.update_one(
{
"_id": ObjectId(userid),
"qus": {
'$elemMatch': {
"qu_id": ObjectId(qu_id)
}
}
}, {'$set': {
'qus.$.info': info
}})
def set_qu_data(userid, qu_id, qu_data):
has_qu = __get_qu(userid, qu_id) is not None
if not has_qu:
User.update_by_id_add(
userid, {'qus': {
'qu_id': ObjectId(qu_id),
'info': {
'auth': 1
}
}})
qu_report = Report.generate_report(qu_id, qu_data)
return User.update_one(
{
"_id": ObjectId(userid),
"qus": {
'$elemMatch': {
"qu_id": ObjectId(qu_id)
}
}
}, {'$set': {
'qus.$.qu_data': qu_data,
'qus.$.qu_report': qu_report
}})