From 9846477bea9d9e8665fffcb039f4cbd9ba6186ee Mon Sep 17 00:00:00 2001
From: Shay Nehmad <shay.nehmad@guardicore.com>
Date: Wed, 15 Jul 2020 18:46:04 +0300
Subject: [PATCH] Change to isort from flake8-import-order

fix issues in envs/ as well
---
 .travis.yml                                   |  5 +++-
 .../analyzers/performance_analyzer.py         |  3 +-
 .../island_client/monkey_island_client.py     |  4 +--
 .../island_client/monkey_island_requests.py   | 11 ++++---
 .../log_handlers/test_logs_handler.py         |  6 ++--
 envs/monkey_zoo/blackbox/test_blackbox.py     | 30 ++++++++++++-------
 .../performance/endpoint_performance_test.py  | 12 +++++---
 .../tests/performance/map_generation.py       |  9 ++++--
 .../map_generation_from_telemetries.py        |  6 ++--
 .../performance/performance_test_workflow.py  |  6 ++--
 .../tests/performance/report_generation.py    |  9 ++++--
 .../report_generation_from_telemetries.py     |  6 ++--
 .../sample_file_parser.py                     |  2 +-
 .../sample_multiplier/fake_monkey.py          |  4 +--
 .../sample_multiplier/sample_multiplier.py    | 12 ++++----
 .../test_fake_ip_generator.py                 |  4 +--
 .../performance/telemetry_performance_test.py | 15 ++++++----
 .../telemetry_performance_test_workflow.py    |  9 ++++--
 envs/os_compatibility/test_compatibility.py   |  4 +--
 monkey/infection_monkey/exploit/mssqlexec.py  |  1 -
 monkey/infection_monkey/network/info.py       |  2 +-
 monkey/monkey_island/cc/models/__init__.py    | 16 +++++-----
 .../monkey_island/cc/resources/local_run.py   |  2 +-
 23 files changed, 108 insertions(+), 70 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 4be40c472..5228275d4 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -20,7 +20,7 @@ install:
 # Python
 - pip freeze
 - pip install -r monkey/monkey_island/requirements.txt  # for unit tests
-- pip install flake8 pytest dlint flake8-import-order # for next stages
+- pip install flake8 pytest dlint isort # for next stages
 - pip install coverage  # for code coverage
 - pip install -r monkey/infection_monkey/requirements.txt  # for unit tests
 - pip install pipdeptree
@@ -69,6 +69,9 @@ script:
 - PYTHON_WARNINGS_AMOUNT_UPPER_LIMIT=120
 - if [ $(tail -n 1 flake8_warnings.txt) -gt $PYTHON_WARNINGS_AMOUNT_UPPER_LIMIT ]; then echo "Too many python linter warnings! Failing this build. Lower the amount of linter errors in this and try again. " && exit 1; fi
 
+## Check import order
+- python -m isort . -c -p common -p infection_monkey -p monkey_island
+
 ## Run unit tests
 - cd monkey  # This is our source dir
 - python -m pytest  # Have to use `python -m pytest` instead of `pytest` to add "{$builddir}/monkey/monkey" to sys.path.
diff --git a/envs/monkey_zoo/blackbox/analyzers/performance_analyzer.py b/envs/monkey_zoo/blackbox/analyzers/performance_analyzer.py
index 4a43ab6a5..e0354530e 100644
--- a/envs/monkey_zoo/blackbox/analyzers/performance_analyzer.py
+++ b/envs/monkey_zoo/blackbox/analyzers/performance_analyzer.py
@@ -3,7 +3,8 @@ from datetime import timedelta
 from typing import Dict
 
 from envs.monkey_zoo.blackbox.analyzers.analyzer import Analyzer
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
 
 LOGGER = logging.getLogger(__name__)
 
diff --git a/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py b/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
index b2370a345..5932022fb 100644
--- a/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
+++ b/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
@@ -1,11 +1,11 @@
 import json
-
 import logging
 from time import sleep
 
 from bson import json_util
 
-from envs.monkey_zoo.blackbox.island_client.monkey_island_requests import MonkeyIslandRequests
+from envs.monkey_zoo.blackbox.island_client.monkey_island_requests import \
+    MonkeyIslandRequests
 
 SLEEP_BETWEEN_REQUESTS_SECONDS = 0.5
 MONKEY_TEST_ENDPOINT = 'api/test/monkey'
diff --git a/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py b/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py
index 98acb5f7f..743cb4146 100644
--- a/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py
+++ b/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py
@@ -1,13 +1,12 @@
-from typing import Dict
+import functools
+import logging
 from datetime import timedelta
-
+from typing import Dict
 
 import requests
-import functools
 
-from envs.monkey_zoo.blackbox.island_client.supported_request_method import SupportedRequestMethod
-
-import logging
+from envs.monkey_zoo.blackbox.island_client.supported_request_method import \
+    SupportedRequestMethod
 
 # SHA3-512 of '1234567890!@#$%^&*()_nothing_up_my_sleeve_1234567890!@#$%^&*()'
 NO_AUTH_CREDS = '55e97c9dcfd22b8079189ddaeea9bce8125887e3237b800c6176c9afa80d2062' \
diff --git a/envs/monkey_zoo/blackbox/log_handlers/test_logs_handler.py b/envs/monkey_zoo/blackbox/log_handlers/test_logs_handler.py
index bae6a9adc..3f5cfc191 100644
--- a/envs/monkey_zoo/blackbox/log_handlers/test_logs_handler.py
+++ b/envs/monkey_zoo/blackbox/log_handlers/test_logs_handler.py
@@ -2,8 +2,10 @@ import logging
 import os
 import shutil
 
-from envs.monkey_zoo.blackbox.log_handlers.monkey_log_parser import MonkeyLogParser
-from envs.monkey_zoo.blackbox.log_handlers.monkey_logs_downloader import MonkeyLogsDownloader
+from envs.monkey_zoo.blackbox.log_handlers.monkey_log_parser import \
+    MonkeyLogParser
+from envs.monkey_zoo.blackbox.log_handlers.monkey_logs_downloader import \
+    MonkeyLogsDownloader
 
 LOG_DIR_NAME = 'logs'
 LOGGER = logging.getLogger(__name__)
diff --git a/envs/monkey_zoo/blackbox/test_blackbox.py b/envs/monkey_zoo/blackbox/test_blackbox.py
index d4e21fb6d..45751452e 100644
--- a/envs/monkey_zoo/blackbox/test_blackbox.py
+++ b/envs/monkey_zoo/blackbox/test_blackbox.py
@@ -1,20 +1,28 @@
-import os
 import logging
-
-import pytest
+import os
 from time import sleep
 
-from envs.monkey_zoo.blackbox.analyzers.communication_analyzer import CommunicationAnalyzer
-from envs.monkey_zoo.blackbox.island_client.island_config_parser import IslandConfigParser
-from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
-from envs.monkey_zoo.blackbox.log_handlers.test_logs_handler import TestLogsHandler
+import pytest
+
+from envs.monkey_zoo.blackbox.analyzers.communication_analyzer import \
+    CommunicationAnalyzer
+from envs.monkey_zoo.blackbox.island_client.island_config_parser import \
+    IslandConfigParser
+from envs.monkey_zoo.blackbox.island_client.monkey_island_client import \
+    MonkeyIslandClient
+from envs.monkey_zoo.blackbox.log_handlers.test_logs_handler import \
+    TestLogsHandler
 from envs.monkey_zoo.blackbox.tests.exploitation import ExploitationTest
-from envs.monkey_zoo.blackbox.tests.performance.map_generation import MapGenerationTest
-from envs.monkey_zoo.blackbox.tests.performance.map_generation_from_telemetries import MapGenerationFromTelemetryTest
-from envs.monkey_zoo.blackbox.tests.performance.report_generation import ReportGenerationTest
+from envs.monkey_zoo.blackbox.tests.performance.map_generation import \
+    MapGenerationTest
+from envs.monkey_zoo.blackbox.tests.performance.map_generation_from_telemetries import \
+    MapGenerationFromTelemetryTest
+from envs.monkey_zoo.blackbox.tests.performance.report_generation import \
+    ReportGenerationTest
 from envs.monkey_zoo.blackbox.tests.performance.report_generation_from_telemetries import \
     ReportGenerationFromTelemetryTest
-from envs.monkey_zoo.blackbox.tests.performance.telemetry_performance_test import TelemetryPerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.telemetry_performance_test import \
+    TelemetryPerformanceTest
 from envs.monkey_zoo.blackbox.utils import gcp_machine_handlers
 
 DEFAULT_TIMEOUT_SECONDS = 5*60
diff --git a/envs/monkey_zoo/blackbox/tests/performance/endpoint_performance_test.py b/envs/monkey_zoo/blackbox/tests/performance/endpoint_performance_test.py
index b8793452d..e08ac2824 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/endpoint_performance_test.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/endpoint_performance_test.py
@@ -1,10 +1,14 @@
 import logging
 
-from envs.monkey_zoo.blackbox.analyzers.performance_analyzer import PerformanceAnalyzer
-from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
-from envs.monkey_zoo.blackbox.island_client.supported_request_method import SupportedRequestMethod
+from envs.monkey_zoo.blackbox.analyzers.performance_analyzer import \
+    PerformanceAnalyzer
+from envs.monkey_zoo.blackbox.island_client.monkey_island_client import \
+    MonkeyIslandClient
+from envs.monkey_zoo.blackbox.island_client.supported_request_method import \
+    SupportedRequestMethod
 from envs.monkey_zoo.blackbox.tests.basic_test import BasicTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
 
 LOGGER = logging.getLogger(__name__)
 
diff --git a/envs/monkey_zoo/blackbox/tests/performance/map_generation.py b/envs/monkey_zoo/blackbox/tests/performance/map_generation.py
index eb95fdc6a..926e5331e 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/map_generation.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/map_generation.py
@@ -1,9 +1,12 @@
 from datetime import timedelta
 
 from envs.monkey_zoo.blackbox.tests.exploitation import ExploitationTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test import PerformanceTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_workflow import PerformanceTestWorkflow
+from envs.monkey_zoo.blackbox.tests.performance.performance_test import \
+    PerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_workflow import \
+    PerformanceTestWorkflow
 
 MAX_ALLOWED_SINGLE_PAGE_TIME = timedelta(seconds=2)
 MAX_ALLOWED_TOTAL_TIME = timedelta(seconds=5)
diff --git a/envs/monkey_zoo/blackbox/tests/performance/map_generation_from_telemetries.py b/envs/monkey_zoo/blackbox/tests/performance/map_generation_from_telemetries.py
index 1b31a8962..1ee1b60da 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/map_generation_from_telemetries.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/map_generation_from_telemetries.py
@@ -1,7 +1,9 @@
 from datetime import timedelta
 
-from envs.monkey_zoo.blackbox.tests.performance.performance_test import PerformanceTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.performance_test import \
+    PerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
 from envs.monkey_zoo.blackbox.tests.performance.telemetry_performance_test_workflow import \
     TelemetryPerformanceTestWorkflow
 
diff --git a/envs/monkey_zoo/blackbox/tests/performance/performance_test_workflow.py b/envs/monkey_zoo/blackbox/tests/performance/performance_test_workflow.py
index 4e708ed9d..5f08c976c 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/performance_test_workflow.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/performance_test_workflow.py
@@ -1,7 +1,9 @@
 from envs.monkey_zoo.blackbox.tests.basic_test import BasicTest
 from envs.monkey_zoo.blackbox.tests.exploitation import ExploitationTest
-from envs.monkey_zoo.blackbox.tests.performance.endpoint_performance_test import EndpointPerformanceTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.endpoint_performance_test import \
+    EndpointPerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
 
 
 class PerformanceTestWorkflow(BasicTest):
diff --git a/envs/monkey_zoo/blackbox/tests/performance/report_generation.py b/envs/monkey_zoo/blackbox/tests/performance/report_generation.py
index e204cc29f..eec8f067d 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/report_generation.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/report_generation.py
@@ -1,9 +1,12 @@
 from datetime import timedelta
 
 from envs.monkey_zoo.blackbox.tests.exploitation import ExploitationTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test import PerformanceTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_workflow import PerformanceTestWorkflow
+from envs.monkey_zoo.blackbox.tests.performance.performance_test import \
+    PerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_workflow import \
+    PerformanceTestWorkflow
 
 MAX_ALLOWED_SINGLE_PAGE_TIME = timedelta(seconds=2)
 MAX_ALLOWED_TOTAL_TIME = timedelta(seconds=5)
diff --git a/envs/monkey_zoo/blackbox/tests/performance/report_generation_from_telemetries.py b/envs/monkey_zoo/blackbox/tests/performance/report_generation_from_telemetries.py
index abc2b35c2..1cba745bf 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/report_generation_from_telemetries.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/report_generation_from_telemetries.py
@@ -1,7 +1,9 @@
 from datetime import timedelta
 
-from envs.monkey_zoo.blackbox.tests.performance.performance_test import PerformanceTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.performance_test import \
+    PerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
 from envs.monkey_zoo.blackbox.tests.performance.telemetry_performance_test_workflow import \
     TelemetryPerformanceTestWorkflow
 
diff --git a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_file_parser.py b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_file_parser.py
index 70e25d8e7..0f0c3311f 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_file_parser.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_file_parser.py
@@ -1,7 +1,7 @@
 import json
 import logging
 from os import listdir, path
-from typing import List, Dict
+from typing import Dict, List
 
 from tqdm import tqdm
 
diff --git a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/fake_monkey.py b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/fake_monkey.py
index 1d140e396..efee81227 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/fake_monkey.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/fake_monkey.py
@@ -1,7 +1,7 @@
 import random
 
-from envs.monkey_zoo.blackbox.tests.performance.\
-    telem_sample_parsing.sample_multiplier.fake_ip_generator import FakeIpGenerator
+from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_multiplier.fake_ip_generator import \
+    FakeIpGenerator
 
 
 class FakeMonkey:
diff --git a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/sample_multiplier.py b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/sample_multiplier.py
index da3c22b05..e5b0a52cd 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/sample_multiplier.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/sample_multiplier.py
@@ -2,14 +2,16 @@ import copy
 import json
 import logging
 import sys
-from typing import List, Dict
+from typing import Dict, List
 
 from tqdm import tqdm
 
-from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_file_parser import SampleFileParser
-from envs.monkey_zoo.blackbox.tests.performance.\
-    telem_sample_parsing.sample_multiplier.fake_ip_generator import FakeIpGenerator
-from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_multiplier.fake_monkey import FakeMonkey
+from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_file_parser import \
+    SampleFileParser
+from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_multiplier.fake_ip_generator import \
+    FakeIpGenerator
+from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_multiplier.fake_monkey import \
+    FakeMonkey
 
 TELEM_DIR_PATH = './tests/performance/telemetry_sample'
 LOGGER = logging.getLogger(__name__)
diff --git a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/test_fake_ip_generator.py b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/test_fake_ip_generator.py
index d8adef827..02cf3a8eb 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/test_fake_ip_generator.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/telem_sample_parsing/sample_multiplier/test_fake_ip_generator.py
@@ -1,7 +1,7 @@
 from unittest import TestCase
 
-from envs.monkey_zoo.blackbox.tests.performance.\
-    telem_sample_parsing.sample_multiplier.fake_ip_generator import FakeIpGenerator
+from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_multiplier.fake_ip_generator import \
+    FakeIpGenerator
 
 
 class TestFakeIpGenerator(TestCase):
diff --git a/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test.py b/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test.py
index 699876cce..75802449e 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test.py
@@ -4,11 +4,16 @@ from datetime import timedelta
 
 from tqdm import tqdm
 
-from envs.monkey_zoo.blackbox.analyzers.performance_analyzer import PerformanceAnalyzer
-from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
-from envs.monkey_zoo.blackbox.island_client.supported_request_method import SupportedRequestMethod
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
-from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_file_parser import SampleFileParser
+from envs.monkey_zoo.blackbox.analyzers.performance_analyzer import \
+    PerformanceAnalyzer
+from envs.monkey_zoo.blackbox.island_client.monkey_island_client import \
+    MonkeyIslandClient
+from envs.monkey_zoo.blackbox.island_client.supported_request_method import \
+    SupportedRequestMethod
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.telem_sample_parsing.sample_file_parser import \
+    SampleFileParser
 
 LOGGER = logging.getLogger(__name__)
 
diff --git a/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test_workflow.py b/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test_workflow.py
index 6d09752ca..b63d904e1 100644
--- a/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test_workflow.py
+++ b/envs/monkey_zoo/blackbox/tests/performance/telemetry_performance_test_workflow.py
@@ -1,7 +1,10 @@
 from envs.monkey_zoo.blackbox.tests.basic_test import BasicTest
-from envs.monkey_zoo.blackbox.tests.performance.endpoint_performance_test import EndpointPerformanceTest
-from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import PerformanceTestConfig
-from envs.monkey_zoo.blackbox.tests.performance.telemetry_performance_test import TelemetryPerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.endpoint_performance_test import \
+    EndpointPerformanceTest
+from envs.monkey_zoo.blackbox.tests.performance.performance_test_config import \
+    PerformanceTestConfig
+from envs.monkey_zoo.blackbox.tests.performance.telemetry_performance_test import \
+    TelemetryPerformanceTest
 
 
 class TelemetryPerformanceTestWorkflow(BasicTest):
diff --git a/envs/os_compatibility/test_compatibility.py b/envs/os_compatibility/test_compatibility.py
index 5e5a38597..17d2d3735 100644
--- a/envs/os_compatibility/test_compatibility.py
+++ b/envs/os_compatibility/test_compatibility.py
@@ -1,7 +1,7 @@
 import pytest
 
-from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
-
+from envs.monkey_zoo.blackbox.island_client.monkey_island_client import \
+    MonkeyIslandClient
 
 machine_list = {
     "10.0.0.36": "centos_6",
diff --git a/monkey/infection_monkey/exploit/mssqlexec.py b/monkey/infection_monkey/exploit/mssqlexec.py
index 6bff6aced..3c8a8c3c1 100644
--- a/monkey/infection_monkey/exploit/mssqlexec.py
+++ b/monkey/infection_monkey/exploit/mssqlexec.py
@@ -4,7 +4,6 @@ import sys
 from time import sleep
 
 import pymssql
-
 from common.utils.exceptions import (ExploitingVulnerableMachineError,
                                      FailedExploitationError)
 from common.utils.exploit_enum import ExploitType
diff --git a/monkey/infection_monkey/network/info.py b/monkey/infection_monkey/network/info.py
index 0aafe0540..f3e411d5f 100644
--- a/monkey/infection_monkey/network/info.py
+++ b/monkey/infection_monkey/network/info.py
@@ -5,11 +5,11 @@ import struct
 from random import randint
 from subprocess import check_output
 
-import netifaces
 import psutil
 import requests
 from requests import ConnectionError
 
+import netifaces
 from common.network.network_range import CidrRange
 from infection_monkey.utils.environment import is_windows_os
 
diff --git a/monkey/monkey_island/cc/models/__init__.py b/monkey/monkey_island/cc/models/__init__.py
index 3ced32ea1..9d0114b78 100644
--- a/monkey/monkey_island/cc/models/__init__.py
+++ b/monkey/monkey_island/cc/models/__init__.py
@@ -2,6 +2,14 @@ from mongoengine import connect
 
 import monkey_island.cc.environment.environment_singleton as env_singleton
 
+from .command_control_channel import CommandControlChannel  # noqa: F401
+# Order of importing matters here, for registering the embedded and referenced documents before using them.
+from .config import Config  # noqa: F401
+from .creds import Creds  # noqa: F401
+from .monkey import Monkey  # noqa: F401
+from .monkey_ttl import MonkeyTtl  # noqa: F401
+from .pba_results import PbaResults  # noqa: F401
+
 # This section sets up the DB connection according to the environment.
 #   If testing, use mongomock which only emulates mongo. for more information, see
 #   http://docs.mongoengine.org/guide/mongomock.html .
@@ -10,11 +18,3 @@ if env_singleton.env.testing:  # See monkey_island.cc.environment.testing
     connect('mongoenginetest', host='mongomock://localhost')
 else:
     connect(db=env_singleton.env.mongo_db_name, host=env_singleton.env.mongo_db_host, port=env_singleton.env.mongo_db_port)
-
-from .command_control_channel import CommandControlChannel  # noqa: F401
-# Order of importing matters here, for registering the embedded and referenced documents before using them.
-from .config import Config  # noqa: F401
-from .creds import Creds  # noqa: F401
-from .monkey import Monkey  # noqa: F401
-from .monkey_ttl import MonkeyTtl  # noqa: F401
-from .pba_results import PbaResults  # noqa: F401
diff --git a/monkey/monkey_island/cc/resources/local_run.py b/monkey/monkey_island/cc/resources/local_run.py
index d4ad2b4a9..d743fc835 100644
--- a/monkey/monkey_island/cc/resources/local_run.py
+++ b/monkey/monkey_island/cc/resources/local_run.py
@@ -1,4 +1,5 @@
 import json
+import logging
 import os
 import sys
 from shutil import copyfile
@@ -15,7 +16,6 @@ from monkey_island.cc.services.node import NodeService
 
 __author__ = 'Barak'
 
-import logging
 
 logger = logging.getLogger(__name__)