sqlalchemy 单表增删改查

本文详细介绍如何使用Python的SQLAlchemy ORM进行数据库操作,包括连接数据库、增删改查等核心功能,以及高级查询技巧如联表查询、聚合函数、排序、分组等。

1、连接数据库,并创建session

from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

engine = create_engine(
        "mysql+pymysql://root:密码@127.0.0.1:3306/数据库?charset=utf8",
        max_overflow=0,  # 超过连接池大小外最多创建的连接
        pool_size=5,  # 连接池大小
        pool_timeout=30,  # 池中没有线程最多等待的时间,否则报错
        pool_recycle=-1  # 多久之后对线程池中的线程进行一次连接的回收(重置)
    )

SessionFactory = sessionmaker(bind=engine)

session = SessionFactory()

2、增

# 单条
obj = Users(name='tom')
session.add(obj)
session.commit()
# 多条
session.add_all([
        Users(name='海贼王'),
        Users(name='死神')
])
session.commit()
session.close()

3、删

session.query(Users).filter(Users.id >= 2).delete()
session.commit()
session.close()

4、改

session.query(Users).filter(Users.id == 4).update({Users.name:'死神'})
session.query(Users).filter(Users.id == 4).update({'name':'火影'})
# 在原本的字段,修改属性
session.query(Users).filter(Users.id == 4).update({'name':Users.name+"DSB"},synchronize_session=False)
session.commit()
session.close()

5、查

# 查询所有
result = session.query(Users).all()
for row in result:
        print(row.id,row.name)
# 根据条件查询
result = session.query(Users).filter(Users.id >= 2)
for row in result:
        print(row.id,row.name)
# 查询id大于2的第一个对象
result = session.query(Users).filter(Users.id >= 2).first()
print(result)

补充

# 1. 指定列, 字段别名
# select id,name as cname from users;
result = session.query(Users.id,Users.name.label('cname')).all()
for item in result:
        print(item[0],item.id,item.cname)
# 2. 默认条件and
session.query(Users).filter(Users.id > 1, Users.name == 'abc').all()
# 3. between
session.query(Users).filter(Users.id.between(1, 3), Users.name == 'abc').all()
# 4. in, not in ~
session.query(Users).filter(Users.id.in_([1,3,4])).all()
session.query(Users).filter(~Users.id.in_([1,3,4])).all()
# 5. 子查询
session.query(Users).filter(Users.id.in_(session.query(Users.id).filter(Users.name=='abc'))).all()
# 6. and 和 or
from sqlalchemy import and_, or_
# 默认 and
session.query(Users).filter(Users.id > 3, Users.name == 'eric').all()
# and
session.query(Users).filter(and_(Users.id > 3, Users.name == 'eric')).all()
# or
session.query(Users).filter(or_(Users.id < 2, Users.name == 'eric')).all()
# and or 一起使用
session.query(Users).filter(
    or_(
        Users.id < 2,
        and_(Users.name == 'eric', Users.id > 3),
        Users.extra != ""
    )).all()

# 7. filter_by,查询内部执行的是filter
session.query(Users).filter_by(name='abc').all()

# 8. 通配符 % 任意个, _一个
ret = session.query(Users).filter(Users.name.like('a_')).all()
ret = session.query(Users).filter(~Users.name.like('e%')).all()

# 9. 切片/分页
result = session.query(Users)[1:2]

# 10.排序
ret = session.query(Users).order_by(Users.name.desc()).all()
ret = session.query(Users).order_by(Users.name.desc(), Users.id.asc()).all()

# 11. group by, having , 聚合函数
from sqlalchemy.sql import func

ret = session.query(
        Users.depart_id,
        func.count(Users.id),
).group_by(Users.depart_id).all()
for item in ret:
        print(item)

# having
from sqlalchemy.sql import func

ret = session.query(
        Users.depart_id,
        func.count(Users.id),
).group_by(Users.depart_id).having(func.count(Users.id) >= 2).all()
for item in ret:
        print(item)

# 12.union 和 union all, unuon 去重 union all 不去重
"""
select id,name from users
UNION
select id,name from users;
"""
# 去重
q1 = session.query(Users.name).filter(Users.id > 2)
q2 = session.query(Favor.caption).filter(Favor.nid < 2)
ret = q1.union(q2).all()
# 不去重
q1 = session.query(Users.name).filter(Users.id > 2)
q2 = session.query(Favor.caption).filter(Favor.nid < 2)
ret = q1.union_all(q2).all()

注意:

1、操作数据结束,关闭session

session.close()

2、增、删、改,要提交数据

session.commit()

 3、去除all获取sql语句

 

转载于:https://www.cnblogs.com/wt7018/p/11616548.html

### ✅ 回答问题: 使用 **SQLAlchemy**(Python 中最流行的 ORM 框架),我们可以封装一套通用的 **增删改查(CRUD)** 操作类。以下是一个基于 SQLAlchemy 的通用 CRUD 封装示例,适用于大多数模型对象的数据库操作。 --- ### 🧪 示例代码: #### 1. 安装依赖(如未安装): ```bash pip install sqlalchemy ``` #### 2. 初始化 SQLAlchemy: ```python # database.py from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker, declarative_base # 数据库连接(以 SQLite 为例) engine = create_engine("sqlite:///./test.db", connect_args={"check_same_thread": False}) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) Base = declarative_base() ``` #### 3. 封装通用 CRUD 类: ```python # crud.py from sqlalchemy.orm import Session from typing import TypeVar, Generic, Optional, List ModelType = TypeVar("ModelType") CreateSchemaType = TypeVar("CreateSchemaType") UpdateSchemaType = TypeVar("UpdateSchemaType") class CRUDBase(Generic[ModelType, CreateSchemaType, UpdateSchemaType]): def __init__(self, model: ModelType): self.model = model def get(self, db: Session, id: int) -> Optional[ModelType]: return db.query(self.model).filter(self.model.id == id).first() def get_all(self, db: Session, *, skip: int = 0, limit: int = 100) -> List[ModelType]: return db.query(self.model).offset(skip).limit(limit).all() def create(self, db: Session, *, obj_in: CreateSchemaType) -> ModelType: db_obj = self.model(**obj_in.dict()) db.add(db_obj) db.commit() db.refresh(db_obj) return db_obj def update(self, db: Session, *, db_obj: ModelType, obj_in: UpdateSchemaType) -> ModelType: update_data = obj_in.dict(exclude_unset=True) for key, value in update_data.items(): setattr(db_obj, key, value) db.add(db_obj) db.commit() db.refresh(db_obj) return db_obj def delete(self, db: Session, *, id: int) -> ModelType: obj = db.query(self.model).get(id) if obj: db.delete(obj) db.commit() return obj ``` #### 4. 使用示例: ##### 定义模型: ```python # models.py from sqlalchemy import Column, Integer, String from database import Base class User(Base): __tablename__ = "users" id = Column(Integer, primary_key=True) name = Column(String(50)) email = Column(String(100), unique=True) ``` ##### 创建 Pydantic 模型(用于数据验证): ```python # schemas.py from pydantic import BaseModel class UserCreate(BaseModel): name: str email: str class UserUpdate(UserCreate): pass class UserOut(BaseModel): id: int name: str email: str class Config: from_attributes = True ``` ##### 使用 CRUD: ```python # main.py from database import SessionLocal, Base from models import User from crud import CRUDBase # 初始化数据库 Base.metadata.create_all(bind=database.engine) # 实例化CRUD user_crud = CRUDBase(User) # 获取数据库会话 db = SessionLocal() # 创建用户 user_create = UserCreate(name="Alice", email="alice@example.com") user = user_crud.create(db, obj_in=user_create) print("创建用户:", user.id) # 查询用户 user = user_crud.get(db, id=1) print("查询用户:", user.email) # 更新用户 user_update = UserUpdate(name="Alice Updated") updated_user = user_crud.update(db, db_obj=user, obj_in=user_update) print("更新用户:", updated_user.name) # 删除用户 deleted_user = user_crud.delete(db, id=1) print("删除用户:", deleted_user.id) ``` --- ### ✅ **优点:** | 特性 | 说明 | |------|------| | 泛型支持 | 支持不同模型的通用操作 | | 可扩展性强 | 可继承并扩展自定义方法 | | 代码简洁 | 减少重复的数据库操作代码 | | 支持 Pydantic | 与 FastAPI 等框架无缝集成 | ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值