index_site_virgool.py

index_site_virgool.py

@app.route('/', methods=('GET', 'POST'))
def index():
    conn = sqlite3.connect('database.db')
    conn.row_factory = sqlite3.Row
    if request.method == 'POST':
        url = request.form['url']
        if not url:
            flash('The URL is required!')
            return redirect(url_for('index'))
        url_data = conn.execute('INSERT INTO urls (original_url) VALUES (?)',
        (url,))  # Write URL data On DB
        conn.commit()
        conn.close()
        url_id = url_data.lastrowid
        hashid = hashids.encode(url_id)
        short_url = request.host_url + hashid
        return render_template('index.html', short_url=short_url)
    return render_template('index.html')