p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

agent: ransomware: Return list instead of filter object when getting files in directory

This commit is contained in:
Shreya 2021-06-22 12:47:38 +05:30
parent 5b64ea5151
commit 3192355f5a
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/infection_monkey/ransomware/utils.py
View File

@ -1,5 +1,5 @@
import os
from typing import Iterator, List
from typing import List
VALID_FILE_EXTENSIONS_FOR_ENCRYPTION = {
".3ds",
@ -90,5 +90,7 @@ def get_files_to_encrypt(dir_path: str) -> List[str]:
return files_to_encrypt
def get_all_files_in_directory(dir_path: str) -> Iterator:
return filter(os.path.isfile, [os.path.join(dir_path, item) for item in os.listdir(dir_path)])
def get_all_files_in_directory(dir_path: str) -> List:
return list(
filter(os.path.isfile, [os.path.join(dir_path, item) for item in os.listdir(dir_path)])
)