Source code for aiida.cmdline.params.types.node
###########################################################################
# 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 #
###########################################################################
"""Module to define the custom click param type for node
"""
from .identifier import IdentifierParamType
__all__ = ('NodeParamType',)
[docs]
class NodeParamType(IdentifierParamType):
"""The ParamType for identifying Node entities or its subclasses"""
name = 'Node'
@property
def orm_class_loader(self):
"""Return the orm entity loader class, which should be a subclass of OrmEntityLoader. This class is supposed
to be used to load the entity for a given identifier
:return: the orm entity loader class for this ParamType
"""
from aiida.orm.utils.loaders import NodeEntityLoader
return NodeEntityLoader