fune/taskcluster/taskgraph/util/platforms.py
Andrew Halberstadt d024a31639 Bug 1726573 - Run 'pyupgrade --py36-plus' on taskcluster/taskgraph directory, r=taskgraph-reviewers,bhearsum
This patch was generated via:

  $ pyupgrade --py36-plus $(find taskcluster/taskgraph -name "*.py" -type f)
  $ autoflake --remove-all-unused-imports -i $(find taskcluster/taskgraph -name "*.py" -type f)

The same set of commands are being applied to standalone taskgraph as well so
they remain in sync.

Differential Revision: https://phabricator.services.mozilla.com/D123234
2021-08-24 19:35:46 +00:00

58 lines
1.6 KiB
Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import re
from .attributes import keymatch
# platform family is extracted from build platform by taking the alphabetic prefix
# and then translating win -> windows
_platform_re = re.compile(r"^[a-z]*")
_renames = {"win": "windows"}
_archive_formats = {
"linux": ".tar.bz2",
"macosx": ".tar.gz",
"windows": ".zip",
}
_executable_extension = {
"linux": "",
"macosx": "",
"windows": ".exe",
}
_architectures = {
r"linux\b.*": "x86",
r"linux64\b.*": "x86_64",
r"macosx64\b.*": "macos-x86_64-aarch64",
r"win32\b.*": "x86",
r"win64\b(?!-aarch64).*": "x86_64",
r"win64-aarch64\b.*": "aarch64",
}
def platform_family(build_platform):
"""Given a build platform, return the platform family (linux, macosx, etc.)"""
family = _platform_re.match(build_platform).group(0)
return _renames.get(family, family)
def archive_format(build_platform):
"""Given a build platform, return the archive format used on the platform."""
return _archive_formats[platform_family(build_platform)]
def executable_extension(build_platform):
"""Given a build platform, return the executable extension used on the platform."""
return _executable_extension[platform_family(build_platform)]
def architecture(build_platform):
matches = keymatch(_architectures, build_platform)
if len(matches) == 1:
return matches[0]
raise Exception(f"Could not determine architecture of platform `{build_platform}`.")