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.client.common_lib.cros.graphite import autotest_stats
8from autotest_lib.scheduler import drones, scheduler_config
9
10HOSTS_JOB_SUBDIR = 'hosts/'
11PARSE_LOG = '.parse.log'
12ENABLE_ARCHIVING =  global_config.global_config.get_config_value(
13        scheduler_config.CONFIG_SECTION, 'enable_archiving', type=bool)
14
15
16class SiteDroneManager(object):
17
18
19    _timer = autotest_stats.Timer('drone_manager')
20
21
22    def copy_to_results_repository(self, process, source_path,
23                                   destination_path=None):
24        """
25        Copy results from the given process at source_path to destination_path
26        in the results repository.
27
28        This site subclassed version will only copy the results back for Special
29        Agent Tasks (Cleanup, Verify, Repair) that reside in the hosts/
30        subdirectory of results if the copy_task_results_back flag has been set
31        to True inside global_config.ini
32
33        It will also only copy .parse.log files back to the scheduler if the
34        copy_parse_log_back flag in global_config.ini has been set to True.
35        """
36        if not ENABLE_ARCHIVING:
37            return
38        copy_task_results_back = global_config.global_config.get_config_value(
39                scheduler_config.CONFIG_SECTION, 'copy_task_results_back',
40                type=bool)
41        copy_parse_log_back = global_config.global_config.get_config_value(
42                scheduler_config.CONFIG_SECTION, 'copy_parse_log_back',
43                type=bool)
44        special_task = source_path.startswith(HOSTS_JOB_SUBDIR)
45        parse_log = source_path.endswith(PARSE_LOG)
46        if (copy_task_results_back or not special_task) and (
47                copy_parse_log_back or not parse_log):
48            super(SiteDroneManager, self).copy_to_results_repository(process,
49                    source_path, destination_path)
50
51
52    def kill_process(self, process):
53        """
54        Kill the given process.
55        """
56        logging.info('killing %s', process)
57        drone = self._get_drone_for_process(process)
58        drone.queue_kill_process(process)
59
60
61    def _add_drone(self, hostname):
62        """
63        Forked from drone_manager.py
64
65        Catches AutoservRunError if the drone fails initialization and does not
66        add it to the list of usable drones.
67
68        @param hostname: Hostname of the drone we are trying to add.
69        """
70        logging.info('Adding drone %s' % hostname)
71        drone = drones.get_drone(hostname)
72        if drone:
73            try:
74                drone.call('initialize', self.absolute_path(''))
75            except error.AutoservRunError as e:
76                logging.error('Failed to initialize drone %s with error: %s',
77                              hostname, e)
78                return
79            self._drones[drone.hostname] = drone
80
81
82    @_timer.decorate
83    def refresh(self):
84       super(SiteDroneManager, self).refresh()
85
86
87    @_timer.decorate
88    def execute_actions(self):
89       super(SiteDroneManager, self).execute_actions()
90