プログラミング学習

【Python】ログイン機能を作る「Flask-Login」⑥

投稿日:

「ログイン機能を作る」も【Part6】となりました。
前回は「ログインできる機能」を追加しました。
今回は「データベース」の設定を追加していこうと思います。

データベースを使っていこう

前回は、決められた「名前」&「メールアドレス」でのみ、ログインできる仕様になっています。

ここから「登録してあるアカウントでのログイン」,「新規アカウント作成」この2つの機能を作っていきましょう。

「データベース」を操作するために「SQLAlchemy」を使っていきます。
コードが長くなってきていますがご了承ください。

SQLAlchemyをインポートしよう

「flask_sqlalchemy」から「SQLAlchemy」をインポートしていきます。

from flask import Flask, render_template, redirect
from flask_login import LoginManager, login_required, UserMixin, login_user, logout_user
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField

------------------------------------------------

from flask_sqlalchemy import SQLAlchemy

------------------------------------------------

app = Flask(__name__)
login_manager = LoginManager()
login_manager.init_app(app)
app.config['SECRET_KEY'] = "secret"

class User(UserMixin):
  def __init__(self,user_id):
    self.id = user_id

class LoginForm(FlaskForm):
  name = StringField('名前')
  mail = StringField('メールアドレス')
  submit = SubmitField('ログイン')

@login_manager.user_loader
def load_user(user_id):
  return User(user_id)

@app.route('/')
def index():
  return render_template('top.html')

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

@app.route('/login', methods=['GET','POST'])
def login():
  form = LoginForm()
  if form.validate_on_submit():
    if form.name.data == 'ANDoblog' and form.mail.data == 'test@mail':
      user = User(form.name.data)
      login_user(user)
      return redirect('/member')
    else:
      return 'ログインに失敗しました'
  return render_template('login.html',form=form)

@app.route('/logout')
def logout():
  logout_user()
  return render_template('logout.html')

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

「SQLAlchemy」についてはこちら。

使用するデータベースを指定しよう

今回は「login.db」と名付けた「dbファイル」を使っていきます。

from flask import Flask, render_template, redirect
from flask_login import LoginManager, login_required, UserMixin, login_user, logout_user
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
login_manager = LoginManager()
login_manager.init_app(app)
app.config['SECRET_KEY'] = "secret"

--------------------------------------------------------

db_uri = 'sqlite:///login.db'
app.config['SQLALCHEMY_DATABASE_URI'] = db_uri
db = SQLAlchemy(app)

--------------------------------------------------------

class User(UserMixin):
  def __init__(self,user_id):
    self.id = user_id

class LoginForm(FlaskForm):
  name = StringField('名前')
  mail = StringField('メールアドレス')
  submit = SubmitField('ログイン')

@login_manager.user_loader
def load_user(user_id):
  return User(user_id)

@app.route('/')
def index():
  return render_template('top.html')

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

@app.route('/login', methods=['GET','POST'])
def login():
  form = LoginForm()
  if form.validate_on_submit():
    if form.name.data == 'ANDoblog' and form.mail.data == 'test@mail':
      user = User(form.name.data)
      login_user(user)
      return redirect('/member')
    else:
      return 'ログインに失敗しました'
  return render_template('login.html',form=form)

@app.route('/logout')
def logout():
  logout_user()
  return render_template('logout.html')

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

Userクラスを変更しよう

前回記述した「Userクラス」の変更を行います。
データベースに格納するモデルを書いていきます。
今回は「id」,「name」,「mail」の3つを作ります。

from flask import Flask, render_template, redirect
from flask_login import LoginManager, login_required, UserMixin, login_user, logout_user
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
login_manager = LoginManager()
login_manager.init_app(app)
app.config['SECRET_KEY'] = "secret"

db_uri = 'sqlite:///login.db'
app.config['SQLALCHEMY_DATABASE_URI'] = db_uri
db = SQLAlchemy(app)

----------------------------------------------

class User(UserMixin, db.Model):
    __tablename__ = 'User'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.Text())
    mail = db.Column(db.Text())
    def __init__(self, name, mail):
      self.name = name
      self.mail = mail

db.create_all()

----------------------------------------------

class LoginForm(FlaskForm):
  name = StringField('名前')
  mail = StringField('メールアドレス')
  submit = SubmitField('ログイン')

@login_manager.user_loader
def load_user(user_id):
  return User(user_id)

@app.route('/')
def index():
  return render_template('top.html')

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

@app.route('/login', methods=['GET','POST'])
def login():
  form = LoginForm()
  if form.validate_on_submit():
    if form.name.data == 'ANDoblog' and form.mail.data == 'test@mail':
      user = User(form.name.data)
      login_user(user)
      return redirect('/member')
    else:
      return 'ログインに失敗しました'
  return render_template('login.html',form=form)

@app.route('/logout')
def logout():
  logout_user()
  return render_template('logout.html')

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

最後に「db.create_all()」でテーブルを作成しています。

「user_loader」を変更しよう

データベースから「ユーザーの情報」を渡してあげるように変更します。

from flask import Flask, render_template, redirect
from flask_login import LoginManager, login_required, UserMixin, login_user, logout_user
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
login_manager = LoginManager()
login_manager.init_app(app)
app.config['SECRET_KEY'] = "secret"

db_uri = 'sqlite:///login.db'
app.config['SQLALCHEMY_DATABASE_URI'] = db_uri
db = SQLAlchemy(app)

class User(UserMixin, db.Model):
    __tablename__ = 'User'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.Text())
    mail = db.Column(db.Text())
    def __init__(self, name, mail):
      self.name = name
      self.mail = mail

db.create_all()

class LoginForm(FlaskForm):
  name = StringField('名前')
  mail = StringField('メールアドレス')
  submit = SubmitField('ログイン')

----------------------------------------------

@login_manager.user_loader
def load_user(user_id):
  return User.query.get(int(user_id))

----------------------------------------------

@app.route('/')
def index():
  return render_template('top.html')

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

@app.route('/login', methods=['GET','POST'])
def login():
  form = LoginForm()
  if form.validate_on_submit():
    if form.name.data == 'ANDoblog' and form.mail.data == 'test@mail':
      user = User(form.name.data)
      login_user(user)
      return redirect('/member')
    else:
      return 'ログインに失敗しました'
  return render_template('login.html',form=form)

@app.route('/logout')
def logout():
  logout_user()
  return render_template('logout.html')

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

まとめ

今回は「データベース」の設定を追加しました。
ここからデータベースにアカウントを登録して、ログインできるかを試していきます。
次回は「ログイン処理」のところを変更しないといけませんね。
それではまた。

-プログラミング学習
-

執筆者:


comment

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

関連記事

【Python】Flask-WTFを使ってみよう③

前回「Flask-WTF」を使って「フォーム」の作成を行いました。今回は「文字数」の入力を制限する「バリデーション」を設定していきたいと思います。 バリデーションを設定してみよう バリデーションについ …

【WTForms】IntegerFieldを使ってみる

今回は「wtforms」の「IntegerField」を使ってみたいと思います。 IntegerFieldを使ってみる 「Integer」とは「整数」という意味です。なので「整数」のみ受け取るフォーム …

【Python】プログラムを実行する方法

「プログラムを実行してください。」と書かれていて、実行する方法がわからない人はいますか?今回は「テキストエディタ」の中で、プログラムを実行する方法を紹介します。 プログラムを実行する方法 今回、使用し …

【エクセル】で「ちょこっとプログラミング」応用編#1

プログラミングをしたことのない人向けにExcelでできる「ちょこっとプログラミング」を紹介します。今回は「空白のセルまで値を取得する」から【応用】として「セルの値で新規ファイルを作成する」を題材にやっ …

【VSCode】テキストの一括置換をしてみよう

コードを書いていて、「変数」を変更したいときに「文字を1つ1つ直した」ことはありますか?今回は「VSCode」の中で使える「テキストの一括置換」について紹介します。 VSCodeで一括置換してみよう …