| 12345678910111213141516171819202122232425262728293031323334 |
- from flask import jsonify
- from flask.views import View
- from pycs.database.Project import Project
- class ListLabelTree(View):
- """
- return a list of labels for a given project
- """
- # pylint: disable=arguments-differ
- methods = ['GET']
- def dispatch_request(self, project_id):
- # find project
- project = Project.get_or_404(project_id)
- labels = [lab.serialize() for lab in project.labels.all()]
- lookup = {}
- for label in labels:
- label["children"] = []
- lookup[label["id"]] = label
- tree = []
- for label in labels:
- parent_id = label["parent_id"]
- if parent_id is None:
- tree.append(label)
- else:
- lookup[parent_id]["children"].append(label)
- return jsonify(tree)
|
