06. 开发班级信息管理

from typing import Optional

from fastapi import APIRouter
from pydantic import create_model, BaseModel, Field
from tortoise.contrib.pydantic import pydantic_model_creator

from common.exception_handler import CustomException
from common.result import Result, PageInfo
from models import Clazz

router = APIRouter(prefix="/clazz")

ClazzPydantic = pydantic_model_creator(Clazz)

ClazzCreatePydantic = create_model(
    "ClazzCreatePydantic",
    **{
        name: (Optional[field.annotation], None)
        for name, field in ClazzPydantic.model_fields.items()
    },
    major_id=(Optional[int], Field(None, alias="majorId"))
)


# 新增
@router.post("/add")
async def add(clazz_create_pydantic: ClazzCreatePydantic):
    db_clazz = await Clazz.get_or_none(no=clazz_create_pydantic.no)
    if db_clazz is not None:
        raise CustomException("班级编号重复")
    # 将参数转换成 字典数据
    create_data = clazz_create_pydantic.model_dump(exclude_unset=True, exclude={"id"})
    await Clazz.create(**create_data)  # no=xxx,name=xxx,college=xxx
    return Result.success()


# 更新
@router.put("/update")
async def add(clazz_create_pydantic: ClazzCreatePydantic):
    if clazz_create_pydantic.id is None:
        raise CustomException("缺少参数ID")
    # 将参数转换成 字典数据
    update_data = clazz_create_pydantic.model_dump(exclude_unset=True, exclude={"id"})
    await Clazz.filter(id=clazz_create_pydantic.id).update(**update_data)  # no=xxx,name=xxx,college=xxx where id = xxx
    return Result.success()


# 删除
@router.delete('/delete/{clazz_id}')
async def delete(clazz_id: int):
    await Clazz.filter(id=clazz_id).delete()
    return Result.success()


# 单个查询
@router.get('/selectById/{clazz_id}')
async def select_by_id(clazz_id: int):
    clazz = await Clazz.get_or_none(id=clazz_id)
    return Result.success(clazz)


# 查询所有数据
@router.get('/selectAll')
async def select_all(name: str = ""):
    clazz_list = await Clazz.filter(name__contains=name)  # name__contains表示根据name进行模糊查询
    return Result.success(clazz_list)


# 分页查询数据
@router.get('/selectPage')
async def select_page(name: str = "", pageNum: int = 1, pageSize: int = 10):
    # name__contains表示根据name进行模糊查询  prefetch_related 关联查询到 major模块的数据
    query = Clazz.filter(name__contains=name).prefetch_related("major")
    clazz_list = await query.order_by("-id").offset((pageNum - 1) * pageSize).limit(pageSize)
    total = await query.count()
    # clazz_list 转成字典数据
    # majorName 怎么返回？？
    # {id=xxx, name=xxx, no=xxx}
    clazz_dict_list = [
        {
            **ClazzPydantic.model_validate(clazz).model_dump(),  # id=xxx,no=xxx,name=xxx
            "majorId": clazz.major.id if clazz.major else None,
            "majorName": clazz.major.name if clazz.major else None
        }
        for clazz in clazz_list
    ]
    page_info = PageInfo(list=clazz_dict_list, total=total)
    return Result.success(page_info)

<template>
  <div>

    <div class="card" style="margin-bottom: 5px;">
      <el-input v-model="data.name" style="width: 300px; margin-right: 10px" placeholder="请输入名称查询"></el-input>
      <el-button type="primary" @click="load">查询</el-button>
      <el-button type="info" style="margin: 0 10px" @click="reset">重置</el-button>
    </div>

    <div class="card" style="margin-bottom: 5px">
      <div style="margin-bottom: 10px">
        <el-button type="primary" @click="handleAdd">新增</el-button>
      </div>
      <el-table :data="data.tableData" stripe>
        <el-table-column label="班级编号" prop="no"></el-table-column>
        <el-table-column label="名称" prop="name"></el-table-column>
        <el-table-column label="所属专业" prop="majorName"></el-table-column>
        <el-table-column label="班主任" prop="teacher"></el-table-column>
        <el-table-column label="操作" align="center" width="160">
          <template #default="scope">
            <el-button type="primary" @click="handleEdit(scope.row)">编辑</el-button>
            <el-button type="danger" @click="handleDelete(scope.row.id)">删除</el-button>
          </template>
        </el-table-column>
      </el-table>
    </div>

    <div class="card">
      <el-pagination @current-change="load" background layout="total, prev, pager, next" v-model:page-size="data.pageSize" v-model:current-page="data.pageNum" :total="data.total"/>
    </div>

    <el-dialog title="班级信息" width="40%" v-model="data.formVisible" :close-on-click-modal="false" destroy-on-close>
      <el-form ref="formRef" :model="data.form" :rules="data.rules" label-width="100px" style="padding-right: 50px">
        <el-form-item label="班级编号" prop="no">
          <el-input placeholder="请输入班级编号" v-model="data.form.no" autocomplete="off" />
        </el-form-item>
        <el-form-item label="名称" prop="name">
          <el-input placeholder="请输入名称" v-model="data.form.name" autocomplete="off" />
        </el-form-item>
        <el-form-item label="所属专业" prop="majorId">
          <el-select placeholder="请选择专业" v-model="data.form.majorId">
            <el-option v-for="item in data.majorList" :key="item.id" :label="item.name" :value="item.id"></el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="班主任" prop="teacher">
          <el-input placeholder="请输入班主任" v-model="data.form.teacher" autocomplete="off" />
        </el-form-item>
      </el-form>
      <template #footer>
      <span class="dialog-footer">
        <el-button @click="data.formVisible = false">取 消</el-button>
        <el-button type="primary" @click="save">保 存</el-button>
      </span>
      </template>
    </el-dialog>

  </div>
</template>

<script setup>
import request from "@/utils/request";
import {reactive, ref} from "vue";
import {ElMessageBox, ElMessage} from "element-plus";

const formRef = ref()
const data = reactive({
  user: JSON.parse(localStorage.getItem('system-user') || '{}'),
  pageNum: 1,
  pageSize: 10,
  total: 0,
  formVisible: false,
  form: {},
  tableData: [],
  majorList: [],
  name: null,
  rules: {
    no: [
      { required: true, message: '请输入专业代码', trigger: 'blur' }
    ],
    name: [
      { required: true, message: '请输入名称', trigger: 'blur' }
    ],
    majorId: [
      { required: true, message: '请选择专业', trigger: 'change' }
    ],
    teacher: [
      { required: true, message: '请输入班主任', trigger: 'blur' }
    ],
  }
})

// 查询专业的信息list
request.get('/major/selectAll').then(res => {
  data.majorList = res.data
})

// 分页查询
const load = () => {
  request.get('/clazz/selectPage', {
    params: {
      pageNum: data.pageNum,
      pageSize: data.pageSize,
      name: data.name
    }
  }).then(res => {
    if (res.code === '200') {
      data.tableData = res.data?.list
      data.total = res.data?.total
    } else {
      ElMessage.error(res.msg)
    }
  })
}
load()

// 新增
const handleAdd = () => {
  data.form = {}
  data.formVisible = true
}

// 编辑
const handleEdit = (row) => {
  data.form = JSON.parse(JSON.stringify(row))
  data.formVisible = true
}

// 新增保存
const add = () => {
  request.post('/clazz/add', data.form).then(res => {
    if (res.code === '200') {
      load()
      ElMessage.success('操作成功')
      data.formVisible = false
    } else {
      ElMessage.error(res.msg)
    }
  })
}

// 编辑保存
const update = () => {
  request.put('/clazz/update', data.form).then(res => {
    if (res.code === '200') {
      load()
      ElMessage.success('操作成功')
      data.formVisible = false
    } else {
      ElMessage.error(res.msg)
    }
  })
}

// 弹窗保存
const save = () => {
  formRef.value.validate(valid => {
    if (valid) {
      // data.form有id就是更新，没有就是新增
      data.form.id ? update() : add()
    }
  })
}

// 删除
const handleDelete = (id) => {
  ElMessageBox.confirm('删除后数据无法恢复，您确定删除吗?', '删除确认', { type: 'warning' }).then(res => {
    request.delete('/clazz/delete/' + id).then(res => {
      if (res.code === '200') {
        load()
        ElMessage.success('操作成功')
      } else {
        ElMessage.error(res.msg)
      }
    })
  }).catch(err => {})
}

// 重置
const reset = () => {
  data.name = null
  load()
}
</script>