deploy_master_system.py

import requests
import json
import time
import glob
import os

ZEEBE_BASE = "http://localhost:8081/v2"

def deploy_all():
    print("--- Deploying Full System ---")
    url = f"{ZEEBE_BASE}/deployments"
    
    files = []
    
    # 1. Master BPMN (Updated to use master_eligibility.bpmn)
    bpmn_path = "master_eligibility.bpmn"
    if os.path.exists(bpmn_path):
        files.append(('resources', (bpmn_path, open(bpmn_path, 'rb'), 'application/xml')))
    else:
        print(f"Error: {bpmn_path} not found.")
        return False

    # 1.5 Forms
    form_files = glob.glob("*.form")
    for f in form_files:
        print(f"Found form: {f}")
        files.append(('resources', (f, open(f, 'rb'), 'application/json')))
        
    # 2. All DMNs
    dmn_files = glob.glob("dmn_output/*.dmn")
    print(f"Found {len(dmn_files)} DMN files.")
    
    # We might need to batch this if too large
    # Let's try sending all? Or batches of 10.
    # Zeebe might have a limit. 
    # Let's do batches of 20
    
    # Close BPMN file after reading? Or keep open?
    # Actually requests reads them.
    
    # Let's map DMN files to tuples
    dmn_tuples = []
    for f in dmn_files:
        dmn_tuples.append(('resources', (os.path.basename(f), open(f, 'rb'), 'application/xml')))
    
    # Deploy BPMN + First 10 DMNs
    batch_size = 20
    current_batch = [files[0]] # Always include BPMN? No, BPMN only needs to be deployed once. 
    # Just deploy BPMN + Forms first
    try:
        print(f"Deploying {len(files)} process resources (BPMN + Forms)...")
        # We need to ensure we pass the list correctly to requests
        # requests.post takes 'files' as a list of tuples
        resp = requests.post(url, files=files)
        if resp.status_code != 200:
            print(f"BPMN/Form Deployment failed: {resp.text}")
            return False
        else:
            print("BPMN/Form Deployment success.")
            print(json.dumps(resp.json(), indent=2))
    except Exception as e:
        print(f"Failed to deploy BPMN: {e}")
        return False

    # Deploy DMNs in batches
    for i in range(0, len(dmn_tuples), batch_size):
        batch = dmn_tuples[i:i+batch_size]
        print(f"Deploying batch {i} to {i+len(batch)}...")
        try:
            resp = requests.post(url, files=batch)
            if resp.status_code != 200:
                print(f"Batch failed: {resp.text}")
            else:
                print("Batch success.")
        except Exception as e:
            print(f"Batch error: {e}")

    # Cleanup
    for f_tuple in files:
        # f_tuple is ('resources', (filename, filehandle, mimetype))
        f_tuple[1][1].close()
    for item in dmn_tuples:
        item[1][1].close()
        
    return True

def start_master_process():
    print("\n--- Starting Master Process ---")
    url = f"{ZEEBE_BASE}/process-instances"
    
    # Get list of Rule IDs from filenames
    # S2R001.dmn -> S2R001
    import glob
    rule_ids = [os.path.basename(f).replace(".dmn", "") for f in glob.glob("dmn_output/*.dmn")]
    # Rule IDs in DMN are usually just the number part as 'item'?
    # In BPMN we used: decisionId="Decision_" + item
    # In DMN file: decision id="Decision_S2R001"
    # So if item="S2R001", then "Decision_S2R001" matches.
    
    payload = {
        "processDefinitionId": "Process_Master_Eligibility",
        "variables": {
            "ruleList": rule_ids,
            # Test Data: A family of 3 with moderate income
            "household": {
                "city": "NYC",
                "members": 3
            },
            "person": {
                "age": 30,
                "disabled": False,
                "blind": False
            },
            "income": 2000.0, # Monthly -> Yearly 24000
            "expenseAmount": 100.0,
            "expenseType": "ChildCare"
        }
    }
    
    try:
        response = requests.post(url, json=payload)
        if response.status_code == 200:
            key = response.json().get('processInstanceKey')
            print(f"Started Master Instance: {key}")
            return key
        else:
            print(f"Start failed: {response.text}")
            return None
    except Exception as e:
        print(f"Error: {e}")
        return None

if __name__ == "__main__":
    if deploy_all():
        start_master_process()
        print("\nNOTE: Ensure 'python aggregation_worker.py' is running in another terminal!")
        print("Then check Tasklist: http://localhost:8083 (demo/demo)")