Use const

Ostorlab · Nov 20, 2023 · 017b16f · 017b16f
1 parent fba90f5
commit 017b16f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/agent/metasploit_agent.py b/agent/metasploit_agent.py
@@ -26,6 +26,7 @@
 logger = logging.getLogger(__name__)
 
 MODULE_TIMEOUT = 300
+VULNERABLE_STATUSES = ["vulnerable", "appears"]
 METASPLOIT_AGENT_KEY = b"agent_metasploit_asset"
 
 
@@ -108,10 +109,10 @@ def process(self, message: m.Message) -> None:
 
                 results = self._get_job_results(client, job_uuid)
 
-                if isinstance(results, dict) and results.get("code") in [
-                    "vulnerable",
-                    "appears",
-                ]:
+                if (
+                    isinstance(results, dict)
+                    and results.get("code") in VULNERABLE_STATUSES
+                ):
                     technical_detail = f"Using `{module_instance.moduletype}` module `{module_instance.modulename}`\n"
                     technical_detail += f"Target: {rhost}:{rport}\n"
                     technical_detail += f'Message: \n```{results["message"]}```'