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
来源:简书