lcny-vue3-antd-admin/src/views/base/crop-yields/base-yields/BaseDrawer.vue

<template>
  <BasicDrawer
    v-bind="$attrs"
    @register="registerDrawer"
    showFooter
    :title="getTitle"
    width="500px"
    @ok="handleSubmit"
  >
    <BasicForm @register="registerForm">
      <template #crop="{ model, field }">
        <TreeSelect
          v-model:value="model[field]"
          @select="select"
          :tree-data="treeData"
          placeholder="请选择"
          :fieldNames="{ children: 'children', label: 'name', value: 'id' }"
        ></TreeSelect>
      </template>
      <!-- 产量 -->
      <template #yield="{ model, field }">
        <Input v-model:value="model[field]" :suffix="company" placeholder="请输入"></Input>
      </template>
      <!-- 扩展 -->
      <template #extends="{ model, field }">
        <Space
          v-for="(sight, index) in model[field]"
          :key="index"
          style="width: 100%"
          align="baseline"
        >
          <FormItem
            class="w-full"
            :label="sight.name"
            :name="['extends', index, 'value']"
            :rules="{
              required: true,
              message: '请输入',
              trigger: 'change',
            }"
            :label-col="{ span: 6 }"
          >
            <Input
              v-model:value="sight.value"
              placeholder="请输入"
              :suffix="sight.unit"
              class="w-full"
              type="number"
            ></Input>
          </FormItem>
        </Space>
      </template>
    </BasicForm>
  </BasicDrawer>
</template>
<script lang="ts" setup>
  import { getTreeData, jsonString } from '/@/utils/index'
  import dayjs from 'dayjs'
  import { ref, computed, unref } from 'vue'
  import { BasicForm, useForm } from '/@/components/Form/index'
  import { accountFormSchema } from './base.data'
  import { BasicDrawer, useDrawerInner } from '/@/components/Drawer'
  import { addCropYields, editCropYields, getcrops, getCropYieldsInfo } from '/@/api/sys/user'
  import { TreeSelect, Input, Space, FormItem } from 'ant-design-vue'
  const emits = defineEmits(['success', 'register'])
  const isUpdate = ref(false)
  const treeData = ref([])
  const cropslist = ref<any>([])
  //单位
  const company = ref('')
  const getTitle = computed(() => (!isUpdate.value ? '新增产量' : '编辑产量'))
  const [registerForm, { resetFields, setFieldsValue, validate }] = useForm({
    labelWidth: 120,
    baseColProps: { span: 24 },
    schemas: accountFormSchema,
    showActionButtonGroup: false,
  })
  const [registerDrawer, { setDrawerProps, closeDrawer }] = useDrawerInner(async (data) => {
    resetFields()
    company.value = ''
    setDrawerProps({ confirmLoading: false })
    if (unref(treeData).length === 0) {
      const res = await getcrops({ type: 'all', crop_type: 1 })
      cropslist.value = res.data
      treeData.value = getTreeData(res.data, 0, 'parent_id', 'id', 'children', 'key')
    }
    isUpdate.value = data?.isUpdate
    if (unref(isUpdate)) {
      company.value = cropslist?.value?.find((e) => e.id === data.crop_id)?.unit ?? ''
      const my_extends = cropslist?.value?.find((e) => e.id === data.crop_id)?.extends ?? null
      const res = await getCropYieldsInfo(data.id)
      await setFieldsValue({
        id: data.id,
        quarter: data.quarter.toString(),
        time_year: dayjs(data.time_year.toString()),
        base_id: data.base_id,
        yield: data.yield,
        output: data.output,
        crop_id: data.crop_id,
        cultivated: data.cultivated,
        extends: formatDataByObject(my_extends, res.extends),
      })
    } else {
      setFieldsValue({
        extends: [],
      })
    }
  })

  const select = (e: number, node: any) => {
    console.log(e)
    company.value = node.unit
    setFieldsValue({
      extends: jsonString(node.extends) ? JSON.parse(node.extends) : node.extends,
    })
  }

  const handleSubmit = async () => {
    try {
      const values = await validate()
      if (values.extends) {
        values.extends = values.extends.reduce(
          (obj, item) => ((obj[item.name] = item.value), obj),
          {},
        )
      }
      values.time_year = dayjs(values.time_year.toString()).format('YYYY')
      values.type = 1
      setDrawerProps({ confirmLoading: true })
      if (values.id) {
        // 修改
        await editCropYields(values.id, values)
      } else {
        // 新增
        await addCropYields(values)
      }
      closeDrawer()
      emits('success')
    } finally {
      setDrawerProps({ confirmLoading: false })
    }
  }
  const formatDataByObject = (e, v) => {
    let arr: any = []
    const arr1 = jsonString(e) ? JSON.parse(e) : e
    if (!!arr1) {
      if (!!v) {
        arr1.forEach((e) => {
          Object.keys(v).forEach((eve) => {
            if (e.name === eve) {
              arr.push({
                name: e.name,
                value: v[eve],
                unit: e.unit,
              })
            }
          })
        })
      } else {
        arr = arr1.map((s) => {
          return {
            name: s.name,
            value: '',
            unit: s.unit,
          }
        })
      }
    }
    return arr
  }
</script>
<style lang="less" scoped>
  :deep(.ant-picker) {
    width: 100%;
  }
</style>