# 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()  # 创建实例
        number=allocator.get_pallet_count_by_batch(container_code)  # 通过实例调用)
        if number is None:
            print("❌ 该批次下无托盘，请检查托盘编码")
            return
        print(f"该批次下托盘总数：{number}")
        
        location_type = allocator.get_location_type(number)  # 获取库位类型
        if not location_type:
            print("❌ 库位类型获取失败，请检查托盘数量")
            return
        print(f"库位类型：{location_type}")
        # 调用生成库位方法
        
        batch_status = allocator.get_batch_status(container_code)  # 获取批次状态
        if not batch_status:
            print("❌ 批次状态获取失败，请检查批次号")
            return
        print(f"批次状态：{batch_status}")

        batch = allocator.get_batch(container_code)
        if not batch:
            print("❌ 批次获取失败，请检查托盘编码")
            return
        print(f"批次信息：{batch}")

        location_list_cnumber = allocator.get_location_by_status(container_code, 'in2', 1)  # 获取库位列表
        if not location_list_cnumber:
            print("❌ 通用库位获取失败，请检查托盘编码")
            return
        print(f"库位列表：{location_list_cnumber}")
        updata_location_container_link = allocator.updata_location_container_link(location_list_cnumber.location_code, container_code)  # 更新库位和托盘的关联关系
        if not updata_location_container_link:
            print("❌ 库位和托盘的关联关系更新失败，请检查托盘编码")
            return
        print(f"库位和托盘的关联关系更新成功！")
        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()