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]' |
以上就是Python下SQLAlchemy的简单介绍的详细内容,更多请关注Work网其它相关文章!