Flaskのログイン機能で、ログイン後にも関わらずUnauthorizedが発生してしまう。

Question

現在Flaskを用いてソースコードのメモアプリを作成中です。今回教えて欲しいこととしては、Flaskのログイン機能に関してです。

具体的には、ユーザを新規登録してから、ログインするとメモアプリのトップページに進む仕様になっており、そこからメモの変更や削除、新規登録をする際に、「Unauthorized」と出てしまいます。

恐らく、＠login_requiredが原因かと思われるのですが、トップページにも追加しているので、ログイン自体は出来ていると思われます。また、「Unauthorized」となった際にフォームを再送信するときちんとメモの更新、削除が行われており、メモアプリとしては機能しています。

以下、実際のメモアプリと、ソースコードになります。
https://memoprogram.herokuapp.com/login

from crypt import methods
from email.policy import default
from enum import unique
from flask import Flask
from flask import render_template, request, redirect
from flask_sqlalchemy import SQLAlchemy
from flask_login import UserMixin, LoginManager, login_user, login_required, logout_user
from flask_bootstrap import Bootstrap

from werkzeug.security import generate_password_hash, check_password_hash
from datetime import datetime
import os
import pytz

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///memo.db'
app.config['SECRET_KEY'] = os.urandom(24)
db = SQLAlchemy(app)
bootstrap = Bootstrap(app)

login_manager = LoginManager()
login_manager.init_app(app)

class Post(db.Model): #データベース定義
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(50), nullable=False)
    url = db.Column(db.String(100), nullable=True)
    body = db.Column(db.String(500), nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, default=datetime.now(pytz.timezone('Asia/Tokyo')))

class User(UserMixin, db.Model): #データベース定義
    id = db.Column(db.Integer, primary_key=True)
    user_name = db.Column(db.String(20), nullable=False, unique=True)
    password = db.Column(db.String(30), nullable=False)

@login_manager.user_loader
def load_user(user_id): #セッション情報取得
    return User.query.get(int(user_id))



@app.route("/", methods=["GET", "POST"])
@login_required #デコレータ追加
def index():
    if request.method == "GET":
        posts = Post.query.all() #Post内の全てのデータをリスト形式で取得
    return render_template("index.html", posts=posts)

@app.route("/signup", methods=["GET", "POST"])
def signup():
    if request.method == "GET":
        return render_template("signup.html")
    else:
        user_name = request.form.get("user_name") #formでPOSTされたデータを取得
        password = request.form.get("password")

        user = User(user_name=user_name, password=generate_password_hash(password, method='sha256')) #パスワードをハッシュ化

        db.session.add(user) #データベースに追加
        db.session.commit() #コミットしないと追加、保存されない
        return redirect("/login") #ログイン画面へリダイレクト

@app.route("/login", methods=["GET", "POST"])
def login():
    if request.method == "GET":
        return render_template("login.html")
    else:
        user_name = request.form.get("user_name") #formでPOSTされたデータを取得
        password = request.form.get("password")

        if (user_name == None) or (password == None):
            return redirect('/login')
        else:
            user = User.query.filter_by(user_name=user_name).first() #ユーザ名を取ってくる
            if check_password_hash(user.password, password): #パスワードハッシュがあっている場合
                login_user(user) #userにログイン
                return redirect("/") #初期画面へリダイレクト

@app.route("/logout")
@login_required #デコレータ追加
def logout():
    logout_user()
    return redirect("/login")

@app.route("/create", methods=["GET", "POST"])
@login_required #デコレータ追加
def create():
    if request.method == "GET":
        return render_template("create.html")
    else:
        title = request.form.get("title") #formでPOSTされたデータを取得
        url = request.form.get("url")
        body = request.form.get("body")

        post = Post(title=title, url=url, body=body)

        db.session.add(post) #データベースに追加
        db.session.commit() #コミットしないと追加、保存されない
        return redirect("/") #初期画面へリダイレクト

@app.route("/<int:id>/update", methods=["GET", "POST"])
@login_required #デコレータ追加
def update(id): #post.idがidに入る
    post = Post.query.get(id) #Post内の特定のデータをリスト形式で取得
    if request.method == "GET":
        return render_template("update.html", post=post)
    else:
        post.title = request.form.get("title") #formでPOSTされたデータを取得
        post.url = request.form.get("url")
        post.body = request.form.get("body")

        db.session.commit() #更新するときはコミットのみでOK
        return redirect("/") #初期画面へリダイレクト

@app.route("/<int:id>/delete", methods=["GET"])
@login_required #デコレータ追加
def delete(id): #post.idがidに入る
    post = Post.query.get(id) #Post内の特定のデータをリスト形式で取得
    
    db.session.delete(post) #データベースのデータを削除
    db.session.commit()
    return redirect("/") #初期画面へリダイレクト

Comments

[kajiaja]

heroku logs --tailでエラー内容を見ると以下のようになっていました。

2022-02-19T06:05:00.695855+00:00 app[web.1]: [2022-02-19 06:05:00,694] ERROR in app: Exception on /login [POST]
2022-02-19T06:05:00.695863+00:00 app[web.1]: Traceback (most recent call last):
2022-02-19T06:05:00.695864+00:00 app[web.1]: File "/app/.heroku/python/lib/python3.9/site-packages/flask/app.py", line 2073, in wsgi_app
2022-02-19T06:05:00.695864+00:00 app[web.1]: response = self.full_dispatch_request()
2022-02-19T06:05:00.695865+00:00 app[web.1]: File "/app/.heroku/python/lib/python3.9/site-packages/flask/app.py", line 1518, in full_dispatch_request
2022-02-19T06:05:00.695865+00:00 app[web.1]: rv = self.handle_user_exception(e)
2022-02-19T06:05:00.695866+00:00 app[web.1]: File "/app/.heroku/python/lib/python3.9/site-packages/flask/app.py", line 1516, in full_dispatch_request
2022-02-19T06:05:00.695866+00:00 app[web.1]: rv = self.dispatch_request()
2022-02-19T06:05:00.695867+00:00 app[web.1]: File "/app/.heroku/python/lib/python3.9/site-packages/flask/app.py", line 1502, in dispatch_request
2022-02-19T06:05:00.695867+00:00 app[web.1]: return self.ensure_sync(self.view_functions[rule.endpoint])(**req.view_args)
2022-02-19T06:05:00.695867+00:00 app[web.1]: File "/app/app.py", line 75, in login
2022-02-19T06:05:00.695868+00:00 app[web.1]: if check_password_hash(user.password, password): #パスワードハッシュがあっている場合
2022-02-19T06:05:00.695868+00:00 app[web.1]: AttributeError: 'NoneType' object has no attribute 'password'

Answer 1

ログに以下の記録があります。

2022-02-19T06:05:00.695867+00:00 app[web.1]: File "/app/app.py", line 75, in login
2022-02-19T06:05:00.695868+00:00 app[web.1]: if check_password_hash(user.password, password): #パスワードハッシュがあっている場合
2022-02-19T06:05:00.695868+00:00 app[web.1]: AttributeError: 'NoneType' object has no attribute 'password'

「AttributeError: 'NoneType' object has no attribute 'password'」とあり、「NoneTypeのオブジェクトにはpasswd属性がない」というメッセージが出ています。

直前に「if check_password_hash(user.password, password):」とあり、ここで「passwd」属性にアクセスしている「user」オブジェクトが「NoneType」になっているようです。

おそらく、その前の行「user = User.query.filter_by(user_name=user_name).first()」で「user」に「None」が格納されています。この行の後に「user」が「None」でないかなどを確認し、「None」だったら適切に処理する（ログインページにリダイレクトするなど）ようなコードを入れると良さそうに見えます。

Comments

[kajiaja]

ご回答ありがとうございます。
実際に、userの中身が何もないのか確認するために、sqliteで直接Userクラスのデータベースを確認したところ、ユーザ登録で登録したはずのuser_nameやpasswordがデータベース内に存在していませんでした。
以下、sqliteで調べた際のデータの中身です。

sqlite> .schema user
CREATE TABLE user (
id INTEGER NOT NULL,
user_name VARCHAR(20) NOT NULL,
password VARCHAR(30) NOT NULL,
PRIMARY KEY (id),
UNIQUE (user_name)
);
sqlite> select user_name from user;
asdfghj

このasdfghjというのは、vscodeのデバッグ機能を利用してエラーを解決しようとした際にローカルで登録したuser_nameになります。

自分で調べたところ、ローカル環境ではログイン後にUnauthorizedとならず、またデータベースに正しくデータが保存されるのですが、サーバー上ではログイン後もUnauthorizedと出ており、その原因が恐らくデータベースに正しく保存されていないため　AttributeError: 'NoneType' object has no attribute 'password'　というエラーを吐いたのだと思われます。

なので、厚かましいことは重々承知しているのですが、サーバー上で正しくデータベースに保存されない理由をもしご存知でしたら、教えていただきたいです。

[msiz07]

以下のような、サーバー上でデータベースにデータを登録した状況が分からないと、情報不足で回答が難しいです。

- サーバ環境でのDBファイル作成やユーザー登録はどのように行ったか（開発環境のSQLiteのDBファイルをそのままアップロードしたか、サーバで稼働させたFlaskアプリを使って登録したのか、など）
- DB作成やデータ登録に関するログ出力など、DBファイル作成やユーザー登録処理で実際に発生した状況の詳細情報

ご質問の件については「解決済み」になったようなので、サーバー上でデータベースに保存されない現象の状況を確認したうえで、自分で解決できないようでしたら別の質問を投稿するとよいのではないかと思います。

[kajiaja]

ご回答ありがとうございます！
自分で調査し、無理ならまた質問を投稿したいと思います。