diff --git a/BackupRotator.py b/BackupRotator.py index ea3265a..c245b7c 100755 --- a/BackupRotator.py +++ b/BackupRotator.py @@ -24,12 +24,19 @@ import yaml class BackupRotator: + __DEFAULT_VALID_EXTENSIONS = [ + "yaml", + "yml" + ] + def __init__(self): self.__dry_run = False self.__configs = [] self.__config_paths = [] self.__calculated_actions = [] + + self.__valid_extensions = self.__DEFAULT_VALID_EXTENSIONS def run(self, configs, dry_run: bool = False): @@ -77,8 +84,9 @@ class BackupRotator: # Use each config path for path in paths: - # If this is a single path + # If this is a single file if os.path.isfile(path): + self._consume_config(path) # If this is a directory @@ -86,7 +94,11 @@ class BackupRotator: # Iterate over each file inside for file_name in os.listdir(path): - self._consume_config(os.path.join(path, file_name)) + + one_file = os.path.join(path, file_name) + + if os.path.isfile(one_file) and self._check_file_extension(file_path=one_file, extensions=None): + self._consume_config(one_file) def _consume_config(self, path: str): @@ -386,3 +398,20 @@ class BackupRotator: self.log("No value found for \"minimum-items\"; Will not enforce minimum item constraint.") return minimum_items + + def _check_file_extension(self, file_path, extensions: list=None): + + if extensions is None: + extensions = self.__valid_extensions + + file_name, file_extension = os.path.splitext(file_path) + if len(file_extension) > 0 and file_extension[0] == ".": + file_extension = file_extension[1:] + file_extension = file_extension.lower() + + for valid_extension in extensions: + #print(file_name, "---", file_extension, "---", valid_extension) + if file_extension == valid_extension: + return True + + return False