def deleteItem(category_id, item_id):
if 'email' in login_session.keys():
delete_item = models.item_get(item_id)
user_id = models.getUserID(login_session['email'])
if delete_item.user_id == user_id:
form = forms.deleteForm(request.form)
category = models.category_get(category_id)
user = models.getUserInfo(user_id)
if request.method == 'POST':
models.item_delete(delete_item)
return redirect(url_for('showCategory',
category_id=category.id))
else:
return render_template('items/delete.html', category=category,
item=delete_item, user=user, form=form)
else:
flash("You aren't the owner for that.")
return redirect(url_for('showCategory', category_id=category_id))
else:
return redirect(url_for('showCategory', category_id=category_id))
def post_delete():
delete_postid = deleteForm()
if delete_postid.validate_on_submit():
pid = delete_postid.pid.data
p = Post.query.filter_by(id=pid).first()
if p is None:
flash(u'该帖子不存在')
else:
comments = Comment.query.filter_by(post_id=p.id).all() # 删除评论
for comment in comments:
db.session.delete(comment)
all_user = User.query.all() # 删除收藏的关系
for user in all_user:
if p in user.collectionPost.all():
user.collectionPost.remove(p)
else:
pass
db.session.delete(p)
db.session.commit()
flash(u'帖子及其评论删除成功')
return render_template('trade/post_delete.html', form=delete_postid)
def deleteCategory(category_id):
if 'email' in login_session.keys():
# Get the category to be deleted out of the DB.
delete_category = models.category_get(category_id)
user_id = models.getUserID(login_session['email'])
if delete_category.user_id == user_id:
form = forms.deleteForm(request.form)
if request.method == 'POST':
# Delete the category out of the DB.
models.category_delete(delete_category)
# Redirect to the index page.
return redirect(url_for('index'))
else:
# If the route is requested via GET render the delete page.
user = models.getUserInfo(user_id)
return render_template('categories/delete.html',
category=delete_category, user=user,
form=form)
else:
flash("You aren't the owner for that.")
return redirect(url_for('index'))
else:
return redirect(url_for('index'))
