make debian installer fetch more generic

1 files changed, 15 insertions, 1 deletions

diff --git a/roles/installer/debian/fetch/filter_plugins/main.py b/roles/installer/debian/fetch/filter_plugins/main.py
index 298e7efd..947db2eb 100644
--- a/roles/installer/debian/fetch/filter_plugins/main.py
+++ b/roles/installer/debian/fetch/filter_plugins/main.py
@@ -4,6 +4,19 @@ __metaclass__ = type
 from ansible import errors
 
 
+def di_dists_path(data):
+    try:
+        if data[0] != 'ubuntu':
+            return data[1]
+
+        if data[1] in ['xenial']:
+            return data[1]+'-updates'
+
+        return data[1]
+    except Exception as e:
+        raise errors.AnsibleFilterError("di_dists_path(): %s" % str(e))
+
+
 def di_images_path(data):
     try:
         if data[0] != 'ubuntu':
@@ -14,12 +27,13 @@ def di_images_path(data):
 
         return 'legacy-images'
     except Exception as e:
-        raise errors.AnsibleFilterError("mountpoint_exists(): %s" % str(e))
+        raise errors.AnsibleFilterError("di_images_path(): %s" % str(e))
 
 
 class FilterModule(object):
 
     filter_map = {
+        'di_dists_path': di_dists_path,
         'di_images_path': di_images_path,
     }