checker.py

# Load the json file generated_form_schemas/F01-MHPSS_Baseline.json and extract all the concept keys and their corresponding UUIDs.
# Check if each UUID exists in the OpenMRS database using the API call "http://localhost/openmrs/ws/rest/v1/concept/{uuid}".
# Update the status of each ID in a CSV file called "concept_status.csv".

import json
import requests
import csv

# filepath to the json files
json_file_path = (
    "./LIME-EMR/sites/mosul/configs/openmrs/initializer_config/ampathforms/"
)

# List of JSON files
json_files = [
    "F01-MHPSS_Baseline.json",
    "F02-MHPSS_Follow-up.json",
    "F03-mhGAP_Baseline.json",
    "F04-mhGAP_Follow-up.json",
    "F05-MH_Closure.json",
    "F06-PHQ-9.json",
]


def find_key_in_json(data, key):
    found_values = []

    if isinstance(data, dict):
        for k, v in data.items():
            if k == key:
                found_values.append((k, v))
            found_values.extend(find_key_in_json(v, key))

    elif isinstance(data, list):
        for item in data:
            found_values.extend(find_key_in_json(item, key))

    return found_values


def main():
    # List to store concept status
    concept_status = []

    # Load the json files (Nutrition.json, F01-MHPSS_Baseline.json, etc.)
    # Create a loop to iterate over all json files
    for form in json_files:
        print(f"Processing {form}")
        # Open the json file and load its data
        with open(json_file_path + form, "r") as f:
            data = json.load(f)

            # Extract concept keys and UUIDs
            concept_uuids = find_key_in_json(data, "concept")

            for concept_key, uuid in concept_uuids:
                # Make API call to check status and give time to OpenMRS to respond
                response = requests.get(
                    f"http://msf-ocg-openmrs3-dev.westeurope.cloudapp.azure.com/openmrs/ws/rest/v1/concept/{uuid}",
                    auth=("admin", "Admin123"),
                )
                if response.status_code == 200:
                    # Get value of key 'display' from response
                    name = response.json().get("display")
                    concept_status.append(
                        {"form": form, "uuid": uuid, "status": "Exists", "name": name}
                    )
                    print(
                        f"Concept {name} with UUID {uuid} exists in the OpenMRS database"
                    )
                else:
                    name = "not found"
                    concept_status.append(
                        {"form": form, "uuid": uuid, "status": "Not found"}
                    )
                    print(
                        f"Error checking concept with UUID {uuid}: {response.status_code}"
                    )

                # Update the status of each ID in a CSV file
                with open("concept_status.csv", "w", newline="") as csvfile:
                    fieldnames = ["form", "uuid", "status", "name"]
                    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
                    writer.writeheader()
                    writer.writerows(concept_status)

    print("Concept status updated successfully!")


if __name__ == "__main__":
    main()