diff --git a/refit/src/modules/librefit.py b/refit/src/modules/librefit.py new file mode 100644 index 0000000..382009f --- /dev/null +++ b/refit/src/modules/librefit.py @@ -0,0 +1,35 @@ +from .refit_logger import logger + + +def get_standard_name_number(current_number, number_str_length): + """returns a number string filled to the length of the input number""" + current_number = str(current_number) + standard_name_number = str.zfill(current_number, number_str_length) + # logger.debug(f"standard_name_number={standard_name_number}") + return standard_name_number + + +def get_standard_folder_name(name): + """returns the default name if name=None else, it returns the first + given name to the argument --name as astring""" + logger.debug("get_standard_folder_name() call") + if name is None: + logger.warning("No name assigned, continue with default.") + standard_folder_name = "directory" + else: + logger.debug(f"{name}, Type: {type(name)}, Length: {len(name)}") + + # Checks if multiple names are passed to the --name argument + # if so, only the first one gets used. + if len(name) > 1: + _NAMES_WARNING = f"{len(name)} names given, only one required.\nContinuing with '{name[0]}' as name." + logger.warning(_NAMES_WARNING) + print(_NAMES_WARNING) + standard_folder_name = name[0] + + logger.debug(f"{standard_folder_name}, Type: {type(standard_folder_name)}") + else: + standard_folder_name = name[0] + logger.debug("Continuing with expected input") + + return standard_folder_name