# 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():
    try:

        from bin.views import LocationAllocation
        container_code = "12345"
        print(f"开始生成库位，托盘编码：{container_code}")
        allocator = LocationAllocation()  # 创建实例
        location_list_cnumber = allocator.get_location_by_status(container_code, 'in2', 1)  # 获取库位列表
        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]库位和托盘的关联关系更新成功！")



 

        
        # location_list = allocator.get_location_by_type(location_type, 'in2', 1)  # 获取库位列表
        # if not location_list:
        #     print("❌ 第一次库位获取失败，请检查库位类型")
        #     return
        # print(f"库位列表：{location_list}")
        # location_list_cnumber = allocator.get_location_by_type_remainder(batch, 1)  # 获取库位列表
        # if not location_list_cnumber:
        #     print("❌ 剩余库位获取失败，请检查托盘编码")
        #     return
        # print(f"库位列表：{location_list_cnumber}")
        print("✅ 方法生成成功！")

        
    except Exception as e:
        print(f"❌ 生成失败：{str(e)}")
        import traceback
        traceback.print_exc()

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