Agent: Modify PBA base class to accept options in its run method

This commit is contained in:
Shreya Malviya 2022-03-30 13:42:58 +05:30
parent 0be6af2d5c
commit 9f8463f707
1 changed files with 2 additions and 2 deletions

View File

@ -1,6 +1,6 @@
import logging
import subprocess
from typing import Iterable
from typing import Dict, Iterable
from common.utils.attack_utils import ScanStatus
from infection_monkey.i_puppet.i_puppet import PostBreachData
@ -30,7 +30,7 @@ class PBA:
self.pba_data = []
self.telemetry_messenger = telemetry_messenger
def run(self) -> Iterable[PostBreachData]:
def run(self, options: Dict) -> Iterable[PostBreachData]:
"""
Runs post breach action command
"""