spiff-arena/bin/import_tickets_for_command_line.py

"""Grabs tickets from csv and makes process instances."""
import csv

from flask_bpmn.models.db import db

from spiffworkflow_backend import get_hacked_up_app_for_script
from spiffworkflow_backend.models.process_instance import ProcessInstanceModel
from spiffworkflow_backend.models.user import UserModel
from spiffworkflow_backend.services.process_instance_processor import (
    ProcessInstanceProcessor,
)
from spiffworkflow_backend.services.process_instance_service import (
    ProcessInstanceService,
)


def print_process_instance_count(process_model_identifier_ticket: str) -> None:
    """Print process instance count."""
    process_instances = ProcessInstanceModel.query.filter_by(
        process_model_identifier=process_model_identifier_ticket
    ).all()
    process_instance_count = len(process_instances)
    print(f"process_instance_count: {process_instance_count}")


def main():
    """Main."""
    app = get_hacked_up_app_for_script()
    with app.app_context():

        process_model_identifier_ticket = "ticket"
        db.session.query(ProcessInstanceModel).filter(
            ProcessInstanceModel.process_model_identifier
            == process_model_identifier_ticket
        ).delete()
        db.session.commit()

        """Print process instance count."""
        process_instances = ProcessInstanceModel.query.filter_by(
            process_model_identifier=process_model_identifier_ticket
        ).all()
        process_instance_count = len(process_instances)
        print(f"process_instance_count: {process_instance_count}")

        columns_to_data_key_mappings = {
            "Month": "month",
            "MS": "milestone",
            "ID": "req_id",
            "Dev Days": "dev_days",
            "Feature": "feature",
            "Priority": "priority",
        }
        columns_to_header_index_mappings = {}

        user = UserModel.query.first()

        with open("tests/files/tickets.csv") as infile:
            reader = csv.reader(infile, delimiter=",")

            # first row is garbage
            next(reader)

            header = next(reader)
            for column_name in columns_to_data_key_mappings:
                columns_to_header_index_mappings[column_name] = header.index(
                    column_name
                )
            id_index = header.index("ID")
            priority_index = header.index("Priority")
            print(f"header: {header}")
            for row in reader:
                ticket_identifier = row[id_index]
                priority = row[priority_index]
                print(f"ticket_identifier: {ticket_identifier}")
                print(f"priority: {priority}")

                process_instance = ProcessInstanceService.create_process_instance(
                    process_model_identifier_ticket,
                    user,
                    process_group_identifier="sartography-admin",
                )
                processor = ProcessInstanceProcessor(process_instance)

                processor.do_engine_steps()
                # processor.save()

                for (
                    column_name,
                    desired_data_key,
                ) in columns_to_data_key_mappings.items():
                    appropriate_index = columns_to_header_index_mappings[column_name]
                    processor.bpmn_process_instance.data[desired_data_key] = row[
                        appropriate_index
                    ]

                print(f"datas: {processor.bpmn_process_instance.data}")
                if processor.bpmn_process_instance.data["month"] == "":
                    continue

                # you at least need a month, or else this row in the csv is considered garbage
                # if processor.bpmn_process_instance.data["month"] is None:
                #     continue

                processor.save()


# if __name__ == "__main__":
main()