"""
待删除物料查询组件
从数据库查询指定负责人需要删除的物料名称
"""

from typing import List, Dict, Any
from db.connection import get_connection


def get_materials_to_delete(manager_name):
    """
    根据负责人名称查询待删除物料名称列表

    Args:
        manager_name: 负责人姓名

    Returns:
        物料名称列表（关键字）
    """
    query = """
        SELECT [MaterialName]
        FROM [dbo].[MaterialsTypeToBeDeleted]
        WHERE [ManagerName] = ?
    """

    with get_connection() as conn:
        results = conn.execute_query(query, (manager_name,))
        # 提取物料名称并去除空值
        material_names = [row["MaterialName"] for row in results if row["MaterialName"]]
        return material_names


def get_all_materials_to_delete() -> List[Dict[str, Any]]:
    """
    获取所有待删除物料记录

    Returns:
        List[Dict[str, Any]]: 包含MaterialName和ManagerName的记录列表
    """
    query = """
        SELECT [MaterialName], [ManagerName]
        FROM [dbo].[MaterialsTypeToBeDeleted]
        WHERE [MaterialName] IS NOT NULL
        ORDER BY [ManagerName], [MaterialName]
    """

    with get_connection() as conn:
        results = conn.execute_query(query)
        return results