使用Flask和Python的Heroku Postgres数据库

时间:2017-05-04 20:56:58

标签: python database postgresql heroku flask

我正在为课堂写一个虚拟网站,而且我在将Heroku数据库连接到我当地的应用程序时遇到了麻烦,直到我推送到Heroku。

我不确定这样做的正确方法是什么,而且我搜索了很多视频/论坛,我似乎无法从他们那里得到直接答案。我会在下面发布一些代码。在dbconnect.py中插入heroku数据库凭据,如URI,host等?

#app.py
from flask import Flask, render_template, redirect, url_for, request, session, flash
from functools import wraps


app = Flask(__name__)

app.secret_key = "Gundam"

# login required decorator
def login_required(f):
    @wraps(f)
    def wrap(*args, **kwargs):
        if 'logged_in' in session:
            return f(*args, **kwargs)
        else:
            flash('You need to login first.')
            return redirect(url_for('login_page'))
    return wrap


@app.route('/')    
def homepage():
    return render_template("main.html")

@app.route('/dashboard/')
@login_required
def dashboard():
    return render_template("dashboard.html")    

@app.errorhandler(404)
def page_not_found(e):
    return render_template("404.html")

@app.route('/login/', methods=["GET", "POST"])    
def login_page():    
    error = ''
    try:
        if request.method == "POST":
            attempted_username = request.form['username']
            attempted_password = request.form['password']

            if attempted_username == "admin" and attempted_password == "password":
                session['logged_in'] = True
                flash('You were just logged in!')
                return redirect(url_for('dashboard'))
            else:
                error = "Invalid Username or Password."    
        return render_template("login.html", error=error)        
    except Exception as e:    
        return render_template("login.html", error=error)

@app.route('/logout/')
def logout():
    session.pop("logged_in", None)        
    return redirect(url_for('homepage'))



if __name__ == '__main__':
    app.run(debug = True)


dbconnect.py

import os
import psycopg2
import urlparse

urlparse.uses_netloc.append("postgres")
url = urlparse.urlparse(os.environ[""])

conn = psycopg2.connect(
    database=url.path[1:],
    user=url.username,
    password=url.password,
    host=url.hostname,
    port=url.port
)

2 个答案:

答案 0 :(得分:2)

您必须先在heroku中安装postgres数据库插件。在计算机中运行heroku工具箱并输入命令heroku addons:create heroku-postgresql:hobby-dev。 Hobby-dev是免费版。

添加Heroku Postgres后,应用配置中将提供DATABASE_URL设置,其中包含用于访问新配置的Heroku Postgres服务的URL。将值用作数据库uri。可以从仪表板访问应用配置。在设置下,点击Reveal config vars。您也可以使用toolbelt命令。见heroku config -h

现在你可以这样做:

url = urlparse.urlparse(os.environ["DATABASE_URL"])

有关详细信息,请参阅https://devcenter.heroku.com/articles/heroku-postgresql

答案 1 :(得分:0)

只需在您的python应用程序上使用以下代码段即可。那应该可以解决问题。

import os
import psycopg2


DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')