jumpserver/apps/assets/tasks/gather_asset_hardware_info.py

# -*- coding: utf-8 -*-
#
import json
import re

from celery import shared_task
from django.utils.translation import ugettext as _, gettext_noop

from common.utils import (
    capacity_convert, sum_capacity, get_logger
)
from orgs.utils import org_aware_func
from . import const
from .utils import clean_ansible_task_hosts


logger = get_logger(__file__)
disk_pattern = re.compile(r'^hd|sd|xvd|vd|nv')
__all__ = [
    'update_assets_hardware_info_util', 'update_asset_hardware_info_manual',
    'update_assets_hardware_info_period',  'update_node_assets_hardware_info_manual',
    'update_assets_hardware_info_manual',
]


def set_assets_hardware_info(assets, result, **kwargs):
    """
    Using ops task run result, to update asset info

    @shared_task must be exit, because we using it as a task callback, is must
    be a celery task also
    :param assets:
    :param result:
    :param kwargs: {task_name: ""}
    :return:
    """
    result_raw = result[0]
    assets_updated = []
    success_result = result_raw.get('ok', {})

    for asset in assets:
        hostname = asset.hostname
        info = success_result.get(hostname, {})
        info = info.get('setup', {}).get('ansible_facts', {})
        if not info:
            logger.error(_("Get asset info failed: {}").format(hostname))
            continue
        ___vendor = info.get('ansible_system_vendor', 'Unknown')
        ___model = info.get('ansible_product_name', 'Unknown')
        ___sn = info.get('ansible_product_serial', 'Unknown')

        for ___cpu_model in info.get('ansible_processor', []):
            if ___cpu_model.endswith('GHz') or ___cpu_model.startswith("Intel"):
                break
        else:
            ___cpu_model = 'Unknown'
        ___cpu_model = ___cpu_model[:48]
        ___cpu_count = info.get('ansible_processor_count', 0)
        ___cpu_cores = info.get('ansible_processor_cores', None) or \
                       len(info.get('ansible_processor', []))
        ___cpu_vcpus = info.get('ansible_processor_vcpus', 0)
        ___memory = '%s %s' % capacity_convert(
            '{} MB'.format(info.get('ansible_memtotal_mb'))
        )
        disk_info = {}
        for dev, dev_info in info.get('ansible_devices', {}).items():
            if disk_pattern.match(dev) and dev_info['removable'] == '0':
                disk_info[dev] = dev_info['size']
        ___disk_total = '%.1f %s' % sum_capacity(disk_info.values())
        ___disk_info = json.dumps(disk_info)

        # ___platform = info.get('ansible_system', 'Unknown')
        ___os = info.get('ansible_distribution', 'Unknown')
        ___os_version = info.get('ansible_distribution_version', 'Unknown')
        ___os_arch = info.get('ansible_architecture', 'Unknown')
        ___hostname_raw = info.get('ansible_hostname', 'Unknown')

        for k, v in locals().items():
            if k.startswith('___'):
                setattr(asset, k.strip('_'), v)
        asset.save()
        assets_updated.append(asset)
    return assets_updated


@shared_task
@org_aware_func("assets")
def update_assets_hardware_info_util(assets, task_name=None):
    """
    Using ansible api to update asset hardware info
    :param assets:  asset seq
    :param task_name: task_name running
    :return: result summary ['contacted': {}, 'dark': {}]
    """
    from ops.utils import update_or_create_ansible_task
    if task_name is None:
        task_name = gettext_noop("Update some assets hardware info. ")
    tasks = const.UPDATE_ASSETS_HARDWARE_TASKS
    hosts = clean_ansible_task_hosts(assets)
    if not hosts:
        return {}
    task, created = update_or_create_ansible_task(
        task_name, hosts=hosts, tasks=tasks,
        pattern='all', options=const.TASK_OPTIONS,
        run_as_admin=True,
    )
    result = task.run()
    set_assets_hardware_info(assets, result)
    return True


@shared_task(queue="ansible")
def update_asset_hardware_info_manual(asset):
    task_name = gettext_noop("Update asset hardware info: ") + str(asset.hostname)
    update_assets_hardware_info_util([asset], task_name=task_name)


@shared_task(queue="ansible")
def update_assets_hardware_info_manual(assets):
    task_name = gettext_noop("Update assets hardware info: ") + str([asset.hostname for asset in assets])
    update_assets_hardware_info_util(assets, task_name=task_name)


@shared_task(queue="ansible")
def update_assets_hardware_info_period():
    """
    Update asset hardware period task
    :return:
    """
    if not const.PERIOD_TASK_ENABLED:
        logger.debug("Period task disabled, update assets hardware info pass")
        return


@shared_task(queue="ansible")
def update_node_assets_hardware_info_manual(node):
    task_name = gettext_noop("Update node asset hardware information: ") + str(node.name)
    assets = node.get_all_assets()
    result = update_assets_hardware_info_util(assets, task_name=task_name)
    return result