VWED_server/scripts/user_save/test1.py

import asyncio
from typing import Dict, Any
EXTERNAL_CALLBACK_URL = "http://roh.vwfawedl.mobi:9001/AGVService/ContainerSendBackRequest" # 生产线到毛坯库任务


async def test1(a: int, b: int) -> int:
    return {"name": a + b}

def name1(*args, **kwargs):
    return {"test": "12333"}

async def validate_task_condition(function_args: Dict[str, Any]) -> Dict[str, Any]:
    """
    任务状态验证功能，用于验证关联任务的状态
    
    Args:
        function_args: 包含以下参数的字典
            - task_code: 任务代码，用于查询外部任务记录
            - task_type: 任务类型，用于判断验证逻辑
            - end_node: 终点节点（仅对GT类型任务需要）
    
    Returns:
        Dict[str, Any]: 验证结果
    """
    from services.external_task_record_service import ExternalTaskRecordService
    from services.task_record_service import TaskRecordService
    from data.session import get_async_session
    from data.models.operate_point_layer import OperatePointLayer
    from data.enum.task_record_enum import TaskStatus
    from sqlalchemy import select
    from utils.logger import get_logger
    
    logger = get_logger("scripts.user_save.test1")
    # print(function_args, "=========================")
    try:
        # 获取参数
        task_code = function_args.get('task_code')
        task_type = function_args.get('task_type') 
        end_node = function_args.get('end_node')
        
        if not task_code:
            return {
                "success": False,
                "message": "task_code参数为空"
            }
        
        if not task_type:
            return {
                "success": False,
                "message": "task_type参数为空"
            }
        
        # 定义任务类型模板映射
        TASK_TYPE_TEMPLATE_MAPPING = {
            "GG2MP": "GG",
            "GGFK2MP": "GG", 
            "GT2MP": "GT",
            "GTFK2MP": "GT",
            "ZG2MP": "ZG",
            "QZ2MP": "QZ",
            "LG2MP": "LG",
            "PHZ2MP": "PHZ",
            "MP2GG": "GG",
            "MP2GGFK": "GG",
            "MP2GT": "GT", 
            "MP2GTFK": "GT",
            "MP2ZG": "ZG",
            "MP2QZ": "QZ",
            "MP2LG": "LG",
            "MP2PHZ": "PHZ"
        }
        
        # 根据TaskCode查询external_task_record表获取task_record_id
        external_records = await ExternalTaskRecordService.get_task_records_by_task_code(task_code)
        logger.info(f"系统相关记录: {external_records}")
        
        if not external_records:
            return {
                "success": False,
                "message": f"找不到TaskCode={task_code}对应的叫料任务记录，无法监控叫料任务状态"
            }
        
        # 获取最新的记录
        latest_record = max(external_records, key=lambda x: x.created_at)
        
        if not latest_record.task_record_id:
            return {
                "success": False,
                "message": "叫料任务记录对应关键字task_record_id值为空"
            }
        
        # 获取TaskType对应的模板类型
        template_type = TASK_TYPE_TEMPLATE_MAPPING.get(task_type, "")
        logger.info(f"TaskCode={task_code}, TaskType={task_type}, TemplateType={template_type}")
        
        while True:
            try:
                async with get_async_session() as session:
                    # 查询end_node对应的库位锁定状态
                    stmt = select(OperatePointLayer).where(
                        OperatePointLayer.layer_name == end_node,
                        OperatePointLayer.is_deleted == False
                    ).limit(1)
                    result = await session.execute(stmt)
                    end_layer = result.scalar_one_or_none()
                    task_detail_result = await TaskRecordService.get_task_record_detail(
                        latest_record.task_record_id)
                    task_detail = task_detail_result.get("data", {})
                    task_status = task_detail.get("status", "")
                    if task_status == TaskStatus.CANCELED:
                        return {
                            "success": True,
                            "message": f"任务被取消: TaskCode={task_code}, Status={task_status}"
                        }
                    if end_layer:
                        if not end_layer.is_locked:
                            return {
                                "success": True,
                                "message": f"任务验证通过，end_node库位已解锁: {end_node}"
                            }
                        else:
                            logger.info(f"任务，end_node库位被锁定，等待解锁: TaskCode={task_code}, end_node={end_node}, locked_by={end_layer.locked_by}")
                            await asyncio.sleep(2)  # 等待2秒后重试
                    else:
                        logger.warning(f"任务，未找到end_node对应的库位，继续执行: TaskCode={task_code}, end_node={end_node}")
                        return {
                            "success": True,
                            "message": f"任务验证通过，未找到对应库位，继续执行: {end_node}"
                        }

            except Exception as e:
                logger.error(f"任务，检查end_node库位锁定状态时出现异常: {str(e)}, TaskCode={task_code}, end_node={end_node}")
                await asyncio.sleep(2)  # 等待2秒后重试

                    
    except Exception as e:
        logger.error(f"任务状态验证异常: {str(e)}")
        return {
            "success": False,
            "message": f"任务状态验证异常: {str(e)}"
        }


async def call_external_callback(req_code: str, task_type: str, arrival_user: str = "000307" ) -> dict:
    """
    调用外部回调接口

    Args:
        req_code: 到货编号（ReqCode）
        task_type: 任务类型
        arrival_user: 到货用户，固定值 000307

    Returns:
        bool: 调用是否成功（返回result为0）
    """
    arrival_no = req_code
    import aiohttp
    from utils.logger import get_logger
    logger = get_logger("scripts.user_save.test1")
    if task_type not in ["GTFK2MP", "GGFK2MP"]:
        return {"message": "不是返空类型 不需要过账"}

    payload = {
        "arrival_no": arrival_no,
        "arrival_user": arrival_user
    }

    max_retries = 100  # 最大重试次数，防止无限循环
    retry_count = 0

    while retry_count < max_retries:
        try:
            async with aiohttp.ClientSession() as sessions:
                async with sessions.post(EXTERNAL_CALLBACK_URL, json=payload) as response:
                    result = await response.json()
                    logger.info(f"外部接口调用响应: {result}, arrival_no={arrival_no}, 重试次数={retry_count}")

                    # 检查响应结果
                    if result.get("result") == "0":
                        logger.info(f"外部接口调用成功: arrival_no={arrival_no}, 总重试次数={retry_count}")
                        return {"message": "空托盘过账成功！"}
                    elif result.get("result") == "1":
                        logger.info(f"外部接口返回result=1，继续重试: arrival_no={arrival_no}, 重试次数={retry_count}")
                        retry_count += 1
                        await asyncio.sleep(5)  # 等待5秒后重试
                    else:
                        logger.error(f"外部接口返回异常结果: {result}, arrival_no={arrival_no}")
                        retry_count += 1
                        await asyncio.sleep(5)

        except Exception as e:
            logger.error(f"调用外部接口异常: {str(e)}, arrival_no={arrival_no}, 重试次数={retry_count}")
            retry_count += 1
            await asyncio.sleep(5)  # 等待5秒后重试

    logger.error(f"外部接口调用失败，已达到最大重试次数: arrival_no={arrival_no}, 最大重试次数={max_retries}")
    return {"message": "空托盘过账失败，重新尝试次数达到最大值"}


async def get_location(location_list: list, location_key: str) -> dict:
    """
    根据当前库位获取下一个库位站点（轮回逻辑）
    
    Args:
        location_list: 库位列表，格式如 ["1-1", "1-2", "1-3", "1-4"]
        location_key: 当前库位的key，如 "1-2"
    
    Returns:
        str: 下一个库位，如 "1-3"
        
    Example:
        location_list = ["1-1", "1-2", "1-3", "1-4"]
        await get_location(location_list, "1-2")  # 返回 "1-3"
        await get_location(location_list, "1-4")  # 返回 "1-1" (轮回到第一个)
    """
    try:
        if not location_list:
            raise ValueError("location_list不能为空")

        if location_key not in location_list:
            raise ValueError(f"location_key '{location_key}' 不存在于location_list中")
        
        # 找到当前库位在列表中的位置
        current_index = location_list.index(location_key)
        
        # 计算下一个位置的索引（轮回逻辑）
        next_index = (current_index + 1) % len(location_list)
        
        # 返回下一个位置的库位
        next_location_key = location_list[next_index]
        
        return {"next_location_key": next_location_key}
        
    except Exception as e:
        # 如果出现异常，返回错误信息或默认值
        from utils.logger import get_logger
        logger = get_logger("scripts.user_save.test1")
        logger.error(f"get_location函数执行异常: {str(e)}")
        
        # 可以选择抛出异常或返回默认值
        raise ValueError(f"get_location执行失败: {str(e)}")



# print(asyncio.run(get_location(["1-1", "1-2", "1-3", "1-4"], "1-4")))


async def find_movable_vehicle_in_loop(vehicle_list: list, location_list: list) -> dict:
    """
    在环线工区中找到可以移动的小车
    根据库位锁定者来判断小车现在所在位置，并检查前方库位是否可用
    
    Args:
        vehicle_list: 小车名称列表，如 ['amd1', 'amd2', 'amd3']
        location_list: 库位名称列表（按环线顺序），如 ['1-1', '1-2', '1-3', '1-4']
                      环线逻辑：1-1->1-2->1-3->1-4->1-1（循环）
    
    Returns:
        dict: {
            'movable_vehicle': str or None,    # 可以移动的小车名称
            'target_location': str or None,    # 前方可用的库位名称
            'current_location': str or None,   # 当前锁定的库位名称
            'message': str                     # 状态描述
        }
    
    Example:
        vehicle_list = ['amd1', 'amd2', 'amd3']
        location_list = ['1-1', '1-2', '1-3', '1-4']
        result = await find_movable_vehicle_in_loop(vehicle_list, location_list)
        # 返回: {'movable_vehicle': 'amd3', 'target_location': '1-4', 'current_location': '1-3', 'message': '...'}
    """
    from data.session import get_async_session
    from data.models.operate_point_layer import OperatePointLayer
    from sqlalchemy import select
    from utils.logger import get_logger
    
    logger = get_logger("scripts.user_save.test1")
    
    try:
        if not vehicle_list:
            return {
                'movable_vehicle': None,
                'target_location': None,
                'current_location': None,
                'message': 'vehicle_list参数为空'
            }
        
        if not location_list:
            return {
                'movable_vehicle': None,
                'target_location': None,
                'current_location': None,
                'message': 'location_list参数为空'
            }
        
        async with get_async_session() as session:
            # 1. 找出所有被指定小车锁定的库位，记录每个小车的当前位置
            vehicle_positions = {}  # {vehicle_name: {'location': str, 'location_index': int, 'location_obj': obj}}
            
            # 查询所有库位的锁定情况
            stmt = select(OperatePointLayer).where(
                OperatePointLayer.layer_name.in_(location_list),
                OperatePointLayer.is_deleted == False,
                OperatePointLayer.is_locked == True,
                OperatePointLayer.locked_by.in_(vehicle_list)
            )
            result = await session.execute(stmt)
            locked_locations = result.scalars().all()
            
            # 建立小车位置映射
            for location in locked_locations:
                vehicle_name = location.locked_by
                if location.layer_name in location_list:
                    location_index = location_list.index(location.layer_name)
                    vehicle_positions[vehicle_name] = {
                        'location': location.layer_name,
                        'location_index': location_index,
                        'location_obj': location
                    }
            
            if not vehicle_positions:
                return {
                    'movable_vehicle': None,
                    'target_location': None,
                    'current_location': None,
                    'message': '环线中没有被指定小车锁定的库位'
                }
            
            logger.info(f"当前小车位置: {[(k, v['location']) for k, v in vehicle_positions.items()]}")
            
            # 2. 按环线顺序检查每个小车前方的库位是否可用
            location_count = len(location_list)
            
            for vehicle_name, position_info in vehicle_positions.items():
                current_location_index = position_info['location_index']
                current_location = position_info['location']
                
                # 计算下一个库位位置（环线循环）
                next_location_index = (current_location_index + 1) % location_count
                next_location_name = location_list[next_location_index]
                
                # 检查下一个库位是否被锁定
                stmt = select(OperatePointLayer).where(
                    OperatePointLayer.layer_name == next_location_name,
                    OperatePointLayer.is_deleted == False
                ).limit(1)
                
                result = await session.execute(stmt)
                next_location_obj = result.scalar_one_or_none()
                
                if next_location_obj:
                    if not next_location_obj.is_locked:
                        # 前方库位没有被锁定，说明这个小车可以移动
                        message = f'小车 {vehicle_name} 可以从 {current_location} 移动到 {next_location_name}'
                        logger.info(message)
                        
                        return {
                            'movable_vehicle': vehicle_name,
                            'target_location': next_location_name,
                            'current_location': current_location,
                            'message': message
                        }
                    else:
                        logger.info(f'小车 {vehicle_name} 前方库位 {next_location_name} 被锁定，锁定者: {next_location_obj.locked_by}')
                else:
                    logger.warning(f'未找到库位 {next_location_name}')
            
            return {
                'movable_vehicle': None,
                'target_location': None,
                'current_location': None,
                'message': '环线中所有小车前方库位都被锁定，暂时无法移动'
            }
            
    except Exception as e:
        error_msg = f'查找可移动小车时发生错误: {str(e)}'
        logger.error(error_msg)
        return {
            'movable_vehicle': None,
            'target_location': None,
            'current_location': None,
            'message': error_msg
        }


async def get_all_movable_vehicles_in_loop(vehicle_list: list, location_list: list) -> dict:
    """
    获取环线工区中所有可以移动的小车列表（用于一轮调度3次的场景）
    
    Args:
        vehicle_list: 小车名称列表，如 ['amd1', 'amd2', 'amd3']
        location_list: 库位名称列表（按环线顺序），如 ['1-1', '1-2', '1-3', '1-4']
        
    Returns:
        dict: {
            'movable_vehicles': list,  # [{'vehicle': str, 'target_location': str, 'current_location': str}, ...]
            'total_count': int,        # 可移动小车总数
            'message': str            # 状态描述
        }
    
    Example:
        result = await get_all_movable_vehicles_in_loop(['amd1', 'amd2', 'amd3'], ['1-1', '1-2', '1-3', '1-4'])
        # 返回: {
        #   'movable_vehicles': [
        #     {'vehicle': 'amd3', 'target_location': '1-4', 'current_location': '1-3'},
        #     {'vehicle': 'amd1', 'target_location': '1-2', 'current_location': '1-1'}
        #   ],
        #   'total_count': 2,
        #   'message': '找到2个可移动的小车'
        # }
    """
    from data.session import get_async_session
    from data.models.operate_point_layer import OperatePointLayer
    from sqlalchemy import select
    from utils.logger import get_logger
    
    logger = get_logger("scripts.user_save.test1")
    
    try:
        movable_vehicles = []
        
        if not vehicle_list or not location_list:
            return {
                'movable_vehicles': [],
                'total_count': 0,
                'message': '参数为空，无法获取可移动小车'
            }
        
        async with get_async_session() as session:
            # 1. 找出所有被指定小车锁定的库位，记录每个小车的当前位置
            vehicle_positions = {}  # {vehicle_name: {'location': str, 'location_index': int, 'location_obj': obj}}
            print("location_list::", location_list)
            print("vehicle_list::", vehicle_list)
            # 查询所有库位的锁定情况
            stmt = select(OperatePointLayer).where(
                OperatePointLayer.layer_name.in_(location_list),
                OperatePointLayer.is_deleted == False,
                OperatePointLayer.is_locked == True,
                OperatePointLayer.locked_by.in_(vehicle_list)
            )
            result = await session.execute(stmt)
            locked_locations = result.scalars().all()
            print(locked_locations, "============")
            # 建立小车位置映射
            for location in locked_locations:
                vehicle_name = location.locked_by
                print(vehicle_name)
                if location.layer_name in location_list:
                    location_index = location_list.index(location.layer_name)
                    vehicle_positions[vehicle_name] = {
                        'location': location.layer_name,
                        'location_index': location_index,
                        'location_obj': location
                    }
            
            if not vehicle_positions:
                return {
                    'movable_vehicles': [],
                    'total_count': 0,
                    'message': '环线中没有被指定小车锁定的库位'
                }
            
            # 2. 检查每个小车前方的库位是否可用
            location_count = len(location_list)
            
            for vehicle_name, position_info in vehicle_positions.items():
                current_location_index = position_info['location_index']
                current_location = position_info['location']
                
                # 计算下一个库位位置（环线循环）
                next_location_index = (current_location_index + 1) % location_count
                next_location_name = location_list[next_location_index]
                
                # 检查下一个库位是否被锁定
                stmt = select(OperatePointLayer).where(
                    OperatePointLayer.layer_name == next_location_name,
                    OperatePointLayer.is_deleted == False
                ).limit(1)
                
                result = await session.execute(stmt)
                next_location_obj = result.scalar_one_or_none()
                
                if next_location_obj and not next_location_obj.is_locked:
                    movable_vehicles.append({
                        'vehicle': vehicle_name,
                        'target_location': next_location_name,
                        'current_location': current_location
                    })
            
            total_count = len(movable_vehicles)
            if total_count > 0:
                vehicle_names = [v['vehicle'] for v in movable_vehicles]
                message = f'找到{total_count}个可移动的小车: {", ".join(vehicle_names)}'
            else:
                message = '环线中所有小车前方库位都被锁定，暂时无法移动'
            
            logger.info(message)
            
            return {
                'movable_vehicles': movable_vehicles[0],
                # 'total_count': total_count,
                # 'message': message
            }
            
    except Exception as e:
        error_msg = f'获取可移动小车列表时发生错误: {str(e)}'
        logger.error(error_msg)
        return {
            'movable_vehicles': [],
            'total_count': 0,
            'message': error_msg
        }


async def judgment_condition(location: str, batteryLevel: float, threshold: float, back_task_id: str) -> dict:
    if location.find("CP") != -1 and batteryLevel >= threshold:
        return {"bool": True}
    else:
        return {"bool": False}
    # pass
# 测试示例（注释掉的测试代码）
# async def test_movable_functions():
#     """测试可移动小车功能的示例代码"""
#     vehicle_list = ['amd1', 'amd2', 'amd3']
#     action_point_list = ['1-1', '1-2', '1-3', '1-4']
#
#     # 测试找到一个可移动的小车
#     movable = await find_movable_vehicle_in_loop(vehicle_list, action_point_list)
#     print("可移动小车:", movable)
#
#     # 测试获取所有可移动的小车
#     # all_movable = await get_all_movable_vehicles_in_loop(vehicle_list, action_point_list)
#     # print("所有可移动小车:", all_movable)
#
# print(asyncio.run(test_movable_functions()))