# -*- coding: utf-8 -*-
###########################################################################
# 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 __future__ import division
from __future__ import print_function
from __future__ import absolute_import
from .identifier import IdentifierParamType
[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