【Python】Flask + MySQL 实现用户注册，登录和登出

今天用Flask + MySQL 实现用户注册，登录和登出。
一、实战场景
Flask 框架实现用户的注册，登录和登出。
二、主要知识点

• flask_login 插件使用
  
• SQLAlchemy 基础操作
  
• 用户基础类设计
  
• Flask 读取配置文件
  
• 蓝图注册与使用
  
• wtforms 表单提交数据
  
• wtforms 表单验证
  
• Bootstrap 集成
  
• Jinjia2 模版继承
  

涉及的知识点和细节很多，我下面就直接贴出注册部分的核心代码。
三、核心代码
马上安排！
1、应用初始化 MySQL 和 flask_login 模块

1. '''
2. Description: 创建应用程序，并注册相关蓝图
3. Python学习群 279199867 加群免费领取软件工具、视频教程、电子书、源码、课件，视频相关源码、福利文件赠送，学习问题交流共同进步！
5. '''
7. from flask import Flask
8. from base.base_model import db
9. from flask_login import LoginManager
11. # 登录插件
12. login_manager = LoginManager()
14. def register_auth_blueprint(app):
15.     # 注册蓝图
16.     from app.auth import auth_bp
17.     app.register_blueprint(auth_bp)
19. def create_app(config=None):
20.     # 创建应用
21.     app = Flask(__name__)
23.     # 加载配置
24.     app.config.from_object('config')
26.     # 注册 SQLAlchemy
27.     db.init_app(app)
28.     #
29.     # 注册 login 模块
30.     login_manager.init_app(app)
32.     # 未登录时候的默认跳转页面
33.     login_manager.login_view = 'auth.login'
34.     # # login_manager.login_message = '请先登录或注册'
36.     register_auth_blueprint(app)
38.     if config is not None:
39.         if isinstance(config, dict):
40.             app.config.update(config)
41.         elif config.endswith('.py'):
42.             app.config.from_pyfile(config)
44.     return app
46. app = create_app()
48. with app.app_context():
49.     db.create_all()
51. if __name__ == '__main__':
52.     # 如果要使用 vscode 调试，需要将 debug 设置为 False，否则无法命中请求断点
53.     app.run(host='0.0.0.0', debug=True)

复制代码

 
2、设置配置文件

1. APP_NAME = "north"
3. SECRET_KEY = "fNqh2TNw3l0Dj8ZCMQyQh7m1YvWVSgDx"
5. DEBUG = True
7. SQLALCHEMY_DATABASE_URI = 'mysql://username:password@ip:3306/dbname'
9. # 设置sqlalchemy自动更跟踪数据库
10. SQLALCHEMY_TRACK_MODIFICATIONS = True
12. # 查询时会显示原始SQL语句
13. SQLALCHEMY_ECHO = True

复制代码

 
3、蓝图初始化

1. from flask import Blueprint
3. auth_bp = Blueprint(
4.     'auth',
5.     __name__,
6. )
8. from app.auth.user import user, auth

复制代码

 
4、编写注册表单

1. <main class="form-signin w-100 m-auto">
2. <form action="https://www.cnblogs.com/{{ url_for('auth.register') }}" method="post">
3.     <img class="mb-4" src="https://www.cnblogs.com/{{ url_for('static', filename="3rd/images/bootstrap-logo.svg") }} " alt="" width="72" height="57">
4.     <h1 class="h3 mb-3 fw-normal">注册信息</h1>
6.    
7.       <input class="form-control"
8.              id="nickname" name="nickname"
9.               value="{{ form.data['nickname'] | default('',true) }}"
10.               placeholder="昵称">
11.       <label for="floatingInput">昵称</label>
12.    
13.    
14.       <input type="email" class="form-control"
15.              id="email" name="email"
16.               value="{{ form.data['email'] | default('',true) }}"
17.               placeholder="Email">
18.       <label for="floatingInput">邮箱</label>
19.    
20.    
21.       <input type="password" class="form-control"
22.               id="password" name="password"
23.               value="{{ form.data['password'] | default('',true) }}"
24.               placeholder="Password">
25.       <label for="floatingPassword">密码</label>
26.    
27.    
28.       <input type="password" class="form-control"
29.               id="confirm_password" name="confirm_password"
30.               value="{{ form.data['confirm_password'] | default('',true) }}"
31.               placeholder="Confirm Password">
32.       <label for="floatingPassword">确认密码</label>
33.    
34.         {% if form and form.errors %}
35.             {% for key, error in form.errors.items() %}
36.                 {{ key }} : {{ error }}
37.             {% endfor %}
38.         {% endif %}
40.     <button class="w-100 btn btn-lg btn-primary" type="submit">注册</button>
41.     <p class="mt-5 mb-3 text-muted">秀儿 © 2017–2022</p>
42.   </form>
43. </main>

复制代码

 
5、提交注册表单

1. @auth_bp.route("/register", methods=['POST', 'GET'])
2. def register():
3.     # 注册逻辑
4.     form = RegisterForm(request.form)
6.     # 检查
7.     if request.method == 'POST' and form.validate():
8.         # 执行正确逻辑
9.         user = User()
10.         user.set_attrs(form.data)
12.         user.name = user.nickname
13.         user.token = user.generate_token()
15.         db.session.add(user)
16.         db.session.commit()
18.         # 执行登录
19.         login_user(user, False)
21.         return redirect(url_for('auth.home'))
23.     return render_template("auth/register.html", form=form)

复制代码

 
6、用户模型

1. import random
3. from sqlalchemy import Column, ForeignKey, func
4. from sqlalchemy import String, Unicode, DateTime, Boolean
5. from sqlalchemy import TIMESTAMP, Integer, Float
6. from flask_login import login_user, login_required, logout_user, current_user, UserMixin
7. from werkzeug.security import generate_password_hash, check_password_hash
10. from common.helpers.str_helper import random_string
11. from north import login_manager
13. from  base.base_model import BaseModel
15. class User(BaseModel, UserMixin):
16.     # UserMixin 继承属性
17.     __tablename__ = 'users'
19.     # 表基础值
20.     phone_number = Column(String(16), unique=True)
21.     email = Column(String(64), unique=True, nullable=False)
22.     token = Column(String(64))
23.     password = Column('password', String(100))
24.     status = Column(Integer, default=1)
25.     type = Column(Integer, default=1)
27.     # 定义一个对象属性，对应表中的 password 字段
28.     _password = Column('password', String(100))
30.     @property
31.     def password(self):
32.         # 定义属性，使用对象属性赋值
33.         return self._password
35.     @password.setter
36.     def password(self, raw):
37.         # 属性赋值
38.         self._password = generate_password_hash(raw)
40.     def check_password(self, raw):
41.         # 检查密码
42.         if not self._password:
43.             return False
44.         return check_password_hash(self._password, raw)
46.     def generate_token(self, expiration=60000):
47.         # 生成 token
48.         return random_string(32)
51. @login_manager.user_loader
52. def get_user(uid):
53.     # 必须, login 插件制定方法
54.     return User.query.get(int(uid))

复制代码

 
7、模型基类

1. import pymysql
2. import datetime
4. from flask_sqlalchemy import SQLAlchemy
5. from sqlalchemy import Column, Integer, SmallInteger
6. from sqlalchemy import String, Unicode, DateTime, Boolean
8. # 初始化数据库类型
9. pymysql.install_as_MySQLdb()
10. db = SQLAlchemy()
12. # 模型基础类
13. class BaseModel(db.Model):
14.     __abstract__ = True
16.     id = Column(Integer, primary_key=True)
17.     name = Column(String(32), nullable=False)
18.     nickname = Column(String(32), nullable=False)
19.     is_enable = Column(SmallInteger, default=1, nullable=False)
20.     created_at = db.Column(db.DateTime, default=datetime.datetime.now)
21.     updated_at = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now)
22.     deleted_at = db.Column(db.DateTime)
24.     def __init__(self):
25.         pass
27.     # 字典赋值, 场景: 表单提交
28.     def set_attrs(self, attrs):
29.         for key, value in attrs.items():
30.             if hasattr(self, key) and key != 'id':
31.                 setattr(self, key, value)

复制代码

 
8、表单验证

1. from wtforms import StringField, PasswordField, Form, validators
2. # from wtforms.validators import Length, Email, \
3. #     ValidationError, EqualTo
4. from app.auth.user.user_model import User
6. class RegisterForm(Form):
7.     nickname = StringField('昵称',
8.         validators = [
9.             validators.DataRequired(),
10.             validators.Length(2, 32)
11.             #   validators.Email(message='电子邮箱不符合规范')
12.         ])
13.     email = StringField('电子邮件',
14.         validators = [
15.             validators.DataRequired(),
16.             validators.Length(10, 50)
18.             #   validators.Email(message='电子邮箱不符合规范')
19.         ])
20.     password = PasswordField('密码', [
21.         validators.DataRequired(),
22.         validators.EqualTo('confirm_password', message='密码需要一致')
23.     ])
24.     confirm_password = PasswordField('Repeat Password', [
25.         validators.DataRequired(),
27.     ])
29.     def validate_email(self, field):
30.         # 自定义验证，命名对应
31.         if User.query.filter_by(email=field.data).first():
32.             raise validators.ValidationError('邮件已被注册')
34.     def validate_nickname(self, field):
35.         if User.query.filter_by(nickname=field.data).first():
36.             raise validators.ValidationError('昵称已存在')

复制代码

 
9、代码主要目录结构

1. ├── app
2. │   ├── __init__.py
3. │   ├── auth
4. │   │   ├── __init__.py
5. │   │   └── user
6. │   └── tools
7. │       ├── __init__.py
8. │       └── db_tools.py
9. ├── base
10. │   ├── __init__.py
11. │   ├── base_blueprint.py
12. │   ├── base_form.py
13. │   └── base_model.py
14. ├── common
15. │   ├── __init__.py
16. │   └── helpers
17. │       ├── __init__.py
18. │       └── str_helper.py
19. ├── config.py
21. ├── north.py

复制代码

四、运行结果
1、注册和验证

2、注册成功登录

3、登录

兄弟们，今天的分享就到这里，下次见！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！