import json
import os
from datetime import datetime

    # Generate output file name with current date
def get_date_str():
    """Returns the current date in 'dd-mm-YYYY' format."""
    return datetime.now().strftime('%d_%m_%Y')
current_date_str = get_date_str()
current_path = os.getcwd()
current_dir_path = os.path.basename(current_path)

def merge_json_files_to_custom_array(input_files, output_file, unique_key, fields):
    merged_data = {}
    
    # Load and merge data from input files
    for file in input_files:
        with open(file, 'r', encoding='utf-8') as f:
            data = json.load(f)
            for item in data:
                key = item.get(unique_key)
                if key:  # Ensure the key exists
                    merged_data[key] = item
    
    # Create the custom array
    custom_array = []
    for item in merged_data.values():
        res = item.get("resurtido", None)
        inv = item.get("inventario", None)
        # if res == "SIN RESURTIDO" and inv != 0:
        custom_entry = [item.get(field, None) for field in fields]
        custom_array.append(custom_entry)
    
    # Write custom array to the output file
    with open(output_file, 'w', encoding='utf-8') as f:
        json.dump(custom_array, f, ensure_ascii=False, indent=2)
    print(f"Custom1 Length: {len(custom_array)} ")
    print(f"Merge completed. Custom array saved to {output_file}")

# Define your input files, output file, unique key, and desired fields
input_files = ["panales_master_1.json", "panales_master_2.json"]
output_file = f"master_panales.json"
unique_key = "codigoBarras"  # Replace with the unique key in your JSON
fields = ["seccion_numero", "codigoBarras", "categoria_real", "proveedor_num", "pasillo", "pedir", "proveedor_num"]  # Desired fields
os.remove("panales_master_1.json")
os.remove("panales_master_2.json")

# Run the function
merge_json_files_to_custom_array(input_files, output_file, unique_key, fields)