【Python】ログイン機能を作る「Flask-Login」⑥
「ログイン機能を作る」も【Part６】となりました。
前回は「ログインできる機能」を追加しました。
今回は「データベース」の設定を追加していこうと思います。
データベースを使っていこう
前回は、決められた「名前」＆「メールアドレス」でのみ、ログインできる仕様になっています。
ここから「登録してあるアカウントでのログイン」,「新規アカウント作成」この２つの機能を作っていきましょう。
「データベース」を操作するために「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」の３つを作ります。
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)まとめ
今回は「データベース」の設定を追加しました。
ここからデータベースにアカウントを登録して、ログインできるかを試していきます。
次回は「ログイン処理」のところを変更しないといけませんね。
それではまた。
