Enable autotest client package fetching based on updated image.

server/site_autotest.py

Index: server/site_autotest.py
diff --git a/server/site_autotest.py b/server/site_autotest.py
index 1709b280fa61ce0543a2011876913ace2a950873..ead71654e483fb66c659485a4b7086ae42916311 100755
--- a/server/site_autotest.py
+++ b/server/site_autotest.py
@@ -1,5 +1,10 @@
 import os
-from autotest_lib.server import installable_object
+from autotest_lib.client.common_lib import global_config
+from autotest_lib.server import autoserv_parser, installable_object
+
+
+parser = autoserv_parser.autoserv_parser
+c = global_config.global_config

[DaleCurtis, 2011/04/21 20:35:14]

Seems strange to have a global variable with a single letter name. Up to you,
but I'd use config or something else.

 
 class SiteAutotest(installable_object.InstallableObject):
 
@@ -11,5 +16,18 @@ class SiteAutotest(installable_object.InstallableObject):
         self.got = True
 
 
+    def get_fetch_location(self):
+        repos = c.get_config_value('PACKAGES', 'fetch_location', type=list,
+                                   default=[])
+        new_repos = []
+        for repo in repos[::-1]:
+            path = parser.options.image.rstrip('/')
+            if repo.endswith('static/archive') and parser.options.image:
+                build = '/'.join(path.split('/')[-2:])
+                repo += '/%s/autotest' % build
+            new_repos.append(repo)
+        return new_repos
+
+
 class _SiteRun(object):
     pass