spiffworkflow-backend/bin/save_all_bpmn.py

"""Grabs tickets from csv and makes process instances."""
from spiffworkflow_backend import get_hacked_up_app_for_script
from spiffworkflow_backend.services.process_model_service import ProcessModelService
from spiffworkflow_backend.services.spec_file_service import SpecFileService

# from lxml.etree import Element as EtreeElement


def main():
    """Main."""
    app = get_hacked_up_app_for_script()
    with app.app_context():
        failing_process_models = []
        process_models = ProcessModelService().get_process_models()
        for process_model in process_models:
            if process_model.primary_file_name:

                bpmn_xml_file_contents = SpecFileService.get_data(
                    process_model, process_model.primary_file_name
                )
                bad_files = [
                    "B.1.0.bpmn",
                    "C.1.0.bpmn",
                    "C.2.0.bpmn",
                    "C.6.0.bpmn",
                    "TC-5.1.bpmn",
                ]
                if process_model.primary_file_name in bad_files:
                    continue
                print(f"primary_file_name: {process_model.primary_file_name}")
                try:
                    SpecFileService.update_file(
                        process_model,
                        process_model.primary_file_name,
                        bpmn_xml_file_contents,
                    )
                except Exception as ex:
                    failing_process_models.append(
                        (
                            f"{process_model.process_group_id}/{process_model.id}/{process_model.primary_file_name}",
                            str(ex),
                        )
                    )
                # files = SpecFileService.get_files(
                #     process_model, extension_filter="bpmn"
                # )
                # bpmn_etree_element: EtreeElement = (
                #     SpecFileService.get_etree_element_from_binary_data(
                #         bpmn_xml_file_contents, process_model.primary_file_name
                #     )
                # )
                # if len(files) == 1:
                # try:
                #     new_bpmn_process_identifier = (
                #         SpecFileService.get_bpmn_process_identifier(
                #             bpmn_etree_element
                #         )
                #     )
                #     if (
                #         process_model.primary_process_id
                #         != new_bpmn_process_identifier
                #     ):
                #         print(
                #             "primary_process_id: ", process_model.primary_process_id
                #         )
                #         # attributes_to_update = {
                #         #     "primary_process_id": new_bpmn_process_identifier
                #         # }
                #         # ProcessModelService().update_spec(
                #         #     process_model, attributes_to_update
                #         # )
                # # except Exception as exception:
                # except Exception:
                #     print(f"BAD ONE: {process_model.id}")
                #     # raise exception
            else:
                failing_process_models.append(
                    (
                        f"{process_model.process_group_id}/{process_model.id}",
                        "primary_file_name not set",
                    )
                )
        for bpmn_errors in failing_process_models:
            print(bpmn_errors)
        if len(failing_process_models) > 0:
            exit(1)


if __name__ == "__main__":
    main()