diff --git a/src/spiffworkflow_backend/scripts/get_data_sizes.py b/src/spiffworkflow_backend/scripts/get_data_sizes.py index 038c6f8a..2f28ffa7 100644 --- a/src/spiffworkflow_backend/scripts/get_data_sizes.py +++ b/src/spiffworkflow_backend/scripts/get_data_sizes.py @@ -5,11 +5,12 @@ from spiffworkflow_backend.models.script_attributes_context import ( ScriptAttributesContext, ) from spiffworkflow_backend.scripts.script import Script -#from spiffworkflow_backend.servces.process_instance_processor import ProcessInstanceProcessor from spiffworkflow_backend.services.process_instance_processor import ( ProcessInstanceProcessor, ) +# from spiffworkflow_backend.servces.process_instance_processor import ProcessInstanceProcessor + class GetDataSizes(Script): """GetDataSizes.""" @@ -32,6 +33,10 @@ class GetDataSizes(Script): """Run.""" task = script_attributes_context.task return { - "cumulative_task_data_size": ProcessInstanceProcessor.get_task_data_size(task.workflow), - "python_env_size": ProcessInstanceProcessor.get_python_env_size(task.workflow), + "cumulative_task_data_size": ProcessInstanceProcessor.get_task_data_size( + task.workflow + ), + "python_env_size": ProcessInstanceProcessor.get_python_env_size( + task.workflow + ), } diff --git a/src/spiffworkflow_backend/services/process_instance_processor.py b/src/spiffworkflow_backend/services/process_instance_processor.py index 1dfea5a5..0d0a641e 100644 --- a/src/spiffworkflow_backend/services/process_instance_processor.py +++ b/src/spiffworkflow_backend/services/process_instance_processor.py @@ -1624,7 +1624,9 @@ class ProcessInstanceProcessor: @classmethod def get_python_env_size(cls, bpmn_process_instance: BpmnWorkflow) -> int: - user_defined_state = bpmn_process_instance.script_engine.environment.user_defined_state() + user_defined_state = ( + bpmn_process_instance.script_engine.environment.user_defined_state() + ) return cls._get_data_size(user_defined_state) def check_task_data_size(self) -> None: