首页　>　文章列表　>　掌握 Python SQLAlchemy：精通关系数据库的技巧

掌握 Python SQLAlchemy：精通关系数据库的技巧

Python ORM 关系数据库数据操作
249 2024-02-25

Python SQLAlchemy 终极攻略：玩转关系数据库的秘诀

一、邂逅 sqlAlchemy：开启数据库之旅

SQLAlchemy 是一个强大的 python ORM（对象关系映射）工具，用于连接 Python 代码与关系数据库，它能够将复杂的数据操作转换为简单的 Python 语法，让开发者可以更加专注于业务逻辑，同时拥有对数据库结构的完全控制。

1. 初识 SQLAlchemy：安装与配置

# 安装 SQLAlchemy
pip install sqlalchemy

# 导入 SQLAlchemy
from sqlalchemy import create_engine, Column, Integer, String, Table

# 创建数据库引擎
engine = create_engine("sqlite:///database.db")

2. 定义模型：搭建数据库结构

# 定义模型类，代表数据库中的表
class User(object):
__tablename__ = "user"
id = Column(Integer, primary_key=True)
name = Column(String(50))
email = Column(String(50))

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

二、掌控数据：增删查改

1. 插入数据：填充数据库

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

# 插入数据
new_user = User(name="张三", email="zhangsan@example.com")
session.add(new_user)

# 提交事务，使数据持久化
session.commit()

# 关闭会话
session.close()

2. 查询数据：提取所需

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

# 查询所有数据
users = session.query(User).all()

# 查询特定数据
user = session.query(User).filter(User.name == "张三").first()

# 关闭会话
session.close()

# 打印查询结果
print(users)
print(user)

3. 更新数据：修改现有

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

# 更新数据
user = session.query(User).filter(User.name == "张三").first()
user.email = "zhangsan@new_example.com"

# 提交事务，使数据持久化
session.commit()

# 关闭会话
session.close()

# 打印更新后的数据
print(user)

4. 删除数据：抹去不需要

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

# 删除数据
user = session.query(User).filter(User.name == "张三").first()
session.delete(user)

# 提交事务，使数据持久化
session.commit()

# 关闭会话
session.close()

# 打印删除后的数据
print(session.query(User).all())

三、关系建模：勾勒数据关联

1. 一对多关系：构建层次结构

# 定义父模型
class Parent(object):
__tablename__ = "parent"
id = Column(Integer, primary_key=True)
name = Column(String(50))

# 定义子模型
class Child(object):
__tablename__ = "child"
id = Column(Integer, primary_key=True)
name = Column(String(50))
parent_id = Column(Integer, ForeignKey("parent.id"))

# 创建表
Parent.metadata.create_all(engine)
Child.metadata.create_all(engine)

# 创建数据
parent = Parent(name="王五")
child1 = Child(name="小红", parent_id=parent.id)
child2 = Child(name="小明", parent_id=parent.id)

# 提交事务，使数据持久化
session = Session(bind=engine)
session.add_all([parent, child1, child2])
session.commit()

# 关闭会话
session.close()

2. 多对多关系：勾连复杂关联

# 定义模型
class Student(object):
__tablename__ = "student"
id = Column(Integer, primary_key=True)
name = Column(String(50))

class Course(object):
__tablename__ = "course"
id = Column(Integer, primary_key=True)
name = Column(String(50))

# 定义关联表
student_course = Table("student_course",
Column("student_id", Integer, ForeignKey("student.id")),
Column("course_id", Integer, ForeignKey("course.id")))

# 创建表
Student.metadata.create_all(engine)
Course.metadata.create_all(engine)
student_course.create(engine)

# 创建数据
student1 = Student(name="李四")
student2 = Student(name="赵六")
course1 = Course(name="数学")
course2 = Course(name="语文")

# 提交事务，使数据持久化
session = Session(bind=engine)
session.add_all([student1, student2, course1, course2])
session.commit()

# 建立关联
student1.courses = [course1, course2]
student2.courses = [course1]

# 提交事务，使关联持久化
session.commit()

# 关闭会话
session.close()