Flask 实战笔记02 - SQLAlchemy实现mysql的应用

概述

对web的路由完成后，我们开始对数据库的增删改查进行进一步的实践和探索，flask使用SQLAlchemy进行对数据库的操作，这里我们以Mysql数据库为例。

使用SQLAlchemy需要安装数据库+驱动包两个组件，安装 Flask-SQLAlchemy + PyMySQL 两个组件，会使我们的开发变的更加简洁和方便。

database类的定义

对于Flask的所有应用和插件而言，都需要注册在App实力中，使用对象进行操作，先定义数据库操作的基类，让其他的模块注册在Base中。

Base中写入所有DB中的公共方法，以减少重复造轮子的过程。

Base 中实例化SQLAlchemy

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()


class Base(db.Model):
    # 忽略基类的主键
    __abstract__ = True

普通数据类继承Base类，以User类为例,使用之前引入db包

from app.models.base import Base


class User(Base):
    __tablename__ = 'user'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(50), nullable=False, default="")
    email = Column(String(120), nullable=False, default="")
    _password = Column('password',String(64))

    @property
    def password(self):
        return self._password

    @password.setter
    def password(self, raw):
        self._password = generate_password_hash(raw)

入口文件加载DB

在加载数据时需要加载数据库初始化的配置，使用SQLALCHEMY_DATABASE_URI指定，有指定的格式。

SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://root:123456@127.0.0.1:3306/admin?charset=utf8'

SQLALCHEMY_DATABASE_URI =数据库+驱动名称://用户名:密码@ip地址:端口号/数据库名称

def create_app():
    app = Flask(__name__)
    
    app.config.from_object('app.config.secure')

    # 初始化数据库
    db.init_app(app)
    with app.app_context():
        db.create_all()

    return app

说明和解释一下with关键字的作用，在python中主要使用的是栈数据结构来运行App实例，with关键字能很好的区分上下文，当资源断开时会自动释放和回收，对程序能启到优化的作用。

mysql的驱动可以使用cymysql,也可以使用pymysql，网上使用pymysql的教程会偏多一点，使用驱动时，如果拿不定主意去github上使用stat数多的插件，选大众型的，这样解决方案也会多一点。

模块中定义方法

model中可以定义模型，常量，直接操作数据库的原子方法，也可以使用db.session的形式来读取数据。

from sqlalchemy import Column, Integer, String, SmallInteger

from app.models.base import Base, db


class tp_manager(Base):

    STATUS_NORMAL = 1
    STATUS_STOP = 0
    
    # ...
    
    @classmethod
    def get_manager_by_name(cls, username):
        r = cls.query.filter(cls.username == username, cls.status == cls.STATUS_NORMAL).one()
        return r
        
        
    @classmethod
    def get_db(cls):
        r = db.session.query(cls).filter(cls.status == cls.STATUS_NORMAL).all()
        return r    

在路由中的使用

在路由中使用时，需要引入models中对应的model包，下面是一个简单使用的Demo，用户的权限可以使用falsk-login组件来存储用户信息。

from flask import request, session, jsonify

from app.libs.BaseRoutes import BaseRoutes
from app.validators.form.login import LoginForm
from app.models.admin.tp_manager import tp_manager

api = BaseRoutes('login')


@api.route('/login/loginHandle', methods=['POST'])
def loginHandle():
    form = LoginForm(request.form)
    if form.validate():
        # 验证通过的逻辑
        r = tp_manager.get_manager_by_name(form.username.data)
        if r:
            msg = '操作成功'
            session['manager_id'] = r.id
        else:
            msg = '账号和密码错误'

        return jsonify({"code": 200, "data": [], "msg": msg})
    else:
        # 错误暂时忽略...
        return form.errors

顺便写一句，flask在使用session前，在配置文件中需要配置SECRET_KEY,不然会报错,key值自定义即可。

SECRET_KEY = '需要加密的key值'

就写到这里，下一篇我们再见。