Source code for aiida.cmdline.utils.repository

###########################################################################
# Copyright (c), The AiiDA team. All rights reserved.                     #
# This file is part of the AiiDA code.                                    #
#                                                                         #
# The code is hosted on GitHub at https://github.com/aiidateam/aiida-core #
# For further information on the license, see the LICENSE.txt file        #
# For further information please visit http://www.aiida.net               #
###########################################################################
"""Utility functions for command line commands operating on the repository."""

from aiida.cmdline.utils import echo




def list_repository_contents(node, path, color):
    """Print the contents of the directory `path` in the repository of the given node to stdout.

    :param node: the node
    :param path: directory path
    :raises FileNotFoundError: if the `path` does not exist in the repository of the given node
    """
    from aiida.repository import FileType

    for entry in node.base.repository.list_objects(path):
        bold = bool(entry.file_type == FileType.DIRECTORY)
        echo.echo(
            entry.name, bold=bold, fg=echo.COLORS['report'] if color and entry.file_type == FileType.DIRECTORY else None
        )