group_manage/ruoyi_admin/controller/system/sys_product_conf.py

# -*- coding: utf-8 -*-
# @Author  : YY

from typing import List
from typing_extensions import Annotated
from pydantic import BeforeValidator
import json
from datetime import datetime

from flask import Response

from ruoyi_common.constant import UserConstants
from ruoyi_common.base.model import AjaxResponse, TableResponse
from ruoyi_common.base.transformer import ids_to_str_list
# from ruoyi_common.domain.entity import SysProductConf
from ruoyi_common.domain.entity import SysProductConf
from ruoyi_common.domain.enum import BusinessType
from ruoyi_common.descriptor.serializer import JsonSerializer, BaseSerializer
from ruoyi_common.descriptor.validator import BodyValidator, QueryValidator, PathValidator, FileDownloadValidator
from ruoyi_common.utils import security_util as SecurityUtil
from ruoyi_system.service.module_conf.product_info_conf import SysModuleProductinfoConfService
from ruoyi_framework.descriptor.log import Log
from ruoyi_framework.descriptor.permission import HasPerm, PreAuthorize
from ... import reg


@reg.api.route("/system/moduleConf/productConf/list", methods=["GET"])
@QueryValidator(is_page=True)
# @PreAuthorize(HasPerm("moduleConf:productConf:list"))
@JsonSerializer()
def system_productinfo_conf_list(dto:SysProductConf):
    '''
        获取单品配置列表
    '''
    rows = SysModuleProductinfoConfService.select_productinfo_conf_list(dto)
    table_response = TableResponse(rows=rows)
    return table_response



@reg.api.route("/system/moduleConf/productConf/<string:id>", methods=["GET"])
@PathValidator()
@PreAuthorize(HasPerm("moduleConf:productConf:query"))
@JsonSerializer()
def module_productinfo_conf(id:str):
    '''
        获取单品配置详情
    '''
    # 注意：如果需要数据权限控制，需要实现 check_productinfo_conf_data_scope 方法
    # SysModuleProductinfoConfService.check_productinfo_conf_data_scope(id)
    productinfoConf = SysModuleProductinfoConfService.select_productinfo_conf_by_id(id)
    ajax_response = AjaxResponse.from_success(data=productinfoConf)
    return ajax_response


@reg.api.route("/system/moduleConf/productConf", methods=["POST"])
@BodyValidator()
@PreAuthorize(HasPerm("moduleConf:productConf:add"))
@Log(title="单品配置管理",business_type=BusinessType.INSERT)
@JsonSerializer()
def system_productinfo_conf_create(dto:SysProductConf):
    '''
        新增单品配置
    '''
    # if UserConstants.NOT_UNIQUE == SysModuleProductinfoConfService.check_productinfo_conf_name_unique(dto):
    #     ajax_response = AjaxResponse.from_error("单品配置名称已存在")
    #     return ajax_response
    # 注意：SysProductConf 不继承 AuditEntity，没有 create_by 等字段，所以不需要调用 create_by_user
    SysModuleProductinfoConfService.insert_conf(dto)
    ajax_response = AjaxResponse.from_success()
    return ajax_response


@reg.api.route("/system/moduleConf/productConf", methods=["PUT"])
@BodyValidator()
@PreAuthorize(HasPerm("moduleConf:productConf:edit"))
@Log(title="单品配置管理",business_type=BusinessType.UPDATE)
@JsonSerializer()
def system_productinfo_conf_update(dto:SysProductConf):
    '''
        修改单品配置
    '''
    # SysModuleProductinfoConfService.check_productinfo_conf_data_scope(dto.productinfoConf_id)
    # if UserConstants.UNIQUE != SysModuleProductinfoConfService.check_productinfo_conf_name_unique(dto):
    #     return AjaxResponse.from_error("单品配置名称已存在")
    # 注意：SysProductConf 不继承 AuditEntity，没有 update_by 等字段，所以不需要调用 update_by_user
    flag = SysModuleProductinfoConfService.update_conf(dto)
    return AjaxResponse.from_success() if flag else AjaxResponse.from_error()


@reg.api.route("/system/moduleConf/productConf/<ids>", methods=["DELETE"])
@PathValidator()
@PreAuthorize(HasPerm("moduleConf:productConf:remove"))
@Log(title="单品配置管理",business_type=BusinessType.DELETE)
@JsonSerializer()
def system_productinfo_conf_delete(
    ids: Annotated[List[str], BeforeValidator(ids_to_str_list)]
):
    '''
        删除单品配置
    '''
    # if SysModuleProductinfoConfService.has_child_by_productinfo_conf_id(id):
    #     return AjaxResponse.from_error("该单品配置存在下级单品配置，不允许删除")
    # if SysModuleProductinfoConfService.check_productinfo_conf_exist_user(id):
    #     return AjaxResponse.from_error("该单品配置存在用户，不允许删除")
    # SysModuleProductinfoConfService.check_productinfo_conf_data_scope(id)
    flag = SysModuleProductinfoConfService.delete_productinfo_conf_by_id(ids)
    return AjaxResponse.from_success() if flag > 0 else AjaxResponse.from_error()


@reg.api.route("/system/moduleConf/productConf/export", methods=["POST"])
@FileDownloadValidator()
@PreAuthorize(HasPerm("moduleConf:productConf:export"))
@Log(title="单品配置管理", business_type=BusinessType.EXPORT)
@BaseSerializer()
def system_productinfo_conf_export(dto: SysProductConf):
    '''
        导出单品配置列表为JSON
    '''
    try:
        rows = SysModuleProductinfoConfService.select_productinfo_conf_list(dto)
        # 将Pydantic模型转换为字典
        data = []
        for row in rows:
            try:
                if hasattr(row, 'model_dump'):
                    data.append(row.model_dump())
                elif hasattr(row, 'dict'):
                    data.append(row.dict())
                else:
                    # 如果是字典类型，直接使用
                    data.append(dict(row) if hasattr(row, '__iter__') and not isinstance(row, str) else row)
            except Exception as e:
                # 如果序列化失败，尝试使用默认方法
                data.append(str(row))
        # 转换为JSON字符串
        json_str = json.dumps(data, ensure_ascii=False, indent=2, default=str)
        # 创建响应 - 使用application/octet-stream确保前端识别为文件下载
        response = Response(
            json_str.encode('utf-8'),
            mimetype='application/octet-stream',
            headers={
                'Content-Disposition': f'attachment; filename=productConf_{datetime.now().strftime("%Y%m%d_%H%M%S")}.json',
                'Content-Type': 'application/json; charset=utf-8'
            }
        )
        return response
    except Exception as e:
        import traceback
        error_msg = f"导出失败: {str(e)}"
        error_json = json.dumps({"code": 500, "msg": error_msg}, ensure_ascii=False)
        error_response = Response(
            error_json,
            mimetype='application/json',
            status=500
        )
        return error_response