71 lines
3.1 KiB
Python
71 lines
3.1 KiB
Python
import glob
|
|
import os
|
|
from typing import NoReturn
|
|
|
|
from lxml import etree
|
|
from spiffworkflow_backend import create_app
|
|
from spiffworkflow_backend.services.file_system_service import FileSystemService
|
|
|
|
|
|
def find_script_tasks_with_get_current_user(bpmn_file_path: str, root_path: str) -> None:
|
|
from spiffworkflow_backend.services.spec_file_service import SpecFileService
|
|
|
|
try:
|
|
with open(bpmn_file_path, "rb") as bpmn_file:
|
|
binary_data = bpmn_file.read()
|
|
tree = SpecFileService.get_etree_from_xml_bytes(binary_data)
|
|
check_script_and_prescript_elements(tree, bpmn_file_path, root_path)
|
|
except etree.XMLSyntaxError:
|
|
print(f"Error parsing XML in file {bpmn_file_path}. Please check for syntax issues.")
|
|
return
|
|
|
|
|
|
def check_script_and_prescript_elements(tree, bpmn_file_path: str, root_path: str) -> None:
|
|
# Define the namespace map to search for elements
|
|
nsmap = {
|
|
"bpmn": "http://www.omg.org/spec/BPMN/20100524/MODEL",
|
|
"spiffworkflow": "http://spiffworkflow.org/bpmn/schema/1.0/core",
|
|
}
|
|
# Find all script tasks and preScript elements
|
|
script_tasks = tree.xpath("//bpmn:scriptTask", namespaces=nsmap)
|
|
pre_scripts = tree.xpath("//spiffworkflow:preScript", namespaces=nsmap)
|
|
|
|
# Check script tasks for get_current_user() calls
|
|
for task in script_tasks:
|
|
script = task.find("bpmn:script", namespaces=nsmap)
|
|
if script is not None and script.text is not None and "get_current_user()" in script.text:
|
|
print(f'Found get_current_user() in script task {task.get("id")} of file {bpmn_file_path}')
|
|
|
|
# Check preScript elements for get_current_user() calls
|
|
check_scripts_for_get_current_user(pre_scripts, bpmn_file_path, "preScript", root_path)
|
|
post_scripts = tree.xpath("//spiffworkflow:postScript", namespaces=nsmap)
|
|
check_scripts_for_get_current_user(post_scripts, bpmn_file_path, "postScript", root_path)
|
|
|
|
|
|
def check_scripts_for_get_current_user(scripts, bpmn_file_path: str, script_type: str, root_path: str) -> None:
|
|
for script in scripts:
|
|
if script is not None and script.text is not None and "get_current_user()" in script.text:
|
|
# Get the parent of the parent to find the actual BPMN element
|
|
parent = script.getparent().getparent()
|
|
relative_path = os.path.relpath(bpmn_file_path, root_path)
|
|
tag_without_namespace = etree.QName(parent.tag).localname
|
|
if tag_without_namespace not in ["manualTask", "userTask"]:
|
|
print(
|
|
f"Found get_current_user() in {script_type} of {tag_without_namespace} "
|
|
f'with id {parent.get("id")} in file {relative_path}'
|
|
)
|
|
|
|
|
|
def main() -> NoReturn:
|
|
app = create_app()
|
|
with app.app_context():
|
|
hot_dir = FileSystemService.root_path()
|
|
# Search for BPMN files and check for get_current_user() calls in script tasks
|
|
bpmn_files = glob.glob(os.path.expanduser(f"{hot_dir}/**/*.bpmn"), recursive=True)
|
|
for bpmn_file in bpmn_files:
|
|
find_script_tasks_with_get_current_user(bpmn_file, hot_dir)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|