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

Added generator for node types of all projects.

Browse Source
This commit is contained in:
Sybren A. Stüvel 2016-05-23 16:50:43 +02:00
parent cf3de95ce1
commit f03f00163a
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pillar/manage.py
View File

@ -68,6 +68,20 @@ def runserver_profile(pfile='profile.stats'):
cProfile.run('runserver(use_reloader=False)', pfile) cProfile.run('runserver(use_reloader=False)', pfile)
def each_project_node_type(node_type_name=None):
"""Generator, yields (project, node_type) tuples for all projects and node types.
When a node type name is given, only yields those node types.
"""
projects_coll = app.data.driver.db['projects']
for project in projects_coll.find():
for node_type in project['node_types']:
if node_type_name is None or node_type['name'] == node_type_name:
yield project, node_type
def post_item(entry, data): def post_item(entry, data):
return post_internal(entry, data) return post_internal(entry, data)