# generate_locations.py
import os
import django
import sys

# fun:get_pallet_count_by_batch: 根据托盘码查询批次下托盘总数
# fun:get_location_type: 根据托盘数目获取库位类型
# fun:updata_location_container_link: 更新库位和托盘的关联关系
# fun:get_batch_status: 获取批次状态
# fun:get_batch: 获取批次
# fun:get_location_list_remainder: 获取可用库位的c_number列表
# fun:get_min_list_index: 获取最小的库位
# fun:get_location_by_type_remainder: 根据库位类型获取库位
# fun:get_location_by_type: 第一次入库，根据库位类型获取库位
# fun:get_location_by_status: 根据库位状态获取库位
def setup_django():
    # 使用原始字符串处理Windows路径
    project_path = "D:/Document/code/vue/greater_wms"
    sys.path.append(project_path)
    
    # 根据实际目录名设置
    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'greaterwms.settings')
    django.setup()

def main():
   

        from bin.views import LocationAllocation
        # code =["12346", "12345", "1", "2", "7"]
        code =["12346", "12347"]
        for container_code in code:
            # container_code = "12346"
            print(f"开始生成库位，托盘编码：{container_code}")
            allocator = LocationAllocation()  # 创建实例

            location_list_cnumber = allocator.get_location_by_status(container_code, 'in1')  # 获取库位列表
            if not location_list_cnumber:
                print("❌ 通用库位获取失败，请检查托盘编码")
                return
            print(f"[1]库位：{location_list_cnumber}")
            
            update_location_status = allocator.update_location_status(location_list_cnumber.location_code, 'reserverd')  # 更新库位状态
            if not update_location_status:
                print("❌ 库位状态更新失败，请检查托盘编码")
                return
            print(f"[2]发送任务，库位状态更新成功！")
            update_location_group_status = allocator.update_location_group_status(location_list_cnumber.location_code)  # 更新库位组状态
            if not update_location_group_status:
                print("❌ 库位组状态更新失败，请检查托盘编码")
                return
            print(f"[3]库位组状态更新成功！")

            update_batch_status = allocator.update_batch_status(container_code, '2')  # 更新批次状态
            if not update_batch_status:
                print("❌ 批次状态更新失败，请检查批次号")
                return
            print(f"[4]批次状态更新成功！")

            update_location_group_batch = allocator.update_location_group_batch(location_list_cnumber, container_code)  # 更新库位组的批次
            if not update_location_group_batch:
                print("❌ 库位组批次更新失败，请检查托盘编码")
                return
            print(f"[5]库位组批次更新成功！")
            
            update_location_status = allocator.update_location_status(location_list_cnumber.location_code, 'occupied')  # 更新库位状态
            if not update_location_status:
                print("❌ 库位状态更新失败，请检查托盘编码")
                return
            print(f"[6]WCS到位，库位状态更新成功！")

            update_location_container_link = allocator.update_location_container_link(location_list_cnumber.location_code, container_code)  # 更新库位和托盘的关联关系
            if not update_location_container_link:
                print("❌ 库位和托盘的关联关系更新失败，请检查托盘编码")
                return
            print(f"[7]库位和托盘的关联关系更新成功！")

        print("✅ 方法生成成功！")



if __name__ == "__main__":
    setup_django()
    main()