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)