Python学习计划——10.2使用SQLAlchemy进行数据库操作

SQLAlchemy是Python中的一个SQL工具包和对象关系映射（ORM）库。它提供了一个完整的数据库API，支持SQL表达式语言和ORM，可以方便地进行数据库操作。

1. 安装SQLAlchemy

首先，需要安装SQLAlchemy库。可以使用pip进行安装：

pip install SQLAlchemy

2. 创建数据库和表

可以使用SQLAlchemy的声明式系统（Declarative System）来定义数据库表和映射类。首先，定义一个基类，然后定义数据库表类。

示例

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

# 创建数据库连接
engine = create_engine('sqlite:///example.db')

# 创建基类
Base = declarative_base()

# 定义User表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String, nullable=False)
    age = Column(Integer, nullable=False)

# 创建表
Base.metadata.create_all(engine)

3. 插入数据

可以使用SQLAlchemy的会话（Session）来管理事务，并使用会话对象插入数据。

示例

from sqlalchemy.orm import sessionmaker

# 创建数据库连接
engine = create_engine('sqlite:///example.db')
Base = declarative_base()

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

# 定义User表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String, nullable=False)
    age = Column(Integer, nullable=False)

# 插入数据
new_user = User(name='Alice', age=30)
session.add(new_user)
session.commit()

4. 查询数据

可以使用会话对象查询数据，并使用查询API进行各种查询操作。

示例

from sqlalchemy.orm import sessionmaker

# 创建数据库连接
engine = create_engine('sqlite:///example.db')
Base = declarative_base()

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

# 定义User表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String, nullable=False)
    age = Column(Integer, nullable=False)

# 查询数据
users = session.query(User).all()
for user in users:
    print(user.id, user.name, user.age)

5. 更新数据

可以使用会话对象更新数据，并提交事务。

示例

from sqlalchemy.orm import sessionmaker

# 创建数据库连接
engine = create_engine('sqlite:///example.db')
Base = declarative_base()

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

# 定义User表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String, nullable=False)
    age = Column(Integer, nullable=False)

# 更新数据
user = session.query(User).filter_by(name='Alice').first()
user.age = 31
session.commit()

6. 删除数据

可以使用会话对象删除数据，并提交事务。

示例

from sqlalchemy.orm import sessionmaker

# 创建数据库连接
engine = create_engine('sqlite:///example.db')
Base = declarative_base()

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

# 定义User表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String, nullable=False)
    age = Column(Integer, nullable=False)

# 删除数据
user = session.query(User).filter_by(name='Alice').first()
session.delete(user)
session.commit()

7. 可运行的Python案例

下面是一个完整的Python程序，演示了如何使用SQLAlchemy进行数据库操作，包括创建数据库和表、插入数据、查询数据、更新数据和删除数据。

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

# 创建数据库连接
engine = create_engine('sqlite:///example.db')
Base = declarative_base()

# 定义User表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String, nullable=False)
    age = Column(Integer, nullable=False)

# 创建表
Base.metadata.create_all(engine)

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

# 插入数据
def insert_data(name, age):
    new_user = User(name=name, age=age)
    session.add(new_user)
    session.commit()

# 查询数据
def query_data():
    users = session.query(User).all()
    for user in users:
        print(user.id, user.name, user.age)

# 更新数据
def update_data(name, new_age):
    user = session.query(User).filter_by(name=name).first()
    if user:
        user.age = new_age
        session.commit()

# 删除数据
def delete_data(name):
    user = session.query(User).filter_by(name=name).first()
    if user:
        session.delete(user)
        session.commit()

# 主程序
if __name__ == '__main__':
    insert_data('Alice', 30)
    insert_data('Bob', 25)
    insert_data('Charlie', 35)
    print("查询所有数据:")
    query_data()
    print("更新Alice的年龄为31:")
    update_data('Alice', 31)
    print("查询所有数据:")
    query_data()
    print("删除Bob的数据:")
    delete_data('Bob')
    print("查询所有数据:")
    query_data()

可以将上面的代码复制到你的IDE中运行，观察程序的输出。这个案例综合了SQLAlchemy数据库操作的基本知识，帮助你理解和掌握这些操作。继续加油，学习Python会越来越有趣和有用！