7de9323fc5
Also made several variable renames in the tests. Change-Id: I675826f2510b6690f6008fcbce9e08ad610a4fa1 Closes-Bug: #1614551
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
|
|
import flask
|
|
import flask_restful
|
|
from flask_restful import fields
|
|
|
|
from tuning_box import db
|
|
from tuning_box import library
|
|
from tuning_box.library import resource_definitions
|
|
|
|
component_fields = {
|
|
'id': fields.Integer,
|
|
'name': fields.String,
|
|
'resource_definitions': fields.List(fields.Nested(
|
|
resource_definitions.resource_definition_fields
|
|
))
|
|
}
|
|
|
|
|
|
class ComponentsCollection(flask_restful.Resource):
|
|
method_decorators = [flask_restful.marshal_with(component_fields)]
|
|
|
|
def get(self):
|
|
return db.Component.query.all()
|
|
|
|
@db.with_transaction
|
|
def post(self):
|
|
component = db.Component(name=flask.request.json['name'])
|
|
component.resource_definitions = []
|
|
for res_def_data in flask.request.json.get('resource_definitions'):
|
|
res_def = db.ResourceDefinition(
|
|
name=res_def_data['name'], content=res_def_data.get('content'))
|
|
component.resource_definitions.append(res_def)
|
|
db.db.session.add(component)
|
|
return component, 201
|
|
|
|
|
|
class Component(flask_restful.Resource):
|
|
method_decorators = [flask_restful.marshal_with(component_fields)]
|
|
|
|
def get(self, component_id):
|
|
return db.Component.query.get_or_404(component_id)
|
|
|
|
@db.with_transaction
|
|
def _perform_update(self, component_id):
|
|
component = db.Component.query.get_or_404(component_id)
|
|
update_by = flask.request.json
|
|
component.name = update_by.get('name', component.name)
|
|
res_definitions = update_by.get('resource_definitions')
|
|
if res_definitions is not None:
|
|
ids = [data['id'] for data in res_definitions]
|
|
resources = library.load_objects(db.ResourceDefinition, ids)
|
|
component.resource_definitions = resources
|
|
|
|
def put(self, component_id):
|
|
return self.patch(component_id)
|
|
|
|
def patch(self, component_id):
|
|
self._perform_update(component_id)
|
|
return None, 204
|
|
|
|
@db.with_transaction
|
|
def delete(self, component_id):
|
|
component = db.Component.query.get_or_404(component_id)
|
|
db.db.session.delete(component)
|
|
return None, 204
|