python ORM框架：SqlAlchemy

ORM,对象关系映射，即Object Relational Mapping的简称，通过ORM框架将编程语言中的对象模型与数据库的关系模型建立映射关系，这样做的目的：简化sql语言操作数据库的繁琐过程（原生sql的编写及拼接等），转而直接使用对象模型来操作数据库做替代

第一部分

SqlAlchemy本身无法直接操作数据库，它是建立在第三方数据库API（如python 中的pymysql库）之上，应用程序调用对象模型进行增删改查等操作时，将对象转化成sql语句，然后再通过API调用执行已经转换好的sql语句

安装

  pip install sqlalchemy

  pip install pymysql #这里笔者使用的数据API是pymysql

应用

- 配置及创建数据库引擎

SqlAlchemy 支持间接调用多种数据库API，根据不能的配置文件调用不同的数据库API

#常见配置文件：即database url，创建数据库引擎需要

MySQL-Python：mysql+mysqldb://<user>:<password>@<host>[:<port>]/<dbname>

pymysql：mysql+pymysql://<username>:<password>@<host>/<dbname>[?<options>]

MySQL-Connector：mysql+mysqlconnector://<user>:<password>@<host>[:<port>]/<dbname>

cx_Oracle：oracle+cx_oracle://user:pass@host:port/dbname[?key=value&key=value...]

- 创建数据库引擎

* 注意 *：带上charset=utf8参数，防止中文乱码

#初始化数据库连接

from sqlalchemy import create_engine

url="mysql+pymysql://[账号]:[密码]@[主机]:[端口]/[数据库]?charset=utf8" #这里笔者使用的mysql数据库，pymysql API与数据库交互,添加的charset可防止中文乱码

engine=create_engine(url,echo=False,encoding="utf-8")  #url为配置文件，echo调试参数，值为为true打印整个sql执行过程

- 创建会话（session）

通过sessionmaker工厂方法，我们得到一个类，默认返回Session类，也可以自定义Session类
方法：
sessionmaker( bind=None, class_=Session, autoflush=True,autocommit=False,expire_on_commit=True,info=None, **kw)

#示例1：直接调用sessionmaker

from sqlalchemy.orm import sessionmaker

SessionClass=sessionmaker(bind=engine)#利用工厂模式获取SessionClass

session_obj=SessionClass() #创建session对象,此时已绑定数据库引擎，但是未关联任何的对象模型

#示例1：使用scoped_session

from sqlalchemy.orm import scoped_session

SessionClass=scoped_session(sessionmaker(bind=engine))#利用工厂模式获取SessionClass

session_obj=SessionClass() #创建session对象,此时已绑定数据库引擎，但是未关联任何的对象模型

scoped_session VS Session
Session：多次创建的Session对象是不同的
scoped_session：首先通过sessionmaker工厂创建Session对象，然后对Session对象进行相应的管理(先在Registry中找之前是否创建过Session，若没有，则创建并注册，有，则直接返回)，这样的目的：同一个线程维护一个session对象，保证了线程的安全性

- 与数据库交互

常用操作：与sql语言一致，主要是增删改查，接下来就简要概述这4大类

******* 增 *******

session.add(object) #在数据库中增加一个对象实例

session.add_all([object]) #在数据库中增加多个对象实例,参数为列表形式

session.commit() #提交，否则未入口

******* 查 *******

#备注：以下均已object代表对象模型，object.prop代表对象object的prop属性

session.query(object) #查询object对应的关系表，相当于select * from tables

session.query(object).first() #查询结果取第一条，没有返回none

session.query(object).filter(object.prop) #条件查询，filter相当于where

session.query(object).order_by(object.prop) #排序，默认升序，降序用desc

session.query(object).order_by(object.prop.desc()).limit(10) #降序，及限制10条

session.query(object.prop.label("别名")).filter(object.prop.like("%同同mony")) #模糊查询及给字段取别名

#使用聚合函数，sum、count等

from sqlalchemy import func

session.query(func.sum(object.prop).label("别名")

******* 改 *******

session.query(object).filter(object.prop>20).update({"prop1": 'values'})#批量更新，若没有过滤条件，这更新表中所有记录

myuser = Session.query(object).filter(object.prop>20).first()

object.prop2= 'value2' #单条更新，可直接修改对象

session.commit()

******* 删 *******

res = session.query(object).filter(object.prop>20).delete()

session.commit()

第二部分

掌握了基本概念和使用，为了更高效、更快捷将关系表转化成对象，不得不提sqlacodegen工具
a.安装
pip install sqlacodegen
b.使用
  sqlacodegen url [opts]
    这里的url就是sqlalchemy中create_engine使用的参数，但是当账号密码含有特殊字符时，同样的url，在sqlacodegen命令中会报错，识别不出账号、密码、端口等信息，此时可通过给账号和密码加引号解决
    opts：
       --tables TABLES 指定表，默认将所有表转化成对象
       --noindexes 忽略索引
       --noviews 忽略视图
       --noclasses 不生成类，仅生成表
       --outfile OUTFILE 输出文件
示例：
sqlacodegen --noviews --noconstraints --noclasses --noindexes --outfile 对应py文件路径 url【同create engine处使用的】
注意：当账号和密码包含特殊字符时，需要用引号，否则自动识别不出账号、密码、端口等信息

示例1：

#不使用 `--noclasses`导出的对象

from sqlalchemy import BIGINT, Column, DateTime, Float, String, text

from sqlalchemy.dialects.mysql.types import TINYINT

from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

metadata = Base.metadata

class TDeviceActivate(Base):

    __tablename__ = 't_device_activate'

    id = Column(BIGINT(10), nullable=False)

    user_id = Column(BIGINT(10), nullable=False)

    device_id = Column(String(32), primary_key=True, nullable=False)

    mac = Column(String(30))

    uuid = Column(String(32))

    firmware_type = Column(String(16), primary_key=True, nullable=False)

    activate_time = Column(DateTime, nullable=False)

    create_date = Column(DateTime, nullable=False)

    ip_address = Column(String(50))

    longitude = Column(Float(10))

    latitude = Column(Float(10))

    expires_time = Column(DateTime)

    type = Column(TINYINT(10), server_default=text("'1'"))

示例2

#使用 `--noclasses`参数导出的对象

from sqlalchemy import BIGINT, Column, DateTime, Float, MetaData, String, Table, text

from sqlalchemy.dialects.mysql.types import TINYINT

metadata = MetaData()

t_t_device_activate = Table(

    't_device_activate', metadata,

    Column('id', BIGINT(10), nullable=False),

    Column('user_id', BIGINT(10), nullable=False),

    Column('device_id', String(32), primary_key=True, nullable=False),

    Column('mac', String(30)),

    Column('uuid', String(32)),

    Column('firmware_type', String(16), primary_key=True, nullable=False),

    Column('activate_time', DateTime, nullable=False),

    Column('create_date', DateTime, nullable=False),

    Column('ip_address', String(50)),

    Column('longitude', Float(10)),

    Column('latitude', Float(10)),

    Column('expires_time', DateTime),

    Column('type', TINYINT(10), server_default=text("'1'"))

)

总结

在UI回归测试脚本中，通常涉及到与数据库打交道，开始时候主要使用pymysql API 与数据交互，但是随着使用频率的增加，直接书写sql变得繁琐，便尝试使用orm的方式

转载来自：
作者：小蜗牛的成长
链接：https://www.jianshu.com/p/84986de8a903
来源：简书