sqlalchemy 对单表的增删改查操作

一、创建表以及数据库连接，这里使用的是mysql

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine
from sqlalchemy import Column, Integer, String
Base = declarative_base() # 相当于model


# 创建表模型
class User(Base):
    __tablename__ = "user"  # 表名
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(32), index=True)


# 创建数据库引擎, 连接数据库
engine = create_engine("mysql+pymysql://root:@127.0.0.1:3306/bms?charset=utf8")
# 把所有继承 Base 表的模型类生成表
Base.metadata.create_all(engine)

root 后面跟的是密码，若没有则不写，我这里没有

二、创建表之后可以对表进行增删改查

1、添加数据操作

# 导入创建的数据库引擎
from crud.create_table import engine
from sqlalchemy.orm import sessionmaker
# 导入创建的表
from crud.create_table import User
# 创建会话窗口
Session = sessionmaker(engine)
# 打开会话窗口
db_session = Session()

# 1添加数据
user_obj = User(name="lihua")
db_session.add(user_obj)

db_session.commit()
db_session.close()

# 2批量添加数据 add_all 方法
db_session.add_all([
    User(name="张三"),
    User(name="李四"),
    User(name="王五"),
])

db_session.commit()
db_session.close()

 2、查询数据

from crud.create_table import User, engine
from sqlalchemy.orm import sessionmaker
Session = sessionmaker(engine)
db_session = Session()

# 一次查询所有
ret = db_session.query(User).all()
# .all()查询结果是列表，可以循环打印
for i in ret:
    print(i.name, i.id, )

# 带条件查询,第一种是filter
user_list = db_session.query(User).filter(User.id == 4).all()
print(user_list[0])

# 第二种的带条件查询
user_list = db_session.query(User).filter_by(id=4)
print(user_list[0])

# 第三种
user_list = db_session.query(User).filter(User.id >= 2).all()
print(user_list)

db_session.commit()
db_session.close()

3、修改数据

from crud.create_table import User
from crud.create_table import engine

from sqlalchemy.orm import sessionmaker
Session = sessionmaker(engine)
db_session = Session()

# 修改数据之前先查询数据 update放入字典
# res = db_session.query(User).filter(User.id == 4).update({"name": "张楠"})

# 一次修改多条数据
res = db_session.query(User).filter(User.id >= 1).update({"name": "张楠"})

db_session.commit()
db_session.close()

4、删除数据

from crud.create_table import User
from crud.create_table import engine

from sqlalchemy.orm import sessionmaker
Session = sessionmaker(engine)
db_session = Session()
# 删除一条数据或多条数据
res = db_session.query(User).filter(User.id >= 4).delete()
print(res)

db_session.commit()
db_session.close()