Python UserRepo.get_user_by_usernameの例

コード例 #1

ファイル: test_user.py プロジェクト: sirrah23/GTDApp

def test_get_user_by_username(resource):
    UserRepo.add_user(username="******",
                      password="******",
                      email="*****@*****.**")
    users = UserRepo.get_all_users()
    assert len(users) == 1
    uid = users[0].id
    stored_user = UserRepo.get_user_by_username("user1")
    assert stored_user.id == uid
    assert stored_user.username == "user1"
    assert stored_user.email == "*****@*****.**"
    not_stored_user = UserRepo.get_user_by_username(
        "randomuser")  #should not be in the database
    assert not_stored_user == None
    resource

コード例 #2

ファイル: index.py プロジェクト: sirrah23/GTDApp

def login():
    if current_user and current_user.is_authenticated:
        return redirect("/home")
    error = None
    if request.method == "POST":
        user = UserRepo.get_user_by_username(request.form["username"])
        if UserRepo.verify_user(request.form["username"], request.form["password"]):
            user = LoggedInUserWrapper(user)
            login_user(user)
            return redirect("/home")
        else:
            error = "Invalid Credentials. Please try again."
    return render_template("login.html", error=error)