Пример #1
0
def add_skill():
    form = SkillForm(request.form)
    if request.method == 'POST' and form.validate():
        skill_name = form.skill_name.data

        skills = skill_name.split(',')
        #print(skills,file=sys.stderr)
        # Create Cursor
        cur = mysql.connection.cursor()

        # Get user id
        result = cur.execute("SELECT id FROM users WHERE username = %s",
                             [session['username']])
        user_id = cur.fetchone()
        print(user_id['id'], file=sys.stderr)

        # Execute
        size = len(skills)
        for i in range(0, size):
            cur.execute(
                "INSERT INTO skills (skill_name, has_skill) VALUES(%r, %s)",
                [skills[i], user_id['id']])

        # Commit to DB
        mysql.connection.commit()

        #Close connection
        cur.close()

        flash('Skill Added', 'success')

        return redirect(url_for('dashboard'))

    return render_template('add_skill.html', form=form)