# +======================================================================
# | DjangoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
# +----------------------------------------------------------------------
# | Copyright (c) 2021-2026 小蚂蚁云团队 版权所有
# +----------------------------------------------------------------------
# | 开源协议: Apache-2.0
# | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
# +----------------------------------------------------------------------
# | 官方网站: https://www.djangoadmin.cn
# +----------------------------------------------------------------------
# | 研发团队: 小蚂蚁云团队 荣誉出品
# +======================================================================
# | 版权和免责声明:
# | 一、知识产权声明
# | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
# | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
# | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
# | 二、使用限制与禁止条款
# | 您理解并同意，禁止将本产品用于以下任何行为：
# | 1. 任何违反中华人民共和国法律法规的用途；
# | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
# | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
# | 三、责任豁免
# | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
# | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
# | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
# | 四、违约与法律责任
# | 如发现任何违反本声明的行为，本团队有权：
# | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
# | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
# | 3. 保留进一步采取刑事控告等法律措施的权利。
# | 五、许可范围说明
# | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
# | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
# | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
# | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
# | 成的一切经济损失及商誉损害。
# | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
# | 许可协议》附件为准。
# +======================================================================

"""
职级管理服务模块
=============================================================

该模块实现了职级管理相关的所有业务逻辑。
包括职级的分页查询、详情查询、添加、更新、删除、导入导出等功能。
职级用于标识用户在组织中的职级层次，如：初级工程师、中级工程师、高级工程师等，
每个职级包含名称、排序、状态等属性，通过 sort 字段控制显示顺序。

命名规范：
- 函数参数使用驼峰命名（如 levelId, pageNo）
- 表单字段使用下划线命名（如 name, sort, status）- 与数据库字段保持一致
- 返回数据使用驼峰命名（如 createUser, updateTime）- 与前端规范保持一致
- 私有函数以下划线开头（如 _apply_filters, _build_page_data）

功能模块：
1. 分页查询：支持按名称模糊搜索、状态筛选，按 sort 字段排序
2. 详情查询：根据ID获取单个职级的详细信息
3. 数据维护：职级的添加、更新、删除（支持批量）
4. 数据导入：从Excel文件批量导入职级数据
5. 数据导出：将职级数据导出为Excel文件
6. 数据列表：获取所有启用的职级列表（用于下拉选择等场景）
"""

import logging
import os
import random
import time

from django.core.paginator import Paginator, PageNotAnInteger, InvalidPage, EmptyPage
from openpyxl.reader.excel import load_workbook
from openpyxl.workbook import Workbook

from application.level import forms
from application.level.constants import LevelConstants
from application.level.models import Level
from application.upload import services
from config.env import DJANGO_TEMP_PATH, DJANGO_UPLOAD_DIR, DJANGO_FILE_URL
from constant.constants import PAGE_SIZE
from utils import R, regular
from utils.file import mkdir
from utils.common import format_datetime, parse_request_body
from utils.security import get_username


# =============================================================
# 分页查询服务
# =============================================================

def get_level_page(request):
    """
    查询职级分页数据

    Args:
        request: HttpRequest对象，包含查询参数
            - pageNo: 当前页码（可选，默认为1）
            - pageSize: 每页记录数（可选，默认为PAGE_SIZE）
            - name: 职级名称（可选，模糊查询）
            - status: 职级状态（可选，精确匹配）

    Returns:
        R: 包含分页数据的JSON响应
    """
    try:
        # 获取分页参数
        page_no = int(request.GET.get('pageNo', 1))
        page_size = int(request.GET.get('pageSize', PAGE_SIZE))

        # 构建基础查询（只查询未删除的记录）
        query = Level.objects.filter(is_delete=False)

        # 应用筛选条件
        query = _apply_filters(query, request)

        # 按排序字段升序排列
        query = query.order_by("sort")

        # 执行分页查询
        paginator = Paginator(query, page_size)
        try:
            level_list = paginator.page(page_no)
        except (PageNotAnInteger, InvalidPage, EmptyPage):
            # 页码无效时默认返回第一页
            level_list = paginator.page(1)
            page_no = 1

        # 构建并返回分页数据
        page_data = _build_page_data(level_list, paginator, page_no, page_size)
        return R.ok(data=page_data)

    except Exception as e:
        logging.error(f"查询职级分页数据异常: {str(e)}")
        return R.failed(msg="查询失败，请稍后重试")


def _apply_filters(queryset, request):
    """
    应用查询筛选条件（私有辅助函数）

    Args:
        queryset: 初始查询集
        request: HttpRequest对象，包含筛选参数

    Returns:
        QuerySet: 应用筛选后的查询集
    """
    # 职级名称模糊筛选（不区分大小写）
    name = request.GET.get('name')
    if name:
        queryset = queryset.filter(name__icontains=name)

    # 职级状态筛选
    status = request.GET.get('status')
    if status and status.isdigit():
        queryset = queryset.filter(status=int(status))

    return queryset


def _build_page_data(level_list, paginator, page_no, page_size):
    """
    构建分页返回数据（私有辅助函数）

    Args:
        level_list: 当前页数据列表（Page对象）
        paginator: 分页器对象
        page_no: 当前页码
        page_size: 每页大小

    Returns:
        dict: 标准格式的分页数据
    """
    records = []
    for item in level_list:
        records.append({
            'id': item.id,
            'name': item.name,
            'sort': item.sort,
            'status': item.status,
            'createUser': item.create_user,
            'createTime': format_datetime(item.create_time),
            'updateUser': item.update_user,
            'updateTime': format_datetime(item.update_time),
        })

    return {
        'records': records,
        'total': paginator.count,  # 总记录数
        'size': page_size,  # 每页大小
        'current': page_no,  # 当前页码
        'pages': paginator.num_pages  # 总页数
    }


# =============================================================
# 详情查询服务
# =============================================================

def get_level_detail(level_id):
    """
    根据ID查询职级详情

    Args:
        level_id: 职级ID

    Returns:
        dict/None: 职级详情字典，不存在时返回None
    """
    try:
        # 查询未删除的职级记录
        level = Level.objects.filter(is_delete=False, id=level_id).first()
        if not level:
            return None

        # 返回详情数据
        return {
            'id': level.id,
            'name': level.name,
            'status': level.status,
            'sort': level.sort,
        }
    except Exception as e:
        logging.error(f"查询职级详情异常，ID: {level_id}, 错误: {str(e)}")
        return None


# =============================================================
# 添加服务
# =============================================================

def add_level(request):
    """
    添加职级

    处理流程：
    1. 解析请求体中的JSON数据
    2. 表单验证
    3. 创建数据库记录

    Args:
        request: HttpRequest对象，包含职级信息的JSON数据

    Returns:
        R: 操作结果（成功/失败）
    """
    # 解析请求参数
    data, error = parse_request_body(request)
    if error:
        return R.failed(msg=error)

    # 表单验证
    form = forms.LevelForm(data=data)
    if not form.is_valid():
        return R.failed(msg=regular.get_err(form))

    try:
        # 创建职级记录
        Level.objects.create(
            name=form.cleaned_data["name"],
            status=int(form.cleaned_data["status"]),
            sort=int(form.cleaned_data["sort"]),
            create_user=get_username(request),
            update_user=get_username(request)
        )
        return R.ok(msg="添加成功")
    except Exception as e:
        logging.error(f"添加职级异常: {str(e)}")
        return R.failed(msg="添加失败，请稍后重试")


# =============================================================
# 更新服务
# =============================================================

def update_level(request):
    """
    更新职级

    处理流程：
    1. 解析请求体中的JSON数据
    2. 验证职级ID
    3. 表单验证
    4. 查询现有记录
    5. 更新各字段
    6. 保存更新

    Args:
        request: HttpRequest对象，包含职级ID和更新信息的JSON数据

    Returns:
        R: 操作结果（成功/失败）
    """
    # 解析请求参数
    data, error = parse_request_body(request)
    if error:
        return R.failed(msg=error)

    # 验证职级ID
    level_id = data.get('id')
    if not level_id or not str(level_id).isdigit() or int(level_id) <= 0:
        return R.failed("职级ID无效")
    level_id = int(level_id)

    # 表单验证
    form = forms.LevelForm(data=data)
    if not form.is_valid():
        return R.failed(msg=regular.get_err(form))

    try:
        # 查询职级记录
        level = Level.objects.filter(id=level_id, is_delete=False).first()
        if not level:
            return R.failed("记录不存在")

        # 更新各字段
        level.name = form.cleaned_data["name"]
        level.status = int(form.cleaned_data["status"])
        level.sort = int(form.cleaned_data["sort"])
        level.update_user = get_username(request)
        level.save()

        return R.ok(msg="更新成功")
    except Exception as e:
        logging.error(f"更新职级异常，ID: {level_id}, 错误: {str(e)}")
        return R.failed(msg="更新失败，请稍后重试")


# =============================================================
# 删除服务（支持批量）
# =============================================================

def delete_levels(level_ids):
    """
    删除职级（批量删除）

    采用逻辑删除方式（is_delete=True），而非物理删除。
    支持逗号分隔的多个ID。

    Args:
        level_ids: 职级ID字符串，多个ID用逗号分隔
                   例如: "1" 或 "1,2,3"

    Returns:
        R: 操作结果，包含删除数量
    """
    if not level_ids:
        return R.failed("记录ID不存在")

    try:
        # 解析ID列表
        id_list = [int(id_str.strip()) for id_str in level_ids.split(',') if id_str.strip()]
        if not id_list:
            return R.failed("无效的记录ID")

        # 批量更新删除标识（逻辑删除）
        updated_count = Level.objects.filter(
            id__in=id_list,
            is_delete=False
        ).update(is_delete=True)

        return R.ok(msg=f"本次共删除{updated_count}条数据")
    except ValueError:
        return R.failed("记录ID格式错误")
    except Exception as e:
        logging.error(f"删除职级异常，IDs: {level_ids}, 错误: {str(e)}")
        return R.failed(msg="删除失败，请稍后重试")


# =============================================================
# 导入服务
# =============================================================

def import_levels(request):
    """
    从Excel导入职级数据

    处理流程：
    1. 上传并读取Excel文件
    2. 解析Excel数据
    3. 批量创建职级记录

    Args:
        request: HttpRequest对象，包含上传的文件

    Returns:
        R: 操作结果，包含导入数量
    """
    try:
        # 上传并读取Excel文件
        file_info = services.upload_file(request, LevelConstants.ALLOWED_EXCEL_EXTENSIONS)
        if not file_info or 'filePath' not in file_info:
            return R.failed("文件上传失败")

        # 解析Excel数据
        levels_data = _parse_excel_file(file_info['filePath'])
        if not levels_data:
            return R.failed("Excel文件中没有有效数据")

        # 批量创建职级
        current_user = get_username(request)
        created_count = _batch_create_levels(levels_data, current_user)

        return R.ok(msg=f"本次共导入【{created_count}】条数据")

    except Exception as e:
        logging.error(f"导入职级Excel异常: {str(e)}")
        return R.failed(msg="导入失败，请检查文件格式")


def _parse_excel_file(file_path):
    """
    解析Excel文件，提取职级数据（私有辅助函数）

    Args:
        file_path: Excel文件路径

    Returns:
        list: 职级数据列表，每个元素为 (name, sort, status) 元组
    """
    wb = load_workbook(file_path)
    ws = wb.worksheets[0]

    data = []
    for row in range(2, ws.max_row + 1):
        name = ws.cell(row, 2).value
        sort = ws.cell(row, 3).value
        status_text = ws.cell(row, 4).value

        # 跳过空数据行
        if not name:
            continue

        # 转换状态
        status = LevelConstants.STATUS_NORMAL if status_text == "正常" else LevelConstants.STATUS_DISABLED

        data.append((name, sort, status))

    return data


def _batch_create_levels(levels_data, current_user):
    """
    批量创建职级（私有辅助函数）

    Args:
        levels_data: 职级数据列表
        current_user: 当前用户名

    Returns:
        int: 创建成功的数量
    """
    level_objects = []
    for name, sort, status in levels_data:
        level_objects.append(Level(
            name=name,
            status=status,
            sort=sort,
            create_user=current_user,
            update_user=current_user
        ))

    return Level.objects.bulk_create(level_objects)


# =============================================================
# 导出服务
# =============================================================

def export_levels(request):
    """
    导出职级数据到Excel

    处理流程：
    1. 查询数据（支持筛选）
    2. 构建导出数据
    3. 生成Excel文件
    4. 返回文件下载URL

    Args:
        request: HttpRequest对象，支持筛选参数
            - name: 按职级名称筛选
            - status: 按职级状态筛选

    Returns:
        R: 包含文件下载URL的JSON响应
    """
    try:
        # 查询数据
        query = Level.objects.filter(is_delete=False)
        query = _apply_filters(query, request)
        query = query.order_by("sort")

        # 构建导出数据
        export_data = _build_export_data(query)

        # 生成Excel文件
        file_url = _generate_excel_file(export_data)

        return R.ok(data=file_url)

    except Exception as e:
        logging.error(f"导出职级Excel异常: {str(e)}")
        return R.failed(msg="导出失败，请稍后重试")


def _build_export_data(queryset):
    """
    构建导出数据（私有辅助函数）

    Args:
        queryset: 职级查询集

    Returns:
        list: 导出数据列表
    """
    data_list = []
    for item in queryset:
        data_list.append({
            'id': item.id,
            'name': item.name,
            'sort': item.sort,
            'status': LevelConstants.STATUS_MAP.get(item.status, "未知"),
            'createTime': format_datetime(item.create_time),
            'updateTime': format_datetime(item.update_time),
        })
    return data_list


def _generate_excel_file(data_list):
    """
    生成Excel文件（私有辅助函数）

    Args:
        data_list: 要导出的数据列表

    Returns:
        str: 文件访问URL
    """
    wb = Workbook()
    ws = wb.worksheets[0]

    # 添加表头
    ws.append(LevelConstants.EXCEL_HEADERS)

    # 添加数据
    for item in data_list:
        ws.append(list(item.values()))

    # 创建保存目录
    save_dir = os.path.join(DJANGO_TEMP_PATH, time.strftime('%Y%m%d'))
    mkdir(save_dir)

    # 生成文件名
    timestamp = time.strftime('%Y%m%d%H%M%S')
    random_suffix = f'{random.randint(0, 100):05d}'
    filename = f"{timestamp}{random_suffix}.xlsx"
    file_path = os.path.join(save_dir, filename)

    # 保存文件
    wb.save(file_path)

    # 返回文件访问URL
    relative_path = file_path.replace(DJANGO_UPLOAD_DIR, "")
    return DJANGO_FILE_URL + relative_path


# =============================================================
# 下拉列表获取服务
# =============================================================

def get_level_list():
    """
    获取所有启用的职级列表

    通常用于前端下拉选择框等场景。
    只返回 status=1（启用）且未删除的记录。

    Returns:
        list: 职级列表，每个元素为职级模型字典
              查询失败时返回空列表
    """
    try:
        # 查询启用的职级数据列表
        # status=1 表示启用，is_delete=False 表示未删除
        return list(Level.objects.filter(
            is_delete=False,
            status=LevelConstants.STATUS_NORMAL
        ).order_by("sort").values())
    except Exception as e:
        logging.error(f"获取职级列表异常: {str(e)}")
        return []