blob: fad9e4aaccc965d1c67041aea5385d48f31aab22 [file] [log] [blame] [edit]
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
class _SiteAbstractDrone(object):
"""
This is a site subclass of _BaseAbstractDrone in drones.py. Any methods
here automatically overload _BaseAbstractDrone and are used to create
_AbstractDrone for consumers.
"""
def __init__(self):
"""
Add a new private variable _processes_to_kill to _AbstractDrone
"""
super(_SiteAbstractDrone, self).__init__()
self._processes_to_kill = []
def queue_kill_process(self, process):
"""Queue a process to kill/abort.
@param process: Process to kill/abort.
"""
self._processes_to_kill.append(process)
def clear_processes_to_kill(self):
"""Reset the list of processes to kill for this tick."""
self._processes_to_kill = []
def execute_queued_calls(self):
"""Overloads execute_queued_calls().
If there are any processes queued to kill, kill them then process the
remaining queued up calls.
"""
if self._processes_to_kill:
self.queue_call('kill_processes', self._processes_to_kill)
self.clear_processes_to_kill()
super(_SiteAbstractDrone, self).execute_queued_calls()