#!/usr/bin/env python3
# RuralPoint OpenBLAS Environment Setup for Python
# Import this script before importing Django

import os

def setup_openblas_environment():
    """Set OpenBLAS environment variables to prevent resource exhaustion."""
    os.environ['OPENBLAS_NUM_THREADS'] = '1'
    os.environ['MKL_NUM_THREADS'] = '1'
    os.environ['OMP_NUM_THREADS'] = '1'
    os.environ['BLAS_NUM_THREADS'] = '1'
    os.environ['LAPACK_NUM_THREADS'] = '1'
    
    print("OpenBLAS environment variables set:")
    print(f"OPENBLAS_NUM_THREADS: {os.environ.get('OPENBLAS_NUM_THREADS')}")
    print(f"MKL_NUM_THREADS: {os.environ.get('MKL_NUM_THREADS')}")
    print(f"OMP_NUM_THREADS: {os.environ.get('OMP_NUM_THREADS')}")

if __name__ == "__main__":
    setup_openblas_environment()