| #!/usr/bin/env python |
| # encoding: utf-8 |
| # |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you 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. |
| # |
| |
| """This module contains functions for handling requests in relation to tags. |
| """ |
| |
| from flask import current_app |
| |
| from ec2stack import helpers, errors |
| from ec2stack.providers.cloudstack import requester |
| |
| |
| @helpers.authentication_required |
| def create_tags(): |
| """ |
| Create a tag. |
| |
| @return: Response. |
| """ |
| response = _create_tag_request() |
| return _create_tag_response(response) |
| |
| |
| def _create_tag_request(): |
| """ |
| Request to create a tag. |
| |
| @return: Response. |
| """ |
| |
| key = helpers.get('Tag.1.Key') |
| value = helpers.get('Tag.1.Value') |
| resource_id = helpers.get('ResourceId.1') |
| |
| if resource_id in current_app.config['RESOURCE_TYPE_MAP']: |
| resource_type = current_app.config['RESOURCE_TYPE_MAP'][resource_id] |
| else: |
| errors.invalid_request( |
| str(resource_id) + " not found in configuration") |
| |
| args = { |
| 'command': 'createTags', |
| 'resourceids': resource_id, |
| 'resourcetype': resource_type, |
| 'tags[0].key': key, |
| 'tags[0].value': value |
| } |
| |
| response = requester.make_request_async(args) |
| |
| return response |
| |
| |
| def _create_tag_response(response): |
| """ |
| Generates a response for a create tag request. |
| |
| @return: Response. |
| """ |
| if 'errortext' in response: |
| if 'Unable to find resource by id' in response['errortext']: |
| errors.invalid_resource_id() |
| |
| return { |
| 'template_name_or_list': 'status.xml', |
| 'response_type': 'CreateTagsResponse', |
| 'return': 'true' |
| } |
| |
| |
| @helpers.authentication_required |
| def delete_tags(): |
| """ |
| delete a tag. |
| |
| @return: Response. |
| """ |
| response = _delete_tag_request() |
| return _delete_tag_response(response) |
| |
| |
| def _delete_tag_request(): |
| """ |
| Request to delete a tag. |
| |
| @return: Response. |
| """ |
| key = helpers.get('Tag.1.Key') |
| resource_id = helpers.get('ResourceId.1') |
| |
| if resource_id in current_app.config['RESOURCE_TYPE_MAP']: |
| resource_type = current_app.config['RESOURCE_TYPE_MAP'][resource_id] |
| else: |
| errors.invalid_request( |
| str(resource_id) + " not found in configuration") |
| |
| args = { |
| 'command': 'deleteTags', |
| 'resourceids': resource_id, |
| 'resourcetype': resource_type, |
| 'tags[0].key': key |
| } |
| |
| response = requester.make_request_async(args) |
| |
| return response |
| |
| |
| def _delete_tag_response(response): |
| """ |
| Generates a response for a delete tag request. |
| |
| @return: Response. |
| """ |
| if 'errortext' in response: |
| if 'Unable to find resource by id' in response['errortext']: |
| errors.invalid_resource_id() |
| |
| return { |
| 'template_name_or_list': 'status.xml', |
| 'response_type': 'DeleteTagsResponse', |
| 'return': 'true' |
| } |
| |
| |
| @helpers.authentication_required |
| def describe_tags(): |
| """ |
| Describe all tags. |
| |
| @return: Response. |
| """ |
| args = {'command': 'listTags'} |
| response = requester.make_request(args) |
| |
| return _describe_tags_response( |
| response |
| ) |
| |
| |
| def _describe_tags_response(response): |
| """ |
| Generates a response for a describe tags request. |
| |
| @param response: Response from Cloudstack. |
| @return: Response. |
| """ |
| return { |
| 'template_name_or_list': 'tags.xml', |
| 'response_type': 'DescribeTagsResponse', |
| 'response': response['listtagsresponse'] |
| } |