Exemplo n.º 1
0
def addItem():

    if 'username' not in login_session:
        return redirect('/login')

    categories = session.query(Category).all()
    new_item = CategoryItem()

    if request.method == 'POST':
        if request.form['name']:
            new_item.name = request.form['name']
        if request.form['description']:
            new_item.description = request.form['description']
        if request.form['category']:
            new_item.category_id = request.form['category']

        new_item.user_id = login_session['user_id']

        session.add(new_item)
        session.commit()
        flash("Category item created!")

        category = session.query(Category).filter_by(
            id=new_item.category_id).one()

        return redirect(url_for('categoryItemPage',
                                category_name=category.name,
                                categoryitem_name=new_item.name))
    else:
        return render_template('addcategoryitem.html',
                               categories=categories)