Home | History | Annotate | Download | only in scheduler 
      1 # Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
      2 # Use of this source code is governed by a BSD-style license that can be
      3 # found in the LICENSE file.
      4 import logging
      5 
      6 from autotest_lib.client.common_lib import global_config, error
      7 from autotest_lib.scheduler import drones, scheduler_config
      8 
      9 HOSTS_JOB_SUBDIR = 'hosts/'
     10 PARSE_LOG = '.parse.log'
     11 ENABLE_ARCHIVING =  global_config.global_config.get_config_value(
     12         scheduler_config.CONFIG_SECTION, 'enable_archiving', type=bool)
     13 
     14 
     15 class SiteDroneManager(object):
     16     """
     17     Site specific DroneManager extensions.
     18 
     19     When importing this class gains BaseDroneManager as a base class.
     20     """
     21     def copy_to_results_repository(self, process, source_path,
     22                                    destination_path=None):
     23         """
     24         Copy results from the given process at source_path to destination_path
     25         in the results repository.
     26 
     27         This site subclassed version will only copy the results back for Special
     28         Agent Tasks (Cleanup, Verify, Repair) that reside in the hosts/
     29         subdirectory of results if the copy_task_results_back flag has been set
     30         to True inside global_config.ini
     31 
     32         It will also only copy .parse.log files back to the scheduler if the
     33         copy_parse_log_back flag in global_config.ini has been set to True.
     34         """
     35         if not ENABLE_ARCHIVING:
     36             return
     37         copy_task_results_back = global_config.global_config.get_config_value(
     38                 scheduler_config.CONFIG_SECTION, 'copy_task_results_back',
     39                 type=bool)
     40         copy_parse_log_back = global_config.global_config.get_config_value(
     41                 scheduler_config.CONFIG_SECTION, 'copy_parse_log_back',
     42                 type=bool)
     43         special_task = source_path.startswith(HOSTS_JOB_SUBDIR)
     44         parse_log = source_path.endswith(PARSE_LOG)
     45         if (copy_task_results_back or not special_task) and (
     46                 copy_parse_log_back or not parse_log):
     47             super(SiteDroneManager, self).copy_to_results_repository(process,
     48                     source_path, destination_path)
     49 
     50 
     51     def kill_process(self, process):
     52         """
     53         Kill the given process.
     54         """
     55         logging.info('killing %s', process)
     56         drone = self._get_drone_for_process(process)
     57         drone.queue_kill_process(process)
     58 
     59 
     60     def _add_drone(self, hostname):
     61         """
     62         Forked from drone_manager.py
     63 
     64         Catches AutoservRunError if the drone fails initialization and does not
     65         add it to the list of usable drones.
     66 
     67         @param hostname: Hostname of the drone we are trying to add.
     68         """
     69         logging.info('Adding drone %s' % hostname)
     70         drone = drones.get_drone(hostname)
     71         if drone:
     72             try:
     73                 drone.call('initialize', self.absolute_path(''))
     74             except error.AutoservRunError as e:
     75                 logging.error('Failed to initialize drone %s with error: %s',
     76                               hostname, e)
     77                 return
     78             self._drones[drone.hostname] = drone
     79