No need to avoid circular imports

This commit is contained in:
Daniel Goldberg 2019-11-13 11:10:26 +02:00
parent 31a60b12ff
commit 4f59a85f0b
2 changed files with 2 additions and 5 deletions

View File

@ -2,17 +2,16 @@ from abc import abstractmethod
from infection_monkey.config import WormConfiguration
from infection_monkey.utils.plugins.plugin import Plugin
import infection_monkey.network
class HostFinger(Plugin):
@staticmethod
def base_package_file():
import infection_monkey.network # avoid circular imports
return infection_monkey.network.__file__
@staticmethod
def base_package_name():
import infection_monkey.network # avoid circular imports
return infection_monkey.network.__package__
@property

View File

@ -7,7 +7,7 @@ from infection_monkey.utils.environment import is_windows_os
from infection_monkey.config import WormConfiguration
from infection_monkey.telemetry.attack.t1064_telem import T1064Telem
from infection_monkey.utils.plugins.plugin import Plugin
import infection_monkey.post_breach.actions
LOG = logging.getLogger(__name__)
__author__ = 'VakarisZ'
@ -22,12 +22,10 @@ class PBA(Plugin):
@staticmethod
def base_package_name():
import infection_monkey.post_breach.actions # avoid circular imports
return infection_monkey.post_breach.actions.__package__
@staticmethod
def base_package_file():
import infection_monkey.post_breach.actions
return infection_monkey.post_breach.actions.__file__
def __init__(self, name="unknown", linux_cmd="", windows_cmd=""):