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()