下一篇文章:
Python-SQLAlchemy:第2节:查询条件设置
SQLAlchemy是Python编程语言下的一款开源软件。提供了SQL工具包及对象关系映射(ORM)工具,SQLAlchemy使用MIT许可证发行。它采用简单的Python语音,为高效和高性能的数据库访问设计,实现了完整的企业级持久模型。SQLAlchemy非常关注数据库的量级和性能。
本节通过一套例子分析SQLAlchemy的使用方法。
使用SQLAlchemy至少需要3部分代码,它们分别是定义表、定义数据库连接、进行增、删、改、查等逻辑操作。
定义表的实例:
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,Integer,String
Base=declarative_base()
class Accout(Base):
__tablename__=u'accout'
id=Column(Integer,primary_key=True)
user_namr=Column(String(50),nullable=False)
password=Column(String(200),nullable=False)
title=Column(String(50))
salary=Column(Integer)
def is_active(self):
#假设所有
return True
def get_id(self):
#返回账号ID,用方法返回属性值提高了表的封装性。
return self.id
def is_authenticated(self):
#假设已经通过验证
return True
def is_anonymous(self):
#具有登陆名和密码的账号不是匿名用户
return False
解析定义表的代码如下:
- SQLAlchemy表之前必须必须引入sqlalchemy.ext.declarative_base,并定义一个它的实例Base。所有表必须继承自Base。本例中定义了一个账户表类Account。
- 通过__tablename__属性定义了表在数据库中实际的名称account。
- 引入sqlalchemy包中的Column、Integer、String类型,因为需要用它们定义表中的列。本例在Account表中定义了5个列,分别是整型id和salary,以及字符串类型的user_name、password、title。
- 在定义列时可以通过给Column传送参数定义约束。本例中通过primary_key参数将id列定义主键,通过nullable参数将user__name和password定义非空。
- 在表中还可以自定义其他函数。本例中定义了用户验证时常用的几个函数:is__activite()、get__id()、is__authenticate()和is_anonymous()。
定义数据库连接的示例代码如下:
from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session,sessionmaker
from contextlib import contextmanager
db_connect_string='mysql://v_user:v_pase@localhost:3306/test_database?charset=utf8'
ssl_args={
'ssl':{
'cert':'/home/ssl/client-cert.pem',
'key':'/home/shouse/ssl/client-key.pem',
'ca':'/home/shouse/ssl/ca-cert.pem'
}
}
engine=create_engine(db_connect_string,connect_args=ssl_args)
SessionType=scoped_session(sessionmaker(bind=engine,expire_on_commit=False))
def GetSession():
return SessionType()
@contextmanager
def session_scope():
session=GetSession()
try:
yield session
session.commit()
except:
session.rollback()
raise
finally:
session.close()
解析此连接数据部分的代码如下:
- 引入数据库和会话引擎:sqlalchemy.create_engine、sqlalchemy.orm.scoped_session、sqlalchemy.orm.sessionmaker。
- 定义连接数据库需要用到的数据库字符串。本例连接MySQL数据库,字符串格式为[databse_type]://[user_name]:[password]@[domain]:[port]/[database]?[parameters]。本例中除了必须的连接信息,还传入了charset参数,指定用utf-8编码方式解码数据库中的字符串。
- 用create_engine建立数据库引擎,如果数据库开启了SSL链路,则在此处需要传入ssl客户端证书的文件路径。
- 用scoped_session(sessionmaker(bind=engine))建立会话类型SessionType,并定义函数GetSession()用以创建SessionType的实例。
至此,已经可以用GetSession()函数创建数据库会话并进行数据库操作了。但为了使之后的数据库操作的代码能够自动进行事务处理,本例中定义了上下文函数session_scope()。在Python中定义上下文函数的方法是为其加入contextlib包中的contextmanager装饰器。在上下文函数中执行如下逻辑:在函数开始时建立数据库会话,此时会自动建立一个数据库事务:当发生异常时回滚(rollback)事务;当退出时关闭(close)连接。在关闭连接时会自动进行事务提交(commit)操作。
进行数据库操作的代码:
from sqlalchemy import or_,orm
def InsertAccount(user,passwd,title,salary): #新增操作
with session_scope() as session:
account=orm.Account(user_name=user,passwd=passwd,title=title,salary=salary)
session.add(account)
def GetAccount(id=None,user_name=None): #查询操作
with session_scope() as session:
return session.query(orm.Account).filter(
or_(orm.Account.id==id,orm.Account.user_name=user_name)
).first()
def DeleteAccount(user_name): #删除操作
with session_scope() as session:
account=GetAccount(user_name=user_name)
if account:
session.delete(account)
def UpdateAccount(id,user_name,password,title,salary): #更新操作
with session_scope() as session:
account=session.query(orm.Account).filter(orm.Account.id==id).first()
if not account:return
account.user_name=user_name
account.password=password
account.salary=salary
account.title=title
InsertAccount("Mark","123","Manager",3000) #调用新增操作
InsertAccount("帅哥","456","Boss",2000) #调用新增操作
GetAccount(2) #调用查询操作
DeleteAccount("Mark")
UpdateAccount(1,"admin","none","System admin",2500)
本例演示了数据库中最常用的4种基于记录的操作:新增、查找、删除、更新。对此部分代码的解析如下:
- 用import语句引入数据表(Account)所在的包orm。引入多条件查询时使用or_。
- 每个函数中都通过with语句启用上下文函数session_scope(),通过它获取到session对象,并自动开启新事物。
- 在InsertAccount中,通过新建一个表Account实例,并通过session.add将其添加到数据库中。由于上下文函数退出时会自动提交事务,所以无需显示的调用session.commit()使新增生效。
- 在GetAccount中通过query语句进行查询,查询条件由filter设置,多个查询条件可以用or_或and_连接。
- 在DeleteAccount中通过GetAccount查询该对象,如果查询到了,则直接调用session.delete()将该对象删除。
- 在InsertAccount()中通过query根据id查询记录,如果查询到了,则通过设置对象的属性实现对记录的修改。
- 查询语句的结果是一个对象集合。查询语句后面的first()函数用于提取该集合中的第一个对象,如果用all()函数替换first()函数,查询则会返回该集合。
主流数据库的连接方式
SQLAlchemy这样的orm数据库操作方式可以对业务开发者屏蔽不同数据库之间的差异,这样当需要进行数据库迁移时(比如MySQL迁移到SQLite),则只需要更换数据库连接字符串。
下表列出了SQLAlchemy连接主流数据库时的数据库字符串的编写方法:
数据库 | 连接字符串 |
---|---|
Microsoft SQLServer | ‘mssql+pymssql://[user]:[pass]@[domain]:[port]/[dbname]’ |
MySQL | ‘mysql://[user]:[pass]@[domain]:[port]/[dbname]’ |
Oracle | ‘oracle://[user]:[pass]@[domain]:[port/[dbname]]’ |
PostgreSQL | ‘postgresql://[user]:[pass]@[domain]:[port]/[dbname]’ |
SQLite | ‘sqlite://[file_pathname]’ |