网站少不了要和数据库打交道，归根到底都是一些增删改查操作，这里做一个简单的用户登录功能来学习一下flask如何操作mysql。
用到的一些知识点：flask-sqlalchemy、flask-login、flask-wtf、pymysql
这里通过一个完整的登录实例来介绍，程序已经成功运行，在未登录时拦截了success.html页面跳转到登录页面，登录成功后才能访问success。
以下是项目的整体结构图：
首先是配置信息，配置了数据库连接等基本的信息，config.py
|
1
2
3
4
|
debug = true sqlalchemy_echo = false sqlalchemy_database_uri = 'mysql+pymysql://root:1011@localhost/rl_project?charset=utf8'secret_key = '*\xff\x93\xc8w\x13\x0e@3\xd6\x82\x0f\x84\x18\xe7\xd9\\|\x04e\xb9(\xfd\xc3' |
common/_init_.py
|
1
2
3
4
5
6
|
# config=utf-8 from flask_sqlalchemy import sqlalchemy __all__ = ['db'] db = sqlalchemy() |
数据库配置类，common/data.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
# config=utf-8 from sqlalchemy import create_engine from sqlalchemy.sql import text from config import sqlalchemy_database_uri, sqlalchemy_echo def db_query(sql, settings=none, echo=none): if settings is none: settings = sqlalchemy_database_uri if echo is none: echo = sqlalchemy_echo return create_engine(settings, echo=echo).connect().execute(text(sql)).fetchall() def db_execute(sql, settings=none, echo=none): if settings is none: settings = sqlalchemy_database_uri if echo is none: echo = sqlalchemy_echo return create_engine(settings, echo=echo).connect().execute(text(sql)).rowcount |
sqlalchemy_database_uri用于连接数据的数据库。
sqlalchemy_echo如果设置成 true，sqlalchemy 将会记录所有 发到标准输出(stderr)的语句，这对调试很有帮助。
当然，我们在setting中设置了基本的连接数据库信息，启动时加载app = create_app('../config.py')，所以这个类删掉也不会报错。
form/login_form.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
# config=utf-8 from flask_wtf import flaskform as form from wtforms import stringfield, passwordfield from wtforms.validators import datarequired class loginform(form): accountnumber = stringfield('accountnumber', validators=[datarequired('accountnumber is null')]) password = passwordfield('password', validators=[datarequired('password is null')]) |
使用flask-wtf做登录的表单验证，这里简单做了账号密码不为空如，当我们不填写密码时，点击登录：
model/_init_.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# config=utf-8 from flask import flask from flask_login import loginmanager from common import db login_manager = loginmanager() login_manager.login_view = "user.login"def create_app(config_filename=none): app = flask(__name__) login_manager.init_app(app) if config_filename is not none: app.config.from_pyfile(config_filename) configure_database(app) return app def configure_database(app): db.init_app(app) |
其中，login_manager.login_view = "user.login" 指定了未登录时跳转的页面，即被拦截后统一跳到user/login这个路由下model/user_model.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
# config=utf-8 from flask_login import usermixin from common import db class user(db.model, usermixin): user_id = db.column('id', db.integer, primary_key=true) accountnumber = db.column(db.string(200), unique=true) password = db.column(db.string(50), unique=true) name = db.column(db.string(20), unique=true) __tablename__ = 'tb_user' def __init__(self, user_id=none, account_number=none, password=none, name="anonymous"): self.user_id = user_id self.accountnumber = account_number self.password = password self.name = name def is_authenticated(self): return true def is_active(self): return true def is_anonymous(self): return false def get_id(self): return unicode(self.user_id) def __repr__(self): return '<user %r>' % (self.accountnumber) |
这里需要注意：
|
1
2
|
def get_id(self): return unicode(self.user_id) |
该方法不可缺少，否则会报：notimplementederror: no `id` attribute - override `get_id`错误。
get_id()
返回一个能唯一识别用户的，并能用于从 user_loader 回调中 加载用户的 unicode 。注意着 必须 是一个 unicode ——如果 id 原本是 一个 int 或其它类型，你需要把它转换为 unicode 。
is_authenticated()
当用户通过验证时，也即提供有效证明时返回 true
is_active()
如果这是一个通过验证、已激活、未被停用的账户返回 true 。
is_anonymous()
如果是一个匿名用户，返回 true 。
login.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#encoding:utf-8 #!/usr/bin/env python from flask import render_template, request, redirect, flask, blueprint from flask_login import login_user, login_required from model.user_model import user from model import login_manager from form.login_form import loginform userroute = blueprint('user', __name__, url_prefix='/user', template_folder='templates', static_folder='static') @login_manager.user_loader def load_user(user_id): return user.query.get(int(user_id)) @userroute.before_request def before_request(): pass @userroute.route('/success') @login_requireddef index(): return render_template('success.html') @userroute.route('/login', methods=['get', 'post']) def login(): form = loginform() if request.method == 'post': if not form.validate_on_submit(): print form.errors return render_template('login.html', form=form) user = user.query.filter(user.accountnumber == form.accountnumber.data, user.password == form.password.data).first() if user: login_user(user) return render_template('success.html') return render_template('login.html', form=form) |
其中，要实现一个load_user()回调方法，这个回调用于从会话中存储的用户 id 重新加载用户对象，id存在则返回对应的用户对象，不存在则返回none。
有些操作是需要用户登录的，有些操作则无需用户登录，这里使用到了@login_required，在每一个需要登录才能访问的路由方法上加@login_required即可。
启动类，runserver.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
# config=utf-8 from login import userroute from model import create_app default_modules = [userroute] app = create_app('../config.py') for module in default_modules: app.register_blueprint(module) @app.before_request def before_request(): passif __name__ == '__main__': app.run(debug=true) |
default_modules = [userroute]是将userroute蓝图注册入app，才能启动login中的userroute路由，我们在login.py中使用了蓝图：userroute = blueprint('user', __name__, url_prefix='/user', template_folder='templates', static_folder='static')
|
1
2
3
4
5
|
@app.before_request def before_request(): pass |
这是一个全局的方法，在请求开始之前被调用，在某些场景下做一些提示或者特殊处理，当然这里没用上，直接pass，这个方法去掉对项目没有影响。基本样式模版，base.html
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
<!doctype html> <html lang="cn"> <head> <meta charset="utf-8"> <title>{% block title %}{% endblock %}</title> <script src="{{ url_for('static', filename='jquery1.42.min.js') }}"></script> {% block head %}{% endblock %} </head> <body> {% block content %}{% endblock %} </body> </html> |
登录前台页面，login.html
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
{% extends "base.html" %} {% block title %}python flask user page{% endblock %} {% block head %} <style type="text/css"></style> {% endblock %} {% block content %} <form action="{{ url_for('user.login') }}" method="post"> {% if form.errors %} <ul> {% for name, errors in form.errors.items() %} {% for error in errors %} <li>{{ error }}</li> {% endfor %} {% endfor %} </ul> {% endif %} 账号：{{ form.accountnumber(size=20) }}<label>{{ form.accountnumber.errors[0] }}</label><br/> 密码：<input name="password" type="password"/><br/> {{ form.hidden_tag() }} <button type="submit">登录</button> </form> {% endblock %} |
到此，一个flask实现简单登录功能就做完了。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://www.cnblogs.com/javazhiyin/p/9034468.html
