From e5a1193233f0ba327dc9d220fa32ed28465f4e52 Mon Sep 17 00:00:00 2001 From: Thorsten Sick Date: Thu, 28 Oct 2021 16:55:30 +0200 Subject: [PATCH] nicer code --- plugins/base/attack.py | 5 ++++- plugins/base/plugin_base.py | 1 - plugins/base/sensor.py | 1 - 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/plugins/base/attack.py b/plugins/base/attack.py index f8b21de..5747107 100644 --- a/plugins/base/attack.py +++ b/plugins/base/attack.py @@ -64,7 +64,10 @@ class AttackPlugin(BasePlugin): """ Inits metasploit """ if self.needs_metasploit(): - self.metasploit = MetasploitInstant(self.metasploit_password, attack_logger=self.attack_logger, attacker=self.attacker_machine_plugin, username=self.metasploit_user) + self.metasploit = MetasploitInstant(self.metasploit_password, + attack_logger=self.attack_logger, + attacker=self.attacker_machine_plugin, + username=self.metasploit_user) # If metasploit requirements are not set, self.metasploit stay None and using metasploit from a plugin not having the requirements will trigger an exception def copy_to_attacker_and_defender(self): diff --git a/plugins/base/plugin_base.py b/plugins/base/plugin_base.py index 75c9f27..e8a510b 100644 --- a/plugins/base/plugin_base.py +++ b/plugins/base/plugin_base.py @@ -8,7 +8,6 @@ from app.exceptions import PluginError # type: ignore import app.exceptions # type: ignore - class BasePlugin(): """ Base class for plugins """ diff --git a/plugins/base/sensor.py b/plugins/base/sensor.py index 684ae61..7b999b7 100644 --- a/plugins/base/sensor.py +++ b/plugins/base/sensor.py @@ -6,7 +6,6 @@ from typing import Optional from plugins.base.plugin_base import BasePlugin - class SensorPlugin(BasePlugin): """ A sensor will be running on the target machine and monitor attacks. To remote control those sensors there are sensor plugins. This is the base class for them