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

posted @ 2019-06-11 10:25  yin_zhaozhao  阅读(443)  评论(0编辑  收藏  举报