From 37889d0b87df8e7c9be2f6273adad7ed18114b11 Mon Sep 17 00:00:00 2001 From: Shreya Date: Tue, 15 Jun 2021 19:00:12 +0530 Subject: [PATCH] island: Extract code to `_get_null_value_for_win32()` in file_utils.py --- monkey/monkey_island/cc/server_utils/file_utils.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/monkey/monkey_island/cc/server_utils/file_utils.py b/monkey/monkey_island/cc/server_utils/file_utils.py index 4b9d4c708..0a9223d60 100644 --- a/monkey/monkey_island/cc/server_utils/file_utils.py +++ b/monkey/monkey_island/cc/server_utils/file_utils.py @@ -96,9 +96,7 @@ def _get_file_descriptor_for_new_secure_file_windows(path: str): security_attributes, file_creation, file_attributes, - win32job.CreateJobObject( - None, "" - ), # https://stackoverflow.com/questions/46800142/in-python-with-pywin32-win32job-the-createjobobject-function-how-do-i-pass-nu # noqa: E501 + _get_null_value_for_win32(), ) return fd @@ -106,3 +104,8 @@ def _get_file_descriptor_for_new_secure_file_windows(path: str): except Exception as ex: LOG.error(f'Could not create a file at "{path}": {str(ex)}') raise ex + + +def _get_null_value_for_win32() -> None: + # https://stackoverflow.com/questions/46800142/in-python-with-pywin32-win32job-the-createjobobject-function-how-do-i-pass-nu # noqa: E501 + return win32job.CreateJobObject(None, "")