From 1b7e36a966ce1a0964e93eefa98c9efcbc4c323d Mon Sep 17 00:00:00 2001
From: Scott Moser <smoser@ubuntu.com>
Date: Thu, 25 Apr 2013 11:58:38 -0400
Subject: re-work maas datasource and headers call backs

Couple of things here:
 * do not re-try on user-data (404 means 'not here')
 * re-generate headers on retry requests

LP: #1172742
---
 cloudinit/sources/DataSourceMAAS.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

(limited to 'cloudinit/sources/DataSourceMAAS.py')

diff --git a/cloudinit/sources/DataSourceMAAS.py b/cloudinit/sources/DataSourceMAAS.py
index de822924..dfe90bc6 100644
--- a/cloudinit/sources/DataSourceMAAS.py
+++ b/cloudinit/sources/DataSourceMAAS.py
@@ -218,14 +218,20 @@ def read_maas_seed_url(seed_url, header_cb=None, timeout=None,
     md = {}
     for name in file_order:
         url = files.get(name)
-        if header_cb:
-            headers = header_cb(url)
+        if not header_cb:
+            def _cb(url):
+                return {}
+            header_cb = _cb
+
+        if name == 'user-data':
+            retries = 0
         else:
-            headers = {}
+            retries = None
+
         try:
             ssl_details = util.fetch_ssl_details(paths)
-            resp = util.read_file_or_url(url,
-                                         headers=headers,
+            resp = util.read_file_or_url(url, retries=retries,
+                                         headers_cb=header_cb,
                                          timeout=timeout,
                                          ssl_details=ssl_details)
             if resp.ok():
-- 
cgit v1.2.3