84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
"""Example_data."""
|
|
import glob
|
|
import os
|
|
|
|
from flask import current_app
|
|
|
|
from spiffworkflow_backend.models.process_model import ProcessModelInfo
|
|
from spiffworkflow_backend.services.process_model_service import ProcessModelService
|
|
from spiffworkflow_backend.services.spec_file_service import SpecFileService
|
|
|
|
|
|
class ExampleDataLoader:
|
|
"""ExampleDataLoader."""
|
|
|
|
def create_spec(
|
|
self,
|
|
id: str,
|
|
display_name: str = "",
|
|
description: str = "",
|
|
filepath: None = None,
|
|
master_spec: bool = False,
|
|
process_group_id: str = "",
|
|
display_order: int = 0,
|
|
from_tests: bool = False,
|
|
standalone: bool = False,
|
|
library: bool = False,
|
|
) -> ProcessModelInfo:
|
|
"""Assumes that a directory exists in static/bpmn with the same name as the given id.
|
|
|
|
further assumes that the [id].bpmn is the primary file for the process model.
|
|
returns an array of data models to be added to the database.
|
|
"""
|
|
global file
|
|
spec = ProcessModelInfo(
|
|
id=id,
|
|
display_name=display_name,
|
|
description=description,
|
|
process_group_id=process_group_id,
|
|
display_order=display_order,
|
|
is_master_spec=master_spec,
|
|
standalone=standalone,
|
|
library=library,
|
|
primary_file_name="",
|
|
primary_process_id="",
|
|
is_review=False,
|
|
libraries=[],
|
|
)
|
|
workflow_spec_service = ProcessModelService()
|
|
workflow_spec_service.add_spec(spec)
|
|
|
|
if not filepath and not from_tests:
|
|
filepath = os.path.join(current_app.root_path, "static", "bpmn", id, "*.*")
|
|
if not filepath and from_tests:
|
|
filepath = os.path.join(
|
|
current_app.instance_path, "..", "..", "tests", "data", id, "*.*"
|
|
)
|
|
|
|
files = glob.glob(filepath)
|
|
for file_path in files:
|
|
if os.path.isdir(file_path):
|
|
continue # Don't try to process sub directories
|
|
|
|
noise, file_extension = os.path.splitext(file_path)
|
|
filename = os.path.basename(file_path)
|
|
is_primary = filename.lower() == id + ".bpmn"
|
|
file = None
|
|
try:
|
|
file = open(file_path, "rb")
|
|
data = file.read()
|
|
SpecFileService.add_file(
|
|
workflow_spec=spec, file_name=filename, binary_data=data
|
|
)
|
|
if is_primary:
|
|
SpecFileService.set_primary_bpmn(spec, filename, data)
|
|
workflow_spec_service = ProcessModelService()
|
|
workflow_spec_service.update_spec(spec)
|
|
except IsADirectoryError:
|
|
# Ignore sub directories
|
|
pass
|
|
finally:
|
|
if file:
|
|
file.close()
|
|
return spec
|