Build lightweight AI agent admin
This commit is contained in:
Codex
@@ -0,0 +1,241 @@
|
||||
from typing import List
|
||||
from fastapi import APIRouter, Depends, Query
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.database import get_db
|
||||
from app.base_schema import ResponseModel
|
||||
from core.region.service import RegionService
|
||||
from core.region.model import Province, City, Area, Street, Village
|
||||
from core.region.schema import (
|
||||
ProvinceResponse, CityResponse, AreaResponse,
|
||||
StreetResponse, VillageResponse
|
||||
)
|
||||
|
||||
router = APIRouter(prefix="/regions", tags=["行政区划管理"])
|
||||
|
||||
|
||||
@router.get("/tree", summary="获取行政区划树形数据")
|
||||
async def get_region_tree(
|
||||
level: int = Query(default=3, ge=1, le=5, description="级联层级：1=省，2=省市，3=省市区，4=省市区街道，5=省市区街道村"),
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""
|
||||
获取行政区划树形结构数据，用于级联选择器
|
||||
|
||||
- level: 1=仅省份，2=省市两级，3=省市区三级，4=省市区街道四级，5=省市区街道村五级
|
||||
"""
|
||||
tree_data = await RegionService.get_tree(db, level=level)
|
||||
return tree_data
|
||||
|
||||
|
||||
@router.get("/provinces", response_model=List[ProvinceResponse], summary="获取省份列表")
|
||||
async def get_provinces(db: AsyncSession = Depends(get_db)):
|
||||
"""获取所有省份列表"""
|
||||
provinces = await RegionService.get_provinces(db)
|
||||
return provinces
|
||||
|
||||
|
||||
@router.post("/provinces", response_model=ProvinceResponse, summary="创建省份")
|
||||
async def create_province(data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""创建省份"""
|
||||
province = await RegionService.create_province(db, data)
|
||||
return province
|
||||
|
||||
|
||||
@router.put("/provinces/{province_id}", response_model=ProvinceResponse, summary="更新省份")
|
||||
async def update_province(province_id: str, data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""更新省份"""
|
||||
province = await RegionService.update_province(db, province_id, data)
|
||||
if not province:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="省份不存在")
|
||||
return province
|
||||
|
||||
|
||||
@router.delete("/provinces/{province_id}", response_model=ResponseModel, summary="删除省份")
|
||||
async def delete_province(province_id: str, db: AsyncSession = Depends(get_db)):
|
||||
"""删除省份"""
|
||||
success = await RegionService.delete_province(db, province_id)
|
||||
if not success:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="省份不存在")
|
||||
return ResponseModel(message="删除成功")
|
||||
|
||||
|
||||
@router.get("/cities", response_model=List[CityResponse], summary="获取城市列表")
|
||||
async def get_cities(
|
||||
province_code: str = Query(None, description="省份代码"),
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""根据省份代码获取城市列表"""
|
||||
if province_code:
|
||||
cities = await RegionService.get_cities_by_province(db, province_code)
|
||||
else:
|
||||
stmt = select(City).order_by(City.sort, City.code)
|
||||
result = await db.execute(stmt)
|
||||
cities = result.scalars().all()
|
||||
return cities
|
||||
|
||||
|
||||
@router.post("/cities", response_model=CityResponse, summary="创建城市")
|
||||
async def create_city(data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""创建城市"""
|
||||
city = await RegionService.create_city(db, data)
|
||||
return city
|
||||
|
||||
|
||||
@router.put("/cities/{city_id}", response_model=CityResponse, summary="更新城市")
|
||||
async def update_city(city_id: str, data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""更新城市"""
|
||||
city = await RegionService.update_city(db, city_id, data)
|
||||
if not city:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="城市不存在")
|
||||
return city
|
||||
|
||||
|
||||
@router.delete("/cities/{city_id}", response_model=ResponseModel, summary="删除城市")
|
||||
async def delete_city(city_id: str, db: AsyncSession = Depends(get_db)):
|
||||
"""删除城市"""
|
||||
success = await RegionService.delete_city(db, city_id)
|
||||
if not success:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="城市不存在")
|
||||
return ResponseModel(message="删除成功")
|
||||
|
||||
|
||||
@router.get("/areas", response_model=List[AreaResponse], summary="获取区县列表")
|
||||
async def get_areas(
|
||||
city_code: str = Query(None, description="城市代码"),
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""根据城市代码获取区县列表"""
|
||||
if city_code:
|
||||
areas = await RegionService.get_areas_by_city(db, city_code)
|
||||
else:
|
||||
stmt = select(Area).order_by(Area.sort, Area.code)
|
||||
result = await db.execute(stmt)
|
||||
areas = result.scalars().all()
|
||||
return areas
|
||||
|
||||
|
||||
@router.post("/areas", response_model=AreaResponse, summary="创建区县")
|
||||
async def create_area(data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""创建区县"""
|
||||
area = await RegionService.create_area(db, data)
|
||||
return area
|
||||
|
||||
|
||||
@router.put("/areas/{area_id}", response_model=AreaResponse, summary="更新区县")
|
||||
async def update_area(area_id: str, data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""更新区县"""
|
||||
area = await RegionService.update_area(db, area_id, data)
|
||||
if not area:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="区县不存在")
|
||||
return area
|
||||
|
||||
|
||||
@router.delete("/areas/{area_id}", response_model=ResponseModel, summary="删除区县")
|
||||
async def delete_area(area_id: str, db: AsyncSession = Depends(get_db)):
|
||||
"""删除区县"""
|
||||
success = await RegionService.delete_area(db, area_id)
|
||||
if not success:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="区县不存在")
|
||||
return ResponseModel(message="删除成功")
|
||||
|
||||
|
||||
@router.get("/streets", response_model=List[StreetResponse], summary="获取街道列表")
|
||||
async def get_streets(
|
||||
area_code: str = Query(None, description="区县代码"),
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""根据区县代码获取街道列表"""
|
||||
if area_code:
|
||||
streets = await RegionService.get_streets_by_area(db, area_code)
|
||||
else:
|
||||
stmt = select(Street).order_by(Street.sort, Street.code).limit(1000)
|
||||
result = await db.execute(stmt)
|
||||
streets = result.scalars().all()
|
||||
return streets
|
||||
|
||||
|
||||
@router.post("/streets", response_model=StreetResponse, summary="创建街道")
|
||||
async def create_street(data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""创建街道"""
|
||||
street = await RegionService.create_street(db, data)
|
||||
return street
|
||||
|
||||
|
||||
@router.put("/streets/{street_id}", response_model=StreetResponse, summary="更新街道")
|
||||
async def update_street(street_id: str, data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""更新街道"""
|
||||
street = await RegionService.update_street(db, street_id, data)
|
||||
if not street:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="街道不存在")
|
||||
return street
|
||||
|
||||
|
||||
@router.delete("/streets/{street_id}", response_model=ResponseModel, summary="删除街道")
|
||||
async def delete_street(street_id: str, db: AsyncSession = Depends(get_db)):
|
||||
"""删除街道"""
|
||||
success = await RegionService.delete_street(db, street_id)
|
||||
if not success:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="街道不存在")
|
||||
return ResponseModel(message="删除成功")
|
||||
|
||||
|
||||
@router.get("/villages", response_model=List[VillageResponse], summary="获取村庄列表")
|
||||
async def get_villages(
|
||||
street_code: str = Query(None, description="街道代码"),
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""根据街道代码获取村庄列表"""
|
||||
if street_code:
|
||||
villages = await RegionService.get_villages_by_street(db, street_code)
|
||||
else:
|
||||
stmt = select(Village).order_by(Village.sort, Village.code).limit(1000)
|
||||
result = await db.execute(stmt)
|
||||
villages = result.scalars().all()
|
||||
return villages
|
||||
|
||||
|
||||
@router.post("/villages", response_model=VillageResponse, summary="创建村庄")
|
||||
async def create_village(data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""创建村庄"""
|
||||
village = await RegionService.create_village(db, data)
|
||||
return village
|
||||
|
||||
|
||||
@router.put("/villages/{village_id}", response_model=VillageResponse, summary="更新村庄")
|
||||
async def update_village(village_id: str, data: dict, db: AsyncSession = Depends(get_db)):
|
||||
"""更新村庄"""
|
||||
village = await RegionService.update_village(db, village_id, data)
|
||||
if not village:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="村庄不存在")
|
||||
return village
|
||||
|
||||
|
||||
@router.delete("/villages/{village_id}", response_model=ResponseModel, summary="删除村庄")
|
||||
async def delete_village(village_id: str, db: AsyncSession = Depends(get_db)):
|
||||
"""删除村庄"""
|
||||
success = await RegionService.delete_village(db, village_id)
|
||||
if not success:
|
||||
from fastapi import HTTPException
|
||||
raise HTTPException(status_code=404, detail="村庄不存在")
|
||||
return ResponseModel(message="删除成功")
|
||||
|
||||
|
||||
@router.get("/name/{code}", response_model=ResponseModel, summary="根据代码获取名称")
|
||||
async def get_region_name(
|
||||
code: str,
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""根据区域代码获取区域名称"""
|
||||
name = await RegionService.get_region_name(db, code)
|
||||
return ResponseModel(data={"name": name})
|
||||
Reference in New Issue
Block a user