From d0fda6b9e5e4a3cc17fad6bf8eb5cbf31a47926f Mon Sep 17 00:00:00 2001 From: VakarisZ Date: Thu, 15 Oct 2020 14:57:59 +0300 Subject: [PATCH] Fixed import styles in gevent modifications. --- monkey/monkey_island.py | 1 + monkey/monkey_island/cc/services/infection_lifecycle.py | 4 ++-- .../services/reporting/report_generation_synchronisation.py | 1 + 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/monkey/monkey_island.py b/monkey/monkey_island.py index 44b347e8c..cd452066c 100644 --- a/monkey/monkey_island.py +++ b/monkey/monkey_island.py @@ -1,4 +1,5 @@ from gevent import monkey as gevent_monkey + gevent_monkey.patch_all() from monkey_island.cc.main import main diff --git a/monkey/monkey_island/cc/services/infection_lifecycle.py b/monkey/monkey_island/cc/services/infection_lifecycle.py index 298db3e25..f29b9ba71 100644 --- a/monkey/monkey_island/cc/services/infection_lifecycle.py +++ b/monkey/monkey_island/cc/services/infection_lifecycle.py @@ -8,8 +8,8 @@ from monkey_island.cc.resources.test.utils.telem_store import TestTelemStore from monkey_island.cc.services.config import ConfigService from monkey_island.cc.services.node import NodeService from monkey_island.cc.services.reporting.report import ReportService -from monkey_island.cc.services.reporting.report_generation_synchronisation import safe_generate_reports, \ - is_report_being_generated +from monkey_island.cc.services.reporting.report_generation_synchronisation import ( + is_report_being_generated, safe_generate_reports) logger = logging.getLogger(__name__) diff --git a/monkey/monkey_island/cc/services/reporting/report_generation_synchronisation.py b/monkey/monkey_island/cc/services/reporting/report_generation_synchronisation.py index 944989b53..990df24e4 100644 --- a/monkey/monkey_island/cc/services/reporting/report_generation_synchronisation.py +++ b/monkey/monkey_island/cc/services/reporting/report_generation_synchronisation.py @@ -1,4 +1,5 @@ import logging + from gevent.lock import BoundedSemaphore logger = logging.getLogger(__name__)