天天看点

[4]数据库编程-【4】ORM

作者:测开人小江

Python中的ORM(对象关系映射)是一种数据库编程技术,它允许开发人员使用面向对象的方式来操作数据库,而不必直接使用SQL语句。ORM将数据库表映射到Python对象,使得数据库操作更加简洁和方便。在本篇学习教程中,我们将介绍Python中ORM的基本概念、使用方法以及一些常见的ORM库。

什么是ORM?

ORM是一种编程技术,它将数据库表和Python对象之间建立了一种映射关系,使得开发人员可以使用面向对象的方式来操作数据库,而不必直接编写SQL语句。ORM将数据库表的结构映射到Python类的属性,将数据库表的行映射到Python类的实例,从而使得对数据库的操作变得类似于对Python对象的操作。

ORM的优点包括:

  • 抽象了数据库的底层细节,屏蔽了SQL语句,使得数据库操作更加简洁和方便。
  • 提供了面向对象的编程方式,使得代码更加可读性和可维护性更好。
  • 可以实现数据库的平台无关性,通过更换ORM库,可以轻松切换不同数据库系统。

Python中的ORM库

Python中有许多优秀的ORM库可供选择,包括但不限于:

  • SQLAlchemy:一个功能强大、灵活且广泛使用的ORM库,支持多种数据库系统,包括MySQL、SQLite、PostgreSQL等。
  • Django ORM:Django框架内置的ORM,提供了强大的数据库操作功能,适用于Django项目。
  • peewee:一个简单且易用的ORM库,支持SQLite、MySQL、PostgreSQL等数据库系统。

在本篇学习教程中,我们以SQLAlchemy为例,介绍Python中ORM的基本使用方法。

安装和导入SQLAlchemy

首先,我们需要安装SQLAlchemy库。可以使用pip命令来安装:

pip install sqlalchemy
           

安装完成后,我们可以在Python代码中导入SQLAlchemy库:

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
           

在这里,我们导入了SQLAlchemy库的一些基本模块,包括create_engine、Column、Integer、String、declarative_base和sessionmaker。

定义模型类

在使用ORM时,我们首先需要定义一个模型类,该类将映射到数据库中的一张表。在SQLAlchemy中,我们可以通过继承declarative_base()类来定义模型类。

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'  # 表名
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
		age = Column(Integer)           

在上面的示例中,我们定义了一个名为User的模型类,它继承自Base类。Base类是SQLAlchemy库中的基类,它提供了一些ORM的基本功能。

在User模型类中,我们使用__tablename__属性定义了数据库中对应的表名为users。接着,我们定义了两个属性id和name,它们分别映射到了数据库表中的id和name列。Column类是SQLAlchemy库中的一个类,它表示了数据库表中的列,并且可以指定列的类型、属性等。

在上面的示例中,我们使用了Integer和String作为列的类型,分别表示整数和字符串。primary_key=True表示这个列是表的主键,用于唯一标识每一行数据。

连接数据库

在使用ORM操作数据库之前,我们需要先连接到数据库。在SQLAlchemy中,我们使用create_engine()函数来创建一个数据库引擎,用于和数据库进行交互。

from sqlalchemy import create_engine

# 创建数据库引擎
engine = create_engine('mysql://username:password@localhost/dbname')
           

在上面的示例中,我们使用了MySQL数据库作为例子,通过在create_engine()函数中传入数据库的连接信息,包括用户名、密码、主机名和数据库名,来创建了一个MySQL数据库引擎。你需要根据自己的数据库情况来修改连接信息。

创建表

在使用ORM时,我们不需要手动创建数据库表,ORM会根据我们定义的模型类自动创建表结构。我们可以使用Base.metadata.create_all()方法来创建所有定义的模型类对应的数据库表。

Base.metadata.create_all(engine)
           

在上面的示例中,Base.metadata.create_all()方法会根据我们之前定义的User模型类创建一个名为users的数据库表。

插入数据

使用ORM插入数据到数据库表中也非常简单,我们只需要创建一个模型类的实例,并将其添加到一个数据库会话(session)中,然后提交会话即可。

from sqlalchemy.orm import sessionmaker

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

# 创建一个User对象
user = User(name='John')

# 添加到会话中
session.add(user)

# 提交会话
session.commit()

# 关闭会话
session.close()
           

在上面的示例中,我们首先创建了一个会话,然后创建了一个名为user的User对象,并将其添加到会话中。最后,我们调用commit()方法提交会话,将数据插入到数据库表中,并关闭会话。

查询数据

使用ORM查询数据也非常简便,我们可以通过会话的查询接口来执行查询操作。

from sqlalchemy.orm import sessionmaker

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

# 查询所有的用户
users = session.query(User).all()

# 查询第一个用户
user = session.query(User).first()

# 查询特定条件的用户
users = session.query(User).filter_by(name='John').all()

# 查询特定条件的用户(使用filter()方法)
from sqlalchemy import text
users = session.query(User).filter(text("name = 'John'")).all()

# 查询特定条件的用户(使用filter()方法和逻辑运算符)
from sqlalchemy import and_, or_
users = session.query(User).filter(and_(User.name == 'John', User.age >= 18)).all()

# 查询特定条件的用户(使用filter()方法和模糊查询)
users = session.query(User).filter(User.name.like('%John%')).all()

# 查询特定条件的用户(使用filter()方法和in操作)
users = session.query(User).filter(User.id.in_([1, 2, 3])).all()

# 查询特定条件的用户并排序
users = session.query(User).filter(User.age >= 18).order_by(User.name.asc()).all()

# 查询特定条件的用户并限制结果集数量
users = session.query(User).filter(User.age >= 18).limit(10).all()

# 查询特定条件的用户并偏移结果集
users = session.query(User).filter(User.age >= 18).offset(5).all()

# 查询特定条件的用户并统计结果数量
count = session.query(User).filter(User.age >= 18).count()

# 关闭会话
session.close()           

在上面的示例中,我们使用了query()方法来构建查询,并通过链式调用不同的方法来添加查询条件,如filter_by()、filter()、like()、in_()等。我们还可以通过order_by()方法来排序查询结果,通过limit()和offset()方法来限制结果集数量和偏移结果集,以及通过count()方法来统计结果数量。

更新数据

使用ORM更新数据也非常简单,我们只需要先查询到需要更新的数据,然后修改其属性,并提交会话即可。

from sqlalchemy.orm import sessionmaker

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

# 查询需要更新的用户
user = session.query(User).filter_by(name='John').first()

# 更新用户的属性
user.name = 'John Smith'
user.age = 30

# 提交会话
session.commit()

# 关闭会话
session.close()
           

在上面的示例中,我们首先查询到名为John的用户,并将其属性修改为John Smith和30,然后提交会话,将修改保存到数据库中。

删除数据

使用ORM删除数据也非常简单,我们只需要先查询到需要删除的数据,然后从会话中删除并提交会话即可。

from sqlalchemy.orm import sessionmaker

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

# 查询需要删除的用户
user = session.query(User).filter_by(name='John').first()

# 删除用户
session.delete(user)

# 提交会话
session.commit()

# 关闭会话
session.close()
           

在上面的示例中,我们首先查询到名为John的用户,并将其从会话中删除,然后提交会话,从数据库中删除该用户。

总结

这就是一个简单的Python中使用ORM进行数据库操作的学习教程。通过SQLAlchemy这样强大的ORM工具,我们可以方便地进行数据库的增、查、改、删操作,而不必直接操作SQL语句。ORM将数据库表映射为Python类,使得数据库操作更加面向对象化和Pythonic,提供了更加灵活和方便的方式来操作数据库。

需要注意的是,在使用ORM进行数据库操作时,我们需要先创建模型类来定义数据库表的结构,然后通过会话来执行数据库操作。会话是一个临时的工作环境,可以用于执行多个数据库操作,并在完成后进行提交,从而将修改持久化到数据库中。

希望这个简单的学习教程对于学习Python中ORM的基本用法和概念有所帮助。通过深入学习SQLAlchemy的文档和实际的实践,你可以进一步掌握ORM的高级功能和更复杂的数据库操作。