ec2-api/ec2api/api/tag.py
Andrey Pavlov f8af53b167 create tags' tests
Change-Id: Ic25c25e13f2c89e4179b5d926d3c8908fa4197b1
2015-03-05 09:35:39 +03:00

116 lines
3.4 KiB
Python

# Copyright 2014
# The Cloudscaling Group, Inc.
#
# 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.
from ec2api.api import common
from ec2api.api import ec2utils
from ec2api.db import api as db_api
from ec2api import exception
from ec2api.i18n import _
"""Tag related API implementation
"""
Validator = common.Validator
RESOURCE_TYPES = {
'dopt': 'dhcp-options',
'ami': 'image',
'aki': 'image',
'ari': 'image',
'i': 'instance',
'igw': 'internet-gateway',
'eni': 'network-interface',
'rtb': 'route-table',
'snap': 'snapshot',
'subnet': 'subnet',
'sg': 'security-group',
'vol': 'volume',
'vpc': 'vpc',
}
def create_tags(context, resource_id, tag):
reason = None
for tag_pair in tag:
if not tag_pair.get('key'):
reason = _('Not empty key must be present')
elif len(tag_pair['key']) > 127:
reason = _('Tag key exceeds the maximum length of 127 characters')
elif tag_pair['key'].startswith('aws:'):
reason = _("Tag keys starting with 'aws:' are reserved for "
"internal use")
elif 'value' not in tag_pair:
reason = _('Value must be present')
elif len(tag_pair['value']) > 255:
reason = _('Tag value exceeds the maximum length of 255 '
'characters')
if reason:
raise exception.InvalidParameterValue(
parameter='Tag', value=str(tag_pair), reason=reason)
for item_id in resource_id:
kind = ec2utils.get_ec2_id_kind(item_id)
if kind not in RESOURCE_TYPES:
raise exception.InvalidID(id=item_id)
# NOTE(ft): check items exist (excluding images because AWS allows to
# create a tag with any image id)
if kind not in ('ami', 'ari', 'aki'):
ec2utils.get_db_item(context, item_id)
tags = [dict(item_id=item_id,
key=tag_pair['key'],
value=tag_pair['value'])
for item_id in resource_id
for tag_pair in tag]
db_api.add_tags(context, tags)
return True
def delete_tags(context, resource_id, tag=None):
db_api.delete_tags(context, resource_id, tag)
return True
class TagDescriber(common.NonOpenstackItemsDescriber):
FILTER_MAP = {'key': 'key',
'resource-id': 'resourceId',
'resource-type': 'resourceType',
'value': 'value'}
def get_db_items(self):
return db_api.get_tags(self.context)
def format(self, item):
return _format_tag(item)
def describe_tags(context, filter=None, max_results=None, next_token=None):
formatted_tags = TagDescriber().describe(context, filter=filter)
return {'tagSet': formatted_tags}
def _format_tag(tag):
kind = ec2utils.get_ec2_id_kind(tag['item_id'])
return {
'resourceType': RESOURCE_TYPES.get(kind, kind),
'resourceId': tag['item_id'],
'key': tag['key'],
'value': tag['value'],
}