SQLALchemy与fastapi结合的基本流程与注意事项整理

一、模块导入

fromfastapiimportFastAPI,HTTPExceptionfrompydanticimportBaseModelfromsqlalchemyimportcreate_engine,Column,Integer,Stringfromsqlalchemy.ormimportsessionmaker,declarative_base

二、初始化FastAPI对象

app=FastAPI()

三、数据库配置

1、定义 SQLite 数据库的连接路径

database_url="sqlite:///./text2.db"

sqlite:///：代表使用sqlite数据库，后面的 ./text2.db 代表该数据库的相对路径

2、创建数据库连接

engine=create_engine(database_url,connect_args={"check_same_thread":False})

第一个参数为要创建的数据库的地址，第二个参数为check_same_thread: 设置为 False 时，允许在不同线程间共享数据库连接，
对于具体区别，可以看作添加该参数：允许跨线程使用数据库连接
不添加该参数：需要确保每个线程都有独立的数据库连接，否则就会报错

3、创建会话

SessionLocal=sessionmaker(autocommit=False,autoflush=False,bind=engine)

这里的参数一是关闭自动提交，参数二为关闭自动刷新，参数三为指定要操作的数据库连接
这里是创建了一个数据库会话工厂 SessionLocal，用于生成数据库会话对象也就是对数据库的操作对象

四、创建数据库模型

1、创建一个基础类

Base=declarative_base()

declarative_base(): 这是 SQLAlchemy 提供的一个工厂函数，用于创建一个基类，这个基类是使用声明式语法定义数据库表模型的基础

2、定义数据库模型

classLogin(Base):# 表名__tablename__="login"id=Column(Integer,primary_key=True,index=True)username=Column(String,unique=True,index=True)password=Column(String)classUser(Base):# 表名__tablename__="user"id=Column(Integer,primary_key=True,index=True)name=Column(String)age=Column(Integer)address=Column(String)numberphone=Column(String)

五、创建数据库表

Base.metadata.create_all(bind=engine)

如果bind=engin的数据库连接对应的数据库表不存在，则创建

六、创建用户模型

classUserCreate(BaseModel):name:strage:int|None=Noneaddress:str|None=Nonenumberphone:str|None=None

小技巧：除一定会传入的参数外，其他参数可以使用设空值的方法来优化后续的传参步骤

七、设置接口实例

@app.get("/login")deflogin(login:LoginCreate):try:session=SessionLocal()user_db=session.query(Login).filter(Login.username==login.username,Login.password==login.password).first()ifuser_db:returnf"{login.username}登录成功！"else:return"用户名或密码错误"exceptExceptionase:returnf"登录失败:{e}"

！！！注意：对于SQLALchemy在接口内对数据库进行操作的前提是要创建一个会议对象，也就是session = SessionLocal()，然后通过session来调用函数进行查询和其他的操作