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

Island: Implement delete_files_by_pattern() in LocalStorageFileRepository

This commit is contained in:
Shreya Malviya 2022-09-28 15:39:14 +05:30
parent d9b55a5c21
commit ce0affb1ed
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/monkey_island/cc/repository/local_storage_file_repository.py
View File

@ -1,3 +1,4 @@
import glob
import logging import logging
import shutil import shutil
from pathlib import Path from pathlib import Path
@ -55,6 +56,10 @@ class LocalStorageFileRepository(IFileRepository):
f'Error retrieving file "{unsafe_file_name}" from the repository: {err}' f'Error retrieving file "{unsafe_file_name}" from the repository: {err}'
) )
def delete_files_by_pattern(self, file_name_pattern: str):
for file_name in glob.iglob(f"{self._storage_directory}/{file_name_pattern}"):
self.delete_file(file_name)
def delete_file(self, unsafe_file_name: str): def delete_file(self, unsafe_file_name: str):
try: try:
safe_file_path = self._get_safe_file_path(unsafe_file_name) safe_file_path = self._get_safe_file_path(unsafe_file_name)