Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor AmiBackend.describe_images: make logic of missing image simpler

Browse Source
This commit is contained in:
Hugo Lopes Tavares 2014-01-28 18:53:12 -05:00
parent 90aa87d53e
commit 4e88d79da2
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
moto/ec2/models.py
View File

@ -219,14 +219,14 @@ class AmiBackend(object):
self.amis[ami_id] = ami
return ami
def describe_images(self, ami_ids=None):
if ami_ids:
images = [image for image in self.amis.values() if image.id in ami_ids]
for missing_ami in set(ami_ids) - set(ami.id for ami in images):
raise InvalidIdError(missing_ami)
else:
images = self.amis.values()
return images
def describe_images(self, ami_ids=()):
images = []
for ami_id in ami_ids:
if ami_id in self.amis:
images.append(self.amis[ami_id])
else:
raise InvalidIdError(ami_id)
return images or self.amis.values()
def deregister_image(self, ami_id):
if ami_id in self.amis: