59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
from fastapi import FastAPI, HTTPException
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from pydantic import BaseModel
|
|
from database import init_db, get_all_mcus, get_mcu, add_mcu, update_mcu, delete_mcu
|
|
|
|
# Pydantic 模型（用于请求体验证）
|
|
class MCU(BaseModel):
|
|
name: str
|
|
core: str
|
|
frequency: int
|
|
|
|
# 初始化 FastAPI
|
|
app = FastAPI(title="MCU Compare API", version="1.0")
|
|
|
|
# 允许跨域
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 初始化数据库
|
|
init_db()
|
|
|
|
# ✅ 1. 获取所有 MCU
|
|
@app.get("/api/mcus")
|
|
def read_mcus():
|
|
return get_all_mcus()
|
|
|
|
# ✅ 2. 获取单个 MCU
|
|
@app.get("/api/mcus/{mcu_id}")
|
|
def read_mcu(mcu_id: int):
|
|
mcu = get_mcu(mcu_id)
|
|
if not mcu:
|
|
raise HTTPException(status_code=404, detail="MCU not found")
|
|
return mcu
|
|
|
|
# ✅ 3. 创建 MCU
|
|
@app.post("/api/mcus")
|
|
def create_mcu(mcu: MCU):
|
|
add_mcu(mcu.name, mcu.core, mcu.frequency)
|
|
return {"message": "MCU added successfully"}
|
|
|
|
# ✅ 4. 更新 MCU
|
|
@app.put("/api/mcus/{mcu_id}")
|
|
def modify_mcu(mcu_id: int, mcu: MCU):
|
|
if not get_mcu(mcu_id):
|
|
raise HTTPException(status_code=404, detail="MCU not found")
|
|
update_mcu(mcu_id, mcu.name, mcu.core, mcu.frequency)
|
|
return {"message": "MCU updated successfully"}
|
|
|
|
# ✅ 5. 删除 MCU
|
|
@app.delete("/api/mcus/{mcu_id}")
|
|
def remove_mcu(mcu_id: int):
|
|
if not get_mcu(mcu_id):
|
|
raise HTTPException(status_code=404, detail="MCU not found")
|
|
delete_mcu(mcu_id)
|
|
return {"message": "MCU deleted successfully"} |