p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests 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 import os
from typing import Iterator, List from typing import List
VALID_FILE_EXTENSIONS_FOR_ENCRYPTION = { VALID_FILE_EXTENSIONS_FOR_ENCRYPTION = {
".3ds", ".3ds",
@ -90,5 +90,7 @@ def get_files_to_encrypt(dir_path: str) -> List[str]:
return files_to_encrypt return files_to_encrypt
def get_all_files_in_directory(dir_path: str) -> Iterator: def get_all_files_in_directory(dir_path: str) -> List:
return filter(os.path.isfile, [os.path.join(dir_path, item) for item in os.listdir(dir_path)]) return list(
filter(os.path.isfile, [os.path.join(dir_path, item) for item in os.listdir(dir_path)])
)