| # Copyright 2024 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import os |
| |
| from pylib.local.device import local_device_ethernet_environment |
| |
| SWARMING_SERVER = 'chromeos-swarming.appspot.com' |
| |
| |
| class SkylabEnvironment( |
| local_device_ethernet_environment.LocalDeviceEthernetEnvironment): |
| """ |
| A subclass of LocalDeviceEthernetEnvironment for Skylab devices. |
| """ |
| |
| def GetDeviceHostname(self): |
| """Return the hostname based on the bot id. |
| |
| Strips the first component of the bot id, e.g. 'cros-clank1' -> 'clank1'. |
| |
| Gets the bot id from the SWARMING_BOT_ID envvar, see |
| https://chromium.googlesource.com/infra/luci/luci-py/+/HEAD/appengine/swarming/doc/Magic-Values.md#bot-environment-variables. |
| """ |
| bot_id = os.environ.get('SWARMING_BOT_ID') |
| if not bot_id: |
| raise ValueError( |
| "device_arg is 'swarming' but SWARMING_BOT_ID is not set") |
| |
| return bot_id[bot_id.index("-") + 1:] |
