Enable autotest client package fetching based on updated image.

server/site_autotest.py

 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):
 
     def get(self, location = None):
         if not location:
             location = os.path.join(self.serverdir, '../client')
             location = os.path.abspath(location)
         installable_object.InstallableObject.get(self, location)
         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