List images return images on compute and controller

Fix #506
This commit is contained in:
Julien Duponchelle committed 2016-06-08 14:14:03 +02:00
1 parent 08e482004f
commit add546158f
6 files changed
+105 -6

No files matched your search

+21
View File
@@ -27,6 +27,27 @@ import logging
log = logging.getLogger(__name__)
def scan_for_images(type):
"""
Scan directories for available image for a type
:param type: emulator type (dynamips, qemu, iou)
"""
files = set()
paths = []
for directory in images_directories(type):
for root, _, filenames in os.walk(directory):
for file in filenames:
path = os.path.join(root, file)
if file not in files:
if (file.endswith(".image") and type == "dynamips") \
or (file.endswith(".bin") and type == "iou") \
or (not file.endswith(".bin") and not file.endswith(".image") and type == "qemu"):
files.add(file)
paths.append(path)
return paths
def images_directories(type):
"""
Return all directory where we will look for images