def get_installed()

in deepracer_systems_pkg/deepracer_systems_pkg/model_loader_module/model_loader_node.py [0:0]

• 16 lines of code
• 7 McCabe index (conditional complexity)

    def get_installed(self):
        """Return the list of all the models in the /opt/aws/deepracer/artifacts folder.

        Returns:
            list: List of directories which contain a checksum file.
        """
        list_of_directories = list()

        try:
            for dir_name in os.listdir(model_loader_config.MODEL_INSTALL_ROOT_DIRECTORY):

                model_directory = os.path.join(model_loader_config.MODEL_INSTALL_ROOT_DIRECTORY, dir_name)
                if not os.path.isdir(model_directory):
                    continue

                if not os.path.isfile(os.path.join(model_directory, model_loader_config.MODEL_CHECKSUM_FILE)):
                    continue

                if model_loader_config.ENABLE_GOLDEN_MODEL \
                   and (dir_name == model_loader_config.GOLDEN_MODEL_TARGET_NAME):
                    continue

                list_of_directories.append(model_directory)

        except Exception as ex:
            self.get_logger().error(f"Failed to list {model_loader_config.MODEL_INSTALL_ROOT_DIRECTORY}: {ex}")

        return list_of_directories