125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
|
# 使用orm结合fastapi接口对数据库的订单表进行增删改查
|
|||
|
|
# ① 连接数据库，定义数据库表模型，以及数据库会话工具
|
|||
|
|
# ② 定义请求体模型，编写post接口对数据库的订单表插入数据
|
|||
|
|
# ③ 编写put接口对订单进行更新操作
|
|||
|
|
# ④ 编写get接口对订单进行查询操作
|
|||
|
|
# ⑤ 编写delete接口对订单进行删除操作
|
|||
|
|
# ⑥ 总结下orm跟fastapi接口的结合使用步骤以及简单的语法结构框架
|
|||
|
|
from fastapi import FastAPI,HTTPException
|
|||
|
|
from sqlalchemy import *
|
|||
|
|
from sqlalchemy.orm import declarative_base,sessionmaker
|
|||
|
|
from datetime import datetime
|
|||
|
|
from pydantic import BaseModel
|
|||
|
|
|
|||
|
|
db_url = "mysql+pymysql://root:123456@127.0.0.1:3306/ai0824?charset=utf8mb4"
|
|||
|
|
engine = create_engine(db_url)
|
|||
|
|
|
|||
|
|
|
|||
|
|
Base = declarative_base()
|
|||
|
|
|
|||
|
|
class Order(Base):
|
|||
|
|
__tablename__ = 'order_detail'
|
|||
|
|
id = Column(Integer
|
|||
|
|
, primary_key=True
|
|||
|
|
, autoincrement=True
|
|||
|
|
,comment='订单编号，自增主键'
|
|||
|
|
)
|
|||
|
|
title= Column( String(50)
|
|||
|
|
,nullable=False
|
|||
|
|
)
|
|||
|
|
user_id= Column(Integer,nullable=False)
|
|||
|
|
create_date = Column(DATETIME
|
|||
|
|
, default=datetime.now
|
|||
|
|
)
|
|||
|
|
update_date = Column(DATETIME
|
|||
|
|
, default=datetime.now
|
|||
|
|
, onupdate=datetime.now
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
Base.metadata.create_all(engine)
|
|||
|
|
|
|||
|
|
Session = sessionmaker(bind=engine
|
|||
|
|
, autoflush=False
|
|||
|
|
, autocommit=False
|
|||
|
|
)
|
|||
|
|
db=Session()
|
|||
|
|
|
|||
|
|
app = FastAPI(title='订单管理系统')
|
|||
|
|
|
|||
|
|
from fastapi import Depends
|
|||
|
|
from sqlalchemy.orm import Session
|
|||
|
|
|
|||
|
|
def get_db():
|
|||
|
|
db = Session()
|
|||
|
|
try:
|
|||
|
|
yield db
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
class OrderRequest(BaseModel):
|
|||
|
|
title:str|None = None
|
|||
|
|
user_id:int|None =None
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.put('/orders/{id}')
|
|||
|
|
def update_order(order: OrderRequest, id: int, db: Session = Depends(get_db)):
|
|||
|
|
db_order = db.query(Order).filter(Order.id == id).first()
|
|||
|
|
if not db_order:
|
|||
|
|
raise HTTPException(status_code=404, detail="订单不存在")
|
|||
|
|
for key, value in order.model_dump(exclude_unset=True).items():
|
|||
|
|
setattr(db_order, key, value)
|
|||
|
|
db.commit()
|
|||
|
|
return '更新订单成功'
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.put('/orders/{id}',summary='订单更新接口')
|
|||
|
|
def get_orders(order:OrderRequest,id:int):
|
|||
|
|
db.query(Order).filter( Order.id == id ).update( order.model_dump(exclude_unset=True) )
|
|||
|
|
db.commit()
|
|||
|
|
db.close()
|
|||
|
|
return '更新订单成功'
|
|||
|
|
|
|||
|
|
@app.get('/orders',summary='订单查询接口')
|
|||
|
|
def get_orders(order_title:str):
|
|||
|
|
r = db.query(Order).filter( Order.title == order_title ).all()
|
|||
|
|
db.close()
|
|||
|
|
return [{"order_id": i.id, 'order_title': i.title, "user_id": i.user_id, "create_date": i.create_date,
|
|||
|
|
"update_date": i.update_date} for i in r]
|
|||
|
|
@app.delete('/orders/{id}',summary='订单删除接口')
|
|||
|
|
def delete_orders(order:OrderRequest,id:int):
|
|||
|
|
db.query(Order).filter( Order.id == id ).delete( )
|
|||
|
|
db.commit()
|
|||
|
|
db.close()
|
|||
|
|
return '删除订单成功'
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
import uvicorn
|
|||
|
|
uvicorn.run("work:app",host='0.0.0.0',port=12345)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 导入依赖
|
|||
|
|
# 创建数据库引擎
|
|||
|
|
# 创建基类
|
|||
|
|
# 定义表模型（继承 Base）
|
|||
|
|
# 创建表（create_all）
|
|||
|
|
# 创建会话工厂
|
|||
|
|
# 创建 FastAPI 应用
|
|||
|
|
# 定义请求体模型（Pydantic）
|
|||
|
|
# 编写增删改查接口
|
|||
|
|
# 启动服务
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
# 思考下：
|
|||
|
|
现在需要创建三个目录，api、model、schema
|
|||
|
|
① api目录里编写一个文件order_api.py ,利用api路由分发，存放编写好的以上四个订单增删改查接口
|
|||
|
|
② model目录里编写一个order_model.py ，存放的是订单表的表模型类
|
|||
|
|
③ schema目录里编写一个 order_schema.py , 存放的是订单表的相关请求体、相应体模型
|
|||
|
|
④ 在项目目录下创建一个 main.py文件，从main.py文件启动整个接口程序
|
|||
|
|
尝试操作下！
|
|||
|
|
'''
|