def set_default_chart_save_location(user_set: bool) -> None:
"""
Set default_chart_save_location, based on user input, or default.
Set and save default_chart_save_location, taking user input, or defaulting
to original location. Creates chart save folder.
:param user_set: Path or str
:return: None
"""
# Initialise default location.
new_default_save_location = APP_DEFAULT_CHART_SAVE_DIR
original_location = definitions.DEFAULT_CHART_SAVE_DIR
if user_set:
new_default_save_location = user_set_chart_save_folder()
new_chart_save_folder_path = Path(new_default_save_location,
CHART_SAVE_DIR_NAME)
# Initialise and save chart save location.
definitions.DEFAULT_CHART_SAVE_DIR = new_chart_save_folder_path
save_new_default_chart_save_location_setting(new_chart_save_folder_path)
create_chart_save_folder(new_chart_save_folder_path, original_location)
def test_user_set_set_chart_folder_blank_input(self, monkeypatch,
user_folder_selection, mock_APP_DEFAULT_CHART_SAVE_FOLDER,
expected_return):
monkeypatch.setattr(settings_functions_UI, 'select_folder_dialogue', lambda **args: user_folder_selection)
monkeypatch.setattr(settings_functions_UI, 'APP_DEFAULT_CHART_SAVE_DIR', mock_APP_DEFAULT_CHART_SAVE_FOLDER)
assert user_set_chart_save_folder() == expected_return
def test_user_set_set_chart_folder_user_selects_location(
self, mocked_select_folder_dialogue, mocked_print):
mocked_select_folder_dialogue.return_value = self.test_user_location
assert user_set_chart_save_folder() == self.test_user_location
mocked_select_folder_dialogue.assert_called_once_with(
title_str=self.test_dialogue_message, start_dir='..')
mocked_print.assert_called_once_with(self.user_feedback)
def test_user_set_set_chart_folder_blank_input(
self, mocked_select_folder_dialogue, mocked_print):
mocked_select_folder_dialogue.return_value = ''
assert user_set_chart_save_folder(
) == self.mocked_APP_DEFAULT_CHART_SAVE_FOLDER
mocked_select_folder_dialogue.assert_called_once_with(
title_str=self.test_dialogue_message, start_dir='..')
mocked_print.assert_not_called()
def set_default_chart_save_location(user_set):
# Initialise default location.
new_default_save_location = APP_DEFAULT_CHART_SAVE_FOLDER
original_location = definitions.DEFAULT_CHART_SAVE_FOLDER
if user_set:
new_default_save_location = user_set_chart_save_folder()
new_chart_save_folder_path = Path(new_default_save_location, CHART_SAVE_FOLDER_NAME)
# Initialise and save chart save location.
definitions.DEFAULT_CHART_SAVE_FOLDER = str(new_chart_save_folder_path)
save_new_default_chart_save_location_setting(new_chart_save_folder_path)
create_chart_save_folder(new_chart_save_folder_path, original_location)