28 lines
914 B
Python
28 lines
914 B
Python
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
from APPData.Crud import CompanyCrud
|
|
from APPData.Schemas import CompanySchemas
|
|
from Utils.DataBase.SqlAlchemyUtils import get_db
|
|
|
|
router = APIRouter(
|
|
tags=["企业管理"],
|
|
prefix="/api/company"
|
|
)
|
|
|
|
|
|
@router.get("/create", summary="新建企业")
|
|
def func01(name: str, db: Session = Depends(get_db)):
|
|
# 根据name查询企业是否存在
|
|
data = CompanyCrud.get_company_by_name(db=db, name=name)
|
|
if not data:
|
|
# 不能存在新建企业再返回cid
|
|
data = CompanyCrud.create_company_by_name(db=db, name=name)
|
|
return {"cid": data.id}
|
|
|
|
|
|
@router.post("/search", summary="查询企业")
|
|
def func02(req_body: CompanySchemas.CompanySearch, db: Session = Depends(get_db)):
|
|
company = CompanyCrud.get_company_total(db, name=req_body.name, page=req_body.page, pagesize=req_body.pagesize)
|
|
return company
|