archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Utility to get a node type from a project document

Browse Source
This commit is contained in:
Francesco Siddi 2016-05-18 17:40:45 +02:00
parent cb87bea47f
commit e9f8090497
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pillar/application/utils/__init__.py
View File

@ -11,6 +11,7 @@ from flask import current_app
__all__ = ('remove_private_keys', 'PillarJSONEncoder') __all__ = ('remove_private_keys', 'PillarJSONEncoder')
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
def remove_private_keys(document): def remove_private_keys(document):
"""Removes any key that starts with an underscore, returns result as new """Removes any key that starts with an underscore, returns result as new
dictionary. dictionary.
@ -67,3 +68,11 @@ def skip_when_testing(func):
return func(*args, **kwargs) return func(*args, **kwargs)
return wrapper return wrapper
def project_get_node_type(project_document, node_type_node_name):
"""Return a node_type subdocument for a project. If none is found, return
None.
"""
return next((node_type for node_type in project_document['node_types']
if node_type['name'] == node_type_node_name), None)