数据持久化技术（Python）的使用

传统数据库连接方式：mysql（PyMySQL）
ORM 模型：SQLAlchemy MyBatis、 HibernatePyMySQL

安装：

代码语言：txt复制

pip install pymysql

简单使用

利用 pymysql.connect 建立数据库连接并执行 SQL 命令（需要提前搭建好数据库）：

代码语言：txt复制

import pymysql

db = pymysql.connect(
    # mysql 地址
    host='182.92.129.158',
    # 账号和密码
    user='tmp',
    password='ceshiren.com',
    # 数据库
    db='tmp',
    charset='utf8mb4'
)

if __name__ == '__main__':
    with db.cursor() as cursor:
        # 查看数据库中有多少表
        sql = "show tables;"
        # 执行 sql 语句
        cursor.execute(sql)
        # 查看所有数据
        print(cursor.fetchall())
        # 查询 name = aaaaaa 的数据
        sql = "select * from test_case_table where name=%s"
        cursor.execute(sql, ["aaaaaa"])
        print(cursor.fetchall())

代码语言：txt复制

(('test_case_table',),)
(('aaaaaa', '新的测试用例', 'test_hello.py', 'def test'),)

ORM

对象关系映射（ object-relational mapping）利用语言特性，操作数据库，比如对 Python 对象的操作，操作内容会映射到数据库里。

SQLALchemy 是 Python 编程语言下的一款 ORM 框架，该框架建立在数据库 API 之上，使用关系对象映射进行数据库操作。

安装

代码语言：txt复制

pip3 install SQLAlchemy

安装完成后可创建数据库连接：

代码语言：txt复制

engine = create_engine("mysql pymysql://tmp:ceshiren.com@182.92.129.158/tmp?charset=utf8",echo=True,)

1.echo: 当设置为 True 时会将 ORM 语句转化为 SQL 语句打印，一般 debug 的时候可用。

2.字段解释：

3.mysql pymysql：连接方式，采用 pymysql 。

4.tmp:ceshiren.com：用户名:密码。

5.182.92.129.158/tmp：数据库地址和数据库名称。

创建数据库

代码语言：txt复制

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import declarative_base

engine = create_engine("mysql pymysql://tmp:ceshiren.com@182.92.129.158/tmp?charset=utf8",
                       echo=True,
                       )
# 其子类将 Python 类和数据库表关联映射起来
Base = declarative_base()
# 继承 Base
class Users(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True)
    name = Column(String(64), unique=True)

    def __init__(self, name):
        self.name = name

if __name__ == '__main__':
    # 生成数据库表，如果有此库会忽略
    Base.metadata.create_all(engine)

declarative_base() 是 SQLALchemy 内部封装的一个方法，可以让其子类将 Python 类和数据库表关联映射起来。

增和查

SQLALchemy 使用 Session 用于创建程序和数据库之间的会话，通过 Session 对象可实现对数据的增删改查。

代码语言：txt复制

from sqlalchemy.orm import sessionmaker
# 创建session
Session = sessionmaker(bind=engine)
session = Session()
# 添加新数据
add_user = Users("student1")
# 提交
session.add(add_user)
session.commit()
# 查询
result = session.query(Users).filter_by(name="student1").first()
print(result.id, result.name)

上述代码新增数据后进行查询，结果如下：

代码语言：txt复制

1 student1

数据持久化技术就先介绍到这里，大家可以试着做一下练习，

我们后面会讲跨平台API对接，请持续关注哦~

原文链接

自动化测试

0 人点赞