def addCategory(request):
if request.method == 'POST':
form = categoryForm(request.POST)
if form.is_valid():
title = request.POST.get("title", "")
slug = request.POST.get("title", "")
cat_obj = Category(title = title, slug = slug)
cat_obj.save()
return HttpResponseRedirect(reverse('index'))
else:
messages.error(request, "Error")
return render(request, 'blog/create_category.html', {
'form': categoryForm()
})
def editCategory(category_id):
# Adding logged in user verification to every route here on out.
if 'email' in login_session.keys():
# Get the category out of the DB.
edit_category = models.category_get(category_id)
# Let's make sure this user is the cateogory owner.
user_id = models.getUserID(login_session['email'])
if edit_category.user_id == user_id:
# Get the form out of the form module.
form = forms.categoryForm(request.form)
# If the form is submitted via POST and is validated:
if request.method == 'POST' and form.validate():
# Update the category with the form data
edit_category.name = form.name.data
edit_category.image = form.image.data
edit_category.description = form.description.data
# Send the updated category back to the DB.
models.category_edit(edit_category)
# Redirect to the index page.
return redirect(url_for('index'))
else:
# If the route is requested via GET render the edit page.
user = models.getUserInfo(user_id)
return render_template('categories/edit.html',
category=edit_category,
form=form, user=user)
else:
flash("You aren't the owner for that.")
return redirect(url_for('index'))
else:
return redirect(url_for('index'))
def newCategory():
# If the user is logged in:
if 'email' in login_session.keys():
# Get the form for categories out of the forms module.
form = forms.categoryForm(request.form)
user_id = models.getUserID(login_session['email'])
user = models.getUserInfo(user_id)
# If the form is submitted via POST and is validated:
if request.method == 'POST' and form.validate():
# Create a new category object to store all data from the form.
new_category = {
"name": form.name.data,
"image": form.image.data,
"description": form.description.data,
"user_id": models.getUserID(login_session['email'])
}
# Pass that object to the DB via the models module.
models.category_new(new_category)
# Redirect to the index page.
return redirect(url_for('index'))
else:
# If the route is requested via GET, render the new category page.
return render_template('categories/new.html', form=form, user=user)
else:
return redirect(url_for('index'))
def new_category(request):
form = categoryForm()
return render_to_response('blog/create_category.html', {
'categories': Category.objects.all(),
'form': form,
})
