Python - Flask 项目封装数据库连接

这里仅针对 Flask 项目

1. 安装

安装 flask-sqlalchemy

pip install flask-sqlalchemy

安装 pymysql

pip install pymysql

2. 项目 - main.py

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
import pymysql

pymysql.install_as_MySQLdb()  # 解决 “No module named 'MySQLdb'” 的错误

# 使用集成方式处理SQLAlchemy
# 示例  mysql://root:XXX@localhost:3306/XXX?charset=utf8
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://数据库用户:密码@IP:端口/数据库名?charset=编码'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False  # True：更新数据库的修改，及时发送信号

# 实例化db对象
db = SQLAlchemy(app)

3. 提取数据库连接 - database.py

from sqlalchemy import MetaData

def dbconnect():
    from main import db
    dbsession = db.session
    DBase = db.Model
    metadata = MetaData(bind=db.engine)
    return (dbsession, metadata, DBase)

4. 使用示例

dbsession, md, DBase = dbconnect()

class Article(DBase):
	'''数据表对应的模型类'''
    __table__ = Table('article', md, autoload=True)

    def find_all(self):
    	'''全查'''
        result = dbsession.query(Article).all()