scripts/sysmon/osinfo_metrics.py

# Copyright 2016 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.

"""System metrics."""

from __future__ import absolute_import

import collections
import platform
import sys

from chromite.lib import metrics


_os_name_metric = metrics.StringMetric(
    'proc/os/name',
    description='OS name on the machine')

_os_version_metric = metrics.StringMetric(
    'proc/os/version',
    description='OS version on the machine')

_os_arch_metric = metrics.StringMetric(
    'proc/os/arch',
    description='OS architecture on this machine')

_python_arch_metric = metrics.StringMetric(
    'proc/python/arch',
    description='python userland architecture on this machine')


def collect_os_info():
  os_info = _get_osinfo()
  _os_name_metric.set(os_info.name)
  _os_version_metric.set(os_info.version)
  _os_arch_metric.set(platform.machine())
  _python_arch_metric.set(_get_python_arch())


class OSInfo(collections.namedtuple('OSInfo', 'name,version')):
  """namedtuple representing OS info (all fields lowercased)."""

  def __new__(cls, name, version):
    return super(OSInfo, cls).__new__(cls, name.lower(), version.lower())


def _get_osinfo():
  """Get OS name and version.

  Returns:
    OSInfo instance
  """
  os_name = platform.system()
  if os_name == 'Linux':
    return _get_linux_osinfo()
  else:
    return OSInfo(name='', version='')


def _get_linux_osinfo():
  # will return something like ('Ubuntu', '14.04', 'trusty')
  try:
    # This is removed in Python 3.8+.
    # pylint: disable=deprecated-method
    os_name, os_version, _ = platform.dist()
  except AttributeError:
    os_name = os_version = ''
  return OSInfo(name=os_name, version=os_version)


def _get_python_arch():
  if sys.maxsize > 2**32:
    return '64'
  else:
    return '32'