def remove_message(): user = current_user.user if request.values.get('all') == 'true': mongo.db.messages.delete_many({'user_id': user['_id']}) elif request.values.get('id'): msg_id = ObjectId(request.values.get('id')) mongo.db.messages.delete_one({'_id': msg_id}) return jsonify(models.BaseResult())
def add(post_id=None): posts_form = forms.PostsForm() if posts_form.is_submitted(): if not posts_form.validate(): return jsonify(models.BaseResult(1, str(posts_form.errors))) utils.verify_num(posts_form.vercode.data) user = current_user.user if not user.get('is_active', False) or user.get('is_disabled', False): return jsonify(code_msg.USER_UN_ACTIVE_OR_DISABLED) user_coin = user.get('coin', 0) if posts_form.reward.data > user_coin: return jsonify(models.R.ok('悬赏金币不能大于拥有的金币,当前账号金币为:' + str(user_coin))) posts = { 'title': posts_form.title.data, 'catalog_id': ObjectId(posts_form.catalog_id.data), # 'is_closed': False, 'content': posts_form.content.data, } post_index = posts.copy() post_index['catalog_id'] = str(posts['catalog_id']) msg = '发帖成功!' reward = posts_form.reward.data if post_id: posts['modify_at'] = datetime.now() mongo.db.posts.update_one({'_id': post_id}, {'$set': posts}) msg = '修改成功!' else: posts['create_at'] = datetime.utcnow() posts['reward'] = reward posts['user_id'] = user['_id'] # 扣除用户发帖悬赏 if reward > 0: mongo.db.users.update_one({'_id': user['_id']}, {'$inc': {'coin': -reward}}) mongo.db.posts.save(posts) post_id = posts['_id'] # 更新索引文档 update_index(mongo.db.posts.find_one_or_404({'_id': post_id})) return jsonify(models.R.ok(msg).put('action', url_for('index.index'))) else: ver_code = utils.gen_verify_num() # session['ver_code'] = ver_code['answer'] posts = None if post_id: posts = mongo.db.posts.find_one_or_404({'_id': post_id}) title = '发帖' if post_id is None else '编辑帖子' return render_template('jie/add.html', page_name='jie', ver_code=ver_code['question'], form=posts_form, is_add=(post_id is None), post=posts, title=title)
def user_set(): if request.method == 'POST': include_keys = ['username', 'avatar', 'desc', 'city', 'sex'] data = request.values update_data = {} for key in data.keys(): if key in include_keys: update_data[key] = data.get(key) # print(update_data) mongo.db.users.update({'_id': current_user.user['_id']}, {'$set': data}) return jsonify(models.BaseResult()) return render_template('user/set.html', user_page='set', page_name='user', title='基本设置')