软件测试学习笔记丨Flask操作数据库-数据库和表的管理

本文转自测试人社区，原文链接：https://ceshiren.com/t/topic/23427

结构分析

• Runner是中间件
• sqlachemy去驱动Runner，根据不同的数据库去驱动不同的中间件

demo

# 导入Flask的类
from flask import Flask
# 实例化 Flask的类，并且绑定module
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import *

app = Flask(__name__)

# mysql数据库库用户名
username = "root"
# mysql 数据库密码
pwd = "123456"
# mysql 数据库host地址
ip = "127.0.0.1"
# mysql 数据库端口
port = "3306"
# 代码使用的数据库名
database = "first"
# 设置mysql 链接方法是
app.config['SQLALCHEMY_DATABASE_URI'] = f'mysql+pymysql://{username}:{pwd}@{ip}:{port}/{database}?charset=utf8'
# 定义应用使用数据库的配置
# 设置SQLALCHEMY_TRACK_MODIFICATIONS参数 不设置该配置的时候会抛出警告
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
# 将app与flask-SQLAlchemy的db进行绑定
db = SQLAlchemy(app)


# 每个类表示一张表
class User(db.Model):
    # 创建字段
    id = Column(Integer, primary_key=True)
    username = Column(String(80))


if __name__ == '__main__':
    # 创建表
    with app.app_context():
        db.create_all()
    # 删除表
    # db.drop_all()

• 可以结合yaml来使用，将数据库的连接配置信息放在yaml文件中，代码里去读取他

import yaml
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
# 实例化app 对象
from sqlalchemy import *

app = Flask(__name__)
with open("./data.yml") as f :
    result = yaml.safe_load(f)
    username = result.get("database").get('username')
    password = result.get("database").get('password')
    server = result.get("database").get('server')
    db = result.get("database").get('db')
app.config['SQLALCHEMY_DATABASE_URI'] = \
    f"mysql+pymysql://{username}:{password}@{server}/{db}?charset=utf8"
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
# SQLAlchemy 绑定app
db = SQLAlchemy(app)


# 定义数据库的表 需要继承 db.Model，db 为 app 启动的时的 SQLAlchemy 绑定的实例
class User(db.Model):
    id = Column(Integer, primary_key=True)
    username = Column(String(80))


if __name__ == '__main__' :
    db.create_all()

注意：

• 创建表和删除表要添加一行

with app.app_context():

示例：

# 创建表
with app.app_context():
    db.create_all()

• 类名相当于表名
• 驼峰命名的类名，会被转化成下划线，例如UserInfo会被转化成user_info
• 自定义表名：

__tablename__ = "testcase"