文章目录
- 一、模块导入
- 二、初始化FastAPI对象
- 三、数据库配置
- 1、定义 SQLite 数据库的连接路径
- 2、创建数据库连接
- 3、创建会话
- 四、创建数据库模型
- 1、创建一个基础类
- 2、定义数据库模型
- 五、创建数据库表
- 六、创建用户模型
- 七、设置接口实例
一、模块导入
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来调用函数进行查询和其他的操作
)
