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='基本设置')
