mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.3 KiB
72 lines
2.3 KiB
# ~*~ coding: utf-8 ~*~
|
|
from celery import shared_task
|
|
import json
|
|
|
|
from ops.tasks import run_AdHoc
|
|
from common.utils import get_object_or_none, capacity_convert, sum_capacity
|
|
from .models import Asset
|
|
|
|
|
|
@shared_task
|
|
def update_assets_hardware_info(assets):
|
|
task_tuple = (
|
|
('setup', ''),
|
|
)
|
|
task_name = ','.join([asset.hostname for asset in assets])
|
|
summary, result = run_AdHoc(task_tuple, assets, record=True, task_name=task_name)
|
|
for hostname, info in result['contacted'].items():
|
|
if info:
|
|
info = info[0]['ansible_facts']
|
|
else:
|
|
continue
|
|
asset = get_object_or_none(Asset, hostname=hostname)
|
|
if not asset:
|
|
continue
|
|
|
|
___vendor = info['ansible_system_vendor']
|
|
___model = info['ansible_product_version']
|
|
___sn = info['ansible_product_serial']
|
|
|
|
for ___cpu_model in info['ansible_processor']:
|
|
if ___cpu_model.endswith('GHz'):
|
|
break
|
|
else:
|
|
___cpu_model = 'Unknown'
|
|
___cpu_count = info['ansible_processor_count']
|
|
___cpu_cores = info['ansible_processor_cores']
|
|
___memory = '%s %s' % capacity_convert('{} MB'.format(info['ansible_memtotal_mb']))
|
|
disk_info = {}
|
|
for dev, dev_info in info['ansible_devices'].items():
|
|
if dev_info['removable'] == '0':
|
|
disk_info[dev] = dev_info['size']
|
|
___disk_total = '%s %s' % sum_capacity(disk_info.values())
|
|
___disk_info = json.dumps(disk_info)
|
|
|
|
___platform = info['ansible_system']
|
|
___os = info['ansible_distribution']
|
|
___os_version = float(info['ansible_distribution_version'])
|
|
___os_arch = info['ansible_architecture']
|
|
___hostname_raw = info['ansible_hostname']
|
|
|
|
for k, v in locals().items():
|
|
if k.startswith('___'):
|
|
setattr(asset, k.strip('_'), v)
|
|
asset.save()
|
|
|
|
|
|
@shared_task(name="asset_test_ping_check")
|
|
def asset_test_ping_check(assets):
|
|
task_tuple = (
|
|
('ping', ''),
|
|
)
|
|
hoc = AdHocRunner(assets)
|
|
result = hoc.run(task_tuple)
|
|
return result['contacted'].keys(), result['dark'].keys()
|
|
|
|
|
|
def get_assets_hardware_info(assets):
|
|
task_tuple = (
|
|
('setup', ''),
|
|
)
|
|
task = run_AdHoc.delay(task_tuple, assets, record=False)
|
|
return task |