# generate_locations.py
import os
import django
import sys

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.location_allocation import LocationAllocation
    from bin.exceptions import AllocationError
    allocator = LocationAllocation()

    try:
        result = allocator.process('12345')
        print('result:', result)
    except AllocationError as e:
        print('error:', e)

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