Skip to content

Documentation for Staticarbitrationpolicy Module

SAP

Bases: ArbitrationPolicy

Source code in nebula/core/situationalawareness/awareness/arbitrationpolicies/staticarbitrationpolicy.py 
 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
class SAP(ArbitrationPolicy):  # Static Arbitatrion Policy
    def __init__(self, verbose):
        self._verbose = verbose
        # Define static weights for SA Agents from SA Components
        self.agent_weights = {"SATraining": 1, "SANetwork": 2, "SAReputation": 3}

    async def init(self, config):
        pass

    async def _get_agent_category(self, sa_command: SACommand) -> str:
        """
        Extract agent category name.
        Example: "SATraining_Agent1" → "SATraining"
        """
        full_name = await sa_command.get_owner()
        return full_name.split("_")[0] if "_" in full_name else full_name

    async def tie_break(self, sac1: SACommand, sac2: SACommand) -> bool:
        """
        Tie break conflcited SA Commands
        """
        if self._verbose:
            logging.info(
                f"Tie break between ({await sac1.get_owner()}, {sac1.get_action().value}) & ({await sac2.get_owner()}, {sac2.get_action().value})"
            )

        async def get_weight(cmd):
            category = await self._get_agent_category(cmd)
            return self.agent_weights.get(category, 0)

        if await get_weight(sac1) > await get_weight(sac2):
            if self._verbose:
                logging.info(
                    f"Tie break resolved, SA Command choosen ({await sac1.get_owner()}, {sac1.get_action().value})"
                )
            return True
        else:
            if self._verbose:
                logging.info(
                    f"Tie break resolved, SA Command choosen ({await sac2.get_owner()}, {sac2.get_action().value})"
                )
            return False

tie_break(sac1, sac2) async

Tie break conflcited SA Commands

Source code in nebula/core/situationalawareness/awareness/arbitrationpolicies/staticarbitrationpolicy.py 
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
async def tie_break(self, sac1: SACommand, sac2: SACommand) -> bool:
    """
    Tie break conflcited SA Commands
    """
    if self._verbose:
        logging.info(
            f"Tie break between ({await sac1.get_owner()}, {sac1.get_action().value}) & ({await sac2.get_owner()}, {sac2.get_action().value})"
        )

    async def get_weight(cmd):
        category = await self._get_agent_category(cmd)
        return self.agent_weights.get(category, 0)

    if await get_weight(sac1) > await get_weight(sac2):
        if self._verbose:
            logging.info(
                f"Tie break resolved, SA Command choosen ({await sac1.get_owner()}, {sac1.get_action().value})"
            )
        return True
    else:
        if self._verbose:
            logging.info(
                f"Tie break resolved, SA Command choosen ({await sac2.get_owner()}, {sac2.get_action().value})"
            )
        return False