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

[DaleCurtis, 2011/04/21 21:51:02]

Alphabetize config/parser.

[ericli, 2011/04/21 23:48:12]

Done.

+config = global_config.global_config
 

[DaleCurtis, 2011/04/21 21:51:02]

Two lines between top level.

[ericli, 2011/04/21 23:48:12]

Done.

 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):
+        if not parser.options.image:
+            return super(SiteAutotest, self).get_fetch_location()
+
+        repos = config.get_config_value('PACKAGES', 'fetch_location', type=list,
+                                        default=[])
+        new_repos = []
+        for repo in repos[::-1]:
+            if repo.endswith('static/archive'):
+                path = parser.options.image.rstrip('/')

[DaleCurtis, 2011/04/21 21:51:02]

No need to compute path or build in loop. Move them out.

[ericli, 2011/04/21 23:48:12]

No, I would say only compute inside loop.

We will only compute the path when certain pattern appears in the repo list, and
we know for sure it will only appear once.

On 2011/04/21 21:51:02, dalec wrote:

+                build = '/'.join(path.split('/')[-2:])
+                repo += '/%s/autotest' % build
+            new_repos.append(repo)
+        return new_repos
+
+
 class _SiteRun(object):
     pass