agent: Change type hint for FileEncryption's `__init__()`'s `filepath`

This commit is contained in:
Shreya 2021-07-06 19:04:42 +05:30
parent 8afd69634c
commit e91d7a6282
1 changed files with 3 additions and 1 deletions

View File

@ -1,3 +1,5 @@
from pathlib import Path
from common.common_consts.telem_categories import TelemCategoryEnum from common.common_consts.telem_categories import TelemCategoryEnum
from infection_monkey.telemetry.base_telem import BaseTelem from infection_monkey.telemetry.base_telem import BaseTelem
from infection_monkey.telemetry.batchable_telem_mixin import BatchableTelemMixin from infection_monkey.telemetry.batchable_telem_mixin import BatchableTelemMixin
@ -5,7 +7,7 @@ from infection_monkey.telemetry.i_batchable_telem import IBatchableTelem
class FileEncryptionTelem(BatchableTelemMixin, IBatchableTelem, BaseTelem): class FileEncryptionTelem(BatchableTelemMixin, IBatchableTelem, BaseTelem):
def __init__(self, filepath: str, success: bool, error: str): def __init__(self, filepath: Path, success: bool, error: str):
""" """
File Encryption telemetry constructor File Encryption telemetry constructor
:param filepath: The path to the file that monkey attempted to encrypt :param filepath: The path to the file that monkey attempted to encrypt