📦 EqualifyEverything / process-controller

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77import json
from utils.watch import logger
from data.insert import insert_axe_items, insert_axe_nodes, insert_axe_subnodes


def process_items(url_id, scan_event_id, items, item_type):
    success = True
    for item in items:
        nodes = process_nodes(url_id, scan_event_id, item["nodes"])
        if not nodes:
            success = False

        item_processed = {
            "type": item_type,
            "impact": item["impact"],
            "tags": item["tags"],
            "area": item["id"],
            "nodes": nodes
        }

        if not insert_axe_items(scan_event_id, url_id, item_type, item["id"], item["impact"], item["tags"]):
            success = False
            logger.debug("No nodes, items processed successfully")

    return success


def process_nodes(url_id, scan_event_id, nodes):
    success = True
    processed_nodes = []
    for node in nodes:
        subnodes_all = process_subnodes(url_id, scan_event_id, node["all"], "all")
        subnodes_any = process_subnodes(url_id, scan_event_id, node["any"], "any")
        subnodes_none = process_subnodes(url_id, scan_event_id, node["none"], "none")

        if not (subnodes_all and subnodes_any and subnodes_none):
            success = False

        processed_node = {
            "html": node["html"],
            "impact": node["impact"],
            "target": node["target"],
            "data": node.get("data", {}),  # Use .get() method with default value
            "all": subnodes_all,
            "any": subnodes_any,
            "none": subnodes_none
        }

        if not insert_axe_nodes(scan_event_id, url_id, node["html"], node["impact"], node["target"], json.dumps(node.get("data", {})), None):  # Convert the dictionary to JSON string
            success = False
            logger.debug("No subnodes, nodes processed successfully")

        processed_nodes.append(processed_node)

    return processed_nodes if success else False


def process_subnodes(url_id, scan_event_id, subnodes, node_type):
    success = True
    processed_subnodes = []
    for subnode in subnodes:
        processed_subnode = {
            "node_id": subnode["id"],
            "impact": subnode["impact"],
            "message": subnode["message"],
            "data": subnode["data"],
            "related_nodes": subnode["relatedNodes"],
            "node_type": node_type
        }

        if not insert_axe_subnodes(scan_event_id, url_id, json.dumps(subnode["data"]), subnode["id"], subnode["impact"], subnode["message"], node_type, subnode["relatedNodes"]):
            success = False

        processed_subnodes.append(processed_subnode)

    return processed_subnodes if success else False