From 97d53dba3cd6761a1a7b4c4bb8ad3f9864ff54c9 Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Wed, 21 Sep 2022 17:58:41 +0000 Subject: [PATCH] Agent: Fix mypy issues in ransomware_options.py --- .../infection_monkey/payload/ransomware/ransomware_options.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/payload/ransomware/ransomware_options.py b/monkey/infection_monkey/payload/ransomware/ransomware_options.py index 505974ae0..e1e2d55e5 100644 --- a/monkey/infection_monkey/payload/ransomware/ransomware_options.py +++ b/monkey/infection_monkey/payload/ransomware/ransomware_options.py @@ -1,4 +1,6 @@ import logging +from pathlib import Path +from typing import Optional from common.utils.file_utils import InvalidPath, expand_path from infection_monkey.utils.environment import is_windows_os @@ -12,7 +14,7 @@ class RansomwareOptions: self.file_extension = options["encryption"]["file_extension"] self.readme_enabled = options["other_behaviors"]["readme"] - self.target_directory = None + self.target_directory: Optional[Path] = None self._set_target_directory(options["encryption"]["directories"]) def _set_target_directory(self, os_target_directories: dict):