def manage_visit_record():
mongo_operator = MongoOperator(**MongoDB_CONFIG)
# 获取用户的uid
uid = current_user.id
# 查询询该用户的日程安排
generator = mongo_operator.find({'user_id': uid, 'status': 1}, 'visit_record').sort("date", -1)
return render_template('visit_record/index.html', visited_records=list(generator))
def make_template_context():
# 如果登录，则尝试拉取未读信息
unread_msg = 0
if current_user.is_authenticated:
uid = current_user.id
mongo_operator = MongoOperator(**MongoDB_CONFIG)
unread_msg = mongo_operator.find(
{
"to_id": uid,
"state": Message.UNCHECKED
}, "message").count()
return dict(unread_msg=unread_msg)
