博客

如何使用Python使用ORM操作MySQL数据库？

  • Post category:Python

ORM（对象关系映射）是一种将对象模型与关系数据库模型进行映射的技术。使用ORM可以使开发人员更轻松地操作数据库，而无需编写复杂的SQL语句。在Python中，有许多ORM库可用于操作MySQL数据库，例如SQLAlchemy、Django ORM等。以下是使用SQLAlchemy ORM操作MySQL数据库的完整攻略：

安装SQLAlchemy

在使用SQLAlchemy ORM之前，需要先安装SQLAlchemy库。可以使用以下命令在命令行中安装SQLAlchemy：

pip install sqlalchemy

连接到MySQL

在使用ORM操作MySQL之前，需要先连接到MySQL。可以使用以下代码连接到MySQL：

from sqlalchemy import create_engine

engine = create_engine('mysql+mysqlconnector://username:password@host:port/database')

在上面的代码中，username和password是MySQL的用户名和密码，host是MySQL的主机名，port是MySQL的端口号，database是要连接的数据库名称。

创建ORM模型

在使用ORM操作MySQL之前，需要先创建ORM模型。ORM模型是Python类，它映射到MySQL数据库中的表。可以使用以下代码创建ORM模型：

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    age = Column(Integer)

在上面的代码中，我们创建了一个名为User的ORM模型，它映射到MySQL数据库中的users表。id、name和age是表中的列。

创建会话

在使用ORM操作MySQL之前，需要先创建会话。会话是ORM操作MySQL的主要接口。可以使用以下代码创建会话：

from sqlalchemy.orm import sessionmaker

Session = sessionmaker(bind=engine)
session = Session()

在上面的代码中，engine是连接到MySQL的引擎对象。

插入数据

在使用ORM操作MySQL时，可以使用ORM模型插入数据。可以使用以下代码将数据插入到MySQL中：

user = User(name='Alice', age=25)
session.add(user)
session.commit()

在上面的代码中，我们创建了一个名为user的User对象，并将其添加到会话中。然后，我们使用commit()方法将数据插入到MySQL中。

查询数据

在使用ORM操作MySQL时，可以使用ORM模型查询数据。可以使用以下代码查询MySQL中的数据：

users = session.query(User).all()
for user in users:
    print(user.name, user.age)

在上面的代码中，我们使用query()方法查询MySQL中的数据，并使用all()方法获取所有数据。然后，我们遍历结果集并打印每个用户的姓名和年龄。

示例1

在这个示例中，我们将使用ORM操作MySQL，并将数据插入到名为users的表中。

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

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    age = Column(Integer)

engine = create_engine('mysql+mysqlconnector://username:password@host:port/database')
Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

user = User(name='Alice', age=25)
session.add(user)
session.commit()

users = session.query(User).all()
for user in users:
    print(user.name, user.age)

在上面的示例中，我们创建了一个名为User的ORM模型，并将其映射到MySQL数据库中的users表。然后，我们创建了一个名为user的User对象，并将其添加到会话中。后，我们查询MySQL中的数据，并打印每个用户的姓名和年龄。

示例2

在这个示例中，我们将使用ORM操作MySQL将数据插入到名为orders的表中。

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

Base = declarative_base()

class Order(Base):
    __tablename__ = 'orders'

    id = Column(Integer, primary_key=True)
    product = Column(String(50))
    price = Column(Integer)

engine = create_engine('mysql+mysqlconnector://username:password@host:port/database')
Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

order = Order(product='Apple', price=10)
session.add(order)
session.commit()

orders = session.query(Order).all()
for order in orders:
    print(order.product, order.price)

在上面的示例中，我们创建了一个名为Order的ORM模型，并将其映射到MySQL数据库中的orders表。然后，我们创建了一个名为order的Order对象，并将其添加到会话中。最后，我们查询MySQL中的数据，并打印每个订单的产品和价格。

以上是使用SQLAlchemy ORM操作MySQL数据库的完整攻略，包括连接到MySQL、创建ORM模型、创建会话、插入数据和查询数据。示例1演示了如何将数据插入到名为users的表中，并查询MySQL中的数据。示例2演示了如何将数据插入到名为orders的表中，并查询MySQL中的数据。

Tags: ,