blob: 01113153e45d41d186e0448e8ecc2fcca2ed150e [file] [log] [blame]
# Copyright (c) 2011 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.
"""Implementation of the 'workon' chromite command."""
import chromite.lib.cros_build_lib as cros_lib
from import subcmd
class WorkonCmd(subcmd.WrappedChrootCmd):
"""Run cros_workon."""
def __init__(self):
"""WorkonCmd constructor."""
# Just call the WrappedChrootCmd superclass, which does most of the work.
super(WorkonCmd, self).__init__(
['./cros_workon', '--host'],
def Run(self, raw_argv, *args, **kwargs):
"""Run the command.
We do just a slight optimization to help users with a common typo.
raw_argv: Command line arguments, including this command's name, but not
the chromite command name or chromite options.
args: The rest of the positional arguments. See _DoWrappedChrootCommand.
kwargs: The keyword arguments. See _DoWrappedChrootCommand.
# Slight optimization, just since I do this all the time...
if len(raw_argv) >= 2:
if raw_argv[1] in ('start', 'stop', 'list', 'list-all', 'iterate'):
cros_lib.Warning('OOPS, looks like you forgot a board name. Pick one.')
raw_argv = raw_argv[:1] + [''] + raw_argv[1:]
super(WorkonCmd, self).Run(raw_argv, *args, **kwargs)