flaskで簡単なログイン機能をつけたい

Question

作りたいもの

・予めＩＤを用意しておく
・ログインフォームからそのＩＤを入力し、一致したら次の画面に遷移
・ＩＤが一致しなかったらエラーメッセージ表示
・sqlはつかわないで、if文だけで作りたいです。

と言うようなセキュリティはガバガバで簡単なログイン機能を作りたいです。

書いたコード

こちらのmain.pyを実行します。

from flask import Flask, request,render_template

app = Flask(__name__)


password = "aaa"
@app.route('/')
def main():
    message = "Passwordを入力して下さい"
    return render_template("main.html", message = message,password=password)

@app.route("/login", methods=["POST"])
def login():
    password = request.form["password"]
    message = "Loginしました"
    if password == "1234":
        return render_template("login.html",message = message)
    else:
        return render_template('main.html')


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

次にtemplatesのログインフォーム画面です。
こちらが、main.htmlです。

{% extends "layout.html" %}
{% block content %}
<h1>Flask Login</h1>
<p>{{ message }}</p>
<form action="/login" method="post">
    <label for="Password">Password</label>
    <input type="password" name = "pass">
    <button type="submit">送信する</button>
</form>
{% endblock %}

次にlogin.htmlでログイン後の画面です。

{% extends "layout.html" %}
{% block content %}
    <h1>書き込みました</h1>
    <p>{{ message }}</p>
    <form action="/" method="get">
        <button type="submit">戻る</button>
    </form>
{% endblock %}

それぞれlayout.htmlから形を整えています。

解決したいこと

エラー文に対する解決策と、他の考え方などお願いします。

どうかアドバイスやコードへのレビューをお願いしたいです。

Comments

[nico25]

英語が苦でなければ（私は苦ですが笑）、
ログイン機能付きのブログアプリの
公式のチュートリアルがあります。
難しいかもしれないですが、とても良いチュートリアルです。
https://flask.palletsprojects.com/en/1.1.x/tutorial/

有料なら Udemy に日本語の動画があります。
こちらもログイン機能付きのブログアプリです。
おそらく公式チュートリアルを参考にして
作られたものだと思います。
まずはここからステップアップして
いってもいいのかなと思います。
Udemy はセールだと 80% off くらいになるので
その時を狙ってください笑
https://www.udemy.com/course/flask_intro/

[oinari03]

すみません。今回求めているのはDBやSQLを使わないやり方ですので、後このチュートリアル難しくて、、、理解できなかったです。。。

Answer 1

ポイント

session, g, flash の 3 つ import して使います。

• セッションには session を使います。
• g と session は区別するといいと思います。
• 認証の成功と失敗のメッセージは flash を使うといいと思います。

まだ書きかけですが...

• Flask の session と g

ログアウト

ログアウトは実装していないので、ログアウトを実装すると練習になるかなと思います。
sessison から 'user_name' を消すとログアウトします。

del session['user_name']

コード

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

app = Flask(__name__)
app.config['SECRET_KEY'] = "your_secret_string"


class User:
    def __init__(self, name, password):
        self.name = name
        self.password = password
        # 本来、平文でのパスワードは原則 NG です。
        # パスワード、ソルトなどで検索してください。


users = [
    User('aaa', 'aaa123'),
    User('bbb', 'bbb123'),
]


@app.before_request
def load_logged_in_user():
    user_name = session.get('user_name')
    if user_name is None:
        g.user = None
        return

    users_ = [user for user in users if user.name == user_name]
    if len(users_) != 1:
        g.user = None
        return

    g.user = users_[0]


def login_required(view):
    def wrapped_view(**kwargs):
        if g.user is None:
            flash('login required')
            return redirect(url_for('index'))
        return view(**kwargs)
    return wrapped_view


@app.route('/', methods=["GET", "POST"])
def index():
    if request.method == 'POST':
        user_name = request.form['user_name']
        password = request.form['password']
        users_ = [user for user in users if user.name == user_name and user.password == password]
        if users_:
            user = users_[0]
            session['user_name'] = user.name
            flash('You were successfully logged in')
            return redirect('profile')
        else:
            flash('login failure')

    return render_template("index.html")


@app.route("/profile", methods=["GET"])
@login_required
def profile():
    return render_template('profile.html')


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

<!-- index.html -->
{% with messages = get_flashed_messages() %}
  {% if messages %}
    <ul>
    {% for message in messages %}
      <li>{{ message }}</li>
    {% endfor %}
    </ul>
  {% endif %}
{% endwith %}

<form action="/" method="post">
  <p>user name: <input type="text" name="user_name" size="40" /></p>
  <p>password:  <input type="password" name="password" size="40" /></p>
  <p>
    <input type="submit" value="submit" />
    <input type="reset" value="reset" />
  </p>
</form>

<!-- profile.html -->
{% with messages = get_flashed_messages() %}
  {% if messages %}
    <ul>
    {% for message in messages %}
      <li>{{ message }}</li>
    {% endfor %}
    </ul>
  {% endif %}
{% endwith %}

Answer 2

以下の部分に判定がないのが原因です。
IDの判定式を入れましょう。

@app.route("/login", methods=["POST"])
def login():
    message = "Loginしました"
    return render_template("login.html",message = message)

あらかじめIDを用意するとか書いてありますが、HTMLやpythonを見る限りPasswordしか無いようです。
用意するIDというのは、プログラム中ではPasswordなのですか？(inputのnameはarticleですし・・・)

個人的にはif文をFlaskで書くやり方が出来れば良いのではないかとおもっています。

ご推察の通りです。
if文で書けば判定可能です。
if文で固定の判定式を書けばDBは不用です。