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.
4import logging
5
6from autotest_lib.client.common_lib import global_config, error
7from autotest_lib.scheduler import drones, scheduler_config
8
9HOSTS_JOB_SUBDIR = 'hosts/'
10PARSE_LOG = '.parse.log'
11ENABLE_ARCHIVING =  global_config.global_config.get_config_value(
12        scheduler_config.CONFIG_SECTION, 'enable_archiving', type=bool)
13
14
15class 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