Added the ability to filter by tag-value and refactored the filters to be generic.

This commit is contained in:
Omer Katz 2014-10-01 15:44:54 +03:00
commit 298cf65569
3 changed files with 55 additions and 8 deletions

View file

@ -70,7 +70,7 @@ from .utils import (
random_volume_id,
random_vpc_id,
random_vpc_peering_connection_id,
is_filter_matching)
generic_filter)
class InstanceState(object):
@ -96,6 +96,9 @@ class TaggedEC2Instance(object):
if filter_name == 'tag-key':
return [tag['key'] for tag in tags]
if filter_name == 'tag-value':
return [tag['value'] for tag in tags]
class NetworkInterface(object):
def __init__(self, subnet, private_ip_address, device_index=0, public_ip_auto_assign=True, group_ids=None):
@ -1156,7 +1159,7 @@ class VPC(TaggedEC2Instance):
filter_value = super(VPC, self).get_filter_value(filter_name)
if not filter_value:
if filter_value is None:
msg = "The filter '{0}' for DescribeVPCs has not been" \
" implemented in Moto yet. Feel free to open an issue at" \
" https://github.com/spulec/moto/issues".format(filter_name)
@ -1195,11 +1198,7 @@ class VPCBackend(object):
else:
vpcs = self.vpcs.values()
if filters:
for (_filter, _filter_value) in filters.items():
vpcs = [ vpc for vpc in vpcs if is_filter_matching(vpc, _filter, _filter_value) ]
return vpcs
return generic_filter(filters, vpcs)
def delete_vpc(self, vpc_id):
# Delete route table if only main route table remains.

View file

@ -286,11 +286,19 @@ def is_filter_matching(obj, filter, filter_value):
try:
value = set(value)
return value.issubset(filter_value) or value.issuperset(filter_value)
return (value and value.issubset(filter_value)) or value.issuperset(filter_value)
except TypeError:
return value in filter_value
def generic_filter(filters, objects):
if filters:
for (_filter, _filter_value) in filters.items():
objects = [obj for obj in objects if is_filter_matching(obj, _filter, _filter_value)]
return objects
# not really random ( http://xkcd.com/221/ )
def random_key_pair():
return {