コード例 #1
0
ファイル: main.py プロジェクト: j0hn/snipptr
def before_request():
    """
    Ran before any HTTP request.
    Connects the database.
    """

    g.db = database
    g.db.connect()

    try:
        g.user = [x for x in User.select().where(id=session["user_id"])].pop()
    except:
        g.user = None
コード例 #2
0
ファイル: snipptr.py プロジェクト: j0hn/snipptr
def login():
    if g.user:  # Allready logged in
        return redirect(url_for("index"))

    error = None
    if request.method == "POST":
        username = request.form.get("username")
        password = request.form.get("password")

        try:
            user = [x for x in User.select().where(username=username, password=password)].pop()
            session["user_id"] = user.id
            return redirect(url_for("index"))
        except Exception, e:
            print e
            error = "Invalid username or password"