jumpserver/apps/assets/models/asset.py

187 lines
6.7 KiB
Python
Raw Normal View History

2016-12-20 16:43:52 +00:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
2017-11-23 06:08:01 +00:00
import uuid
import logging
2018-03-23 11:46:46 +00:00
import random
2017-11-23 06:08:01 +00:00
2016-12-20 16:43:52 +00:00
from django.db import models
from django.utils.translation import ugettext_lazy as _
2017-12-10 16:29:25 +00:00
from django.core.cache import cache
2016-12-20 16:43:52 +00:00
2017-12-15 07:50:15 +00:00
from ..const import ASSET_ADMIN_CONN_CACHE_KEY
from .user import AdminUser, SystemUser
2016-12-20 16:43:52 +00:00
2017-03-24 06:48:18 +00:00
__all__ = ['Asset']
2016-12-20 16:43:52 +00:00
logger = logging.getLogger(__name__)
2018-01-09 15:33:14 +00:00
def default_cluster():
from .cluster import Cluster
name = "Default"
defaults = {"name": name}
cluster, created = Cluster.objects.get_or_create(
defaults=defaults, name=name
)
2018-01-10 04:03:35 +00:00
return cluster.id
2018-01-09 15:33:14 +00:00
def default_node():
try:
2018-02-09 07:24:44 +00:00
from .node import Node
return Node.root()
except:
return None
2016-12-20 16:43:52 +00:00
class Asset(models.Model):
2017-03-15 16:19:47 +00:00
# Important
PLATFORM_CHOICES = (
('Linux', 'Linux'),
('Unix', 'Unix'),
('MacOS', 'MacOS'),
('BSD', 'BSD'),
('Windows', 'Windows'),
('Other', 'Other'),
)
2017-11-23 06:08:01 +00:00
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
2016-12-20 16:43:52 +00:00
ip = models.GenericIPAddressField(max_length=32, verbose_name=_('IP'), db_index=True)
hostname = models.CharField(max_length=128, unique=True, verbose_name=_('Hostname'))
port = models.IntegerField(default=22, verbose_name=_('Port'))
2018-03-23 11:46:46 +00:00
domain = models.ForeignKey("assets.Domain", null=True, blank=True, related_name='assets', verbose_name=_("Domain"))
nodes = models.ManyToManyField('assets.Node', default=default_node, related_name='assets', verbose_name=_("Nodes"))
2017-03-15 16:19:47 +00:00
is_active = models.BooleanField(default=True, verbose_name=_('Is active'))
# Auth
2018-02-25 10:40:15 +00:00
admin_user = models.ForeignKey('assets.AdminUser', on_delete=models.PROTECT, null=True, verbose_name=_("Admin user"))
2017-03-15 16:19:47 +00:00
# Some information
2017-12-07 05:01:33 +00:00
public_ip = models.GenericIPAddressField(max_length=32, blank=True, null=True, verbose_name=_('Public IP'))
2017-03-15 16:19:47 +00:00
number = models.CharField(max_length=32, null=True, blank=True, verbose_name=_('Asset number'))
# Collect
vendor = models.CharField(max_length=64, null=True, blank=True, verbose_name=_('Vendor'))
model = models.CharField(max_length=54, null=True, blank=True, verbose_name=_('Model'))
2016-12-20 16:43:52 +00:00
sn = models.CharField(max_length=128, null=True, blank=True, verbose_name=_('Serial number'))
2017-03-15 16:19:47 +00:00
cpu_model = models.CharField(max_length=64, null=True, blank=True, verbose_name=_('CPU model'))
cpu_count = models.IntegerField(null=True, verbose_name=_('CPU count'))
cpu_cores = models.IntegerField(null=True, verbose_name=_('CPU cores'))
memory = models.CharField(max_length=64, null=True, blank=True, verbose_name=_('Memory'))
disk_total = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('Disk total'))
disk_info = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('Disk info'))
platform = models.CharField(max_length=128, choices=PLATFORM_CHOICES, default='Linux', verbose_name=_('Platform'))
2017-03-15 16:19:47 +00:00
os = models.CharField(max_length=128, null=True, blank=True, verbose_name=_('OS'))
2017-04-07 08:59:09 +00:00
os_version = models.CharField(max_length=16, null=True, blank=True, verbose_name=_('OS version'))
os_arch = models.CharField(max_length=16, blank=True, null=True, verbose_name=_('OS arch'))
2017-03-15 16:19:47 +00:00
hostname_raw = models.CharField(max_length=128, blank=True, null=True, verbose_name=_('Hostname raw'))
2018-01-16 08:32:06 +00:00
labels = models.ManyToManyField('assets.Label', blank=True, related_name='assets', verbose_name=_("Labels"))
2016-12-20 16:43:52 +00:00
created_by = models.CharField(max_length=32, null=True, blank=True, verbose_name=_('Created by'))
2017-04-07 08:59:09 +00:00
date_created = models.DateTimeField(auto_now_add=True, null=True, blank=True, verbose_name=_('Date created'))
2016-12-20 16:43:52 +00:00
comment = models.TextField(max_length=128, default='', blank=True, verbose_name=_('Comment'))
2017-11-01 15:23:11 +00:00
def __str__(self):
return self.hostname
2016-12-20 17:03:52 +00:00
2016-12-20 16:43:52 +00:00
@property
def is_valid(self):
warning = ''
if not self.is_active:
warning += ' inactive'
else:
return True, ''
return False, warning
def is_unixlike(self):
if self.platform not in ("Windows",):
return True
else:
return False
@property
2017-12-12 04:19:45 +00:00
def hardware_info(self):
if self.cpu_count:
return '{} Core {} {}'.format(
self.cpu_count * self.cpu_cores,
self.memory, self.disk_total
)
else:
return ''
@property
2017-12-12 04:19:45 +00:00
def is_connective(self):
if not self.is_unixlike():
return True
2017-12-15 07:50:15 +00:00
val = cache.get(ASSET_ADMIN_CONN_CACHE_KEY.format(self.hostname))
2017-12-12 04:19:45 +00:00
if val == 1:
return True
else:
2017-12-12 04:19:45 +00:00
return False
2016-12-20 17:03:52 +00:00
def to_json(self):
2018-03-23 11:46:46 +00:00
info = {
'id': self.id,
'hostname': self.hostname,
'ip': self.ip,
'port': self.port,
}
2018-03-23 11:46:46 +00:00
if self.domain and self.domain.gateway_set.all():
info["gateways"] = [d.id for d in self.domain.gateway_set.all()]
return info
2016-12-20 16:43:52 +00:00
def get_auth_info(self):
if self.admin_user:
return {
'username': self.admin_user.username,
'password': self.admin_user.password,
'private_key': self.admin_user.private_key_file,
'become': self.admin_user.become_info,
}
2017-03-05 03:38:02 +00:00
def _to_secret_json(self):
2017-12-07 08:25:50 +00:00
"""
2017-12-12 04:19:45 +00:00
Ansible use it create inventory, First using asset user,
otherwise using cluster admin user
2017-03-05 03:38:02 +00:00
2017-12-07 08:25:50 +00:00
Todo: May be move to ops implements it
"""
data = self.to_json()
if self.admin_user:
admin_user = self.admin_user
2017-12-07 05:01:33 +00:00
data.update({
2017-12-12 04:19:45 +00:00
'username': admin_user.username,
'password': admin_user.password,
'private_key': admin_user.private_key_file,
2017-12-28 06:25:56 +00:00
'become': admin_user.become_info,
'groups': [node.value for node in self.nodes.all()],
2017-12-07 05:01:33 +00:00
})
return data
2016-12-20 16:43:52 +00:00
class Meta:
unique_together = ('ip', 'port')
2018-01-05 09:57:02 +00:00
verbose_name = _("Asset")
2016-12-20 16:43:52 +00:00
@classmethod
def generate_fake(cls, count=100):
from random import seed, choice
import forgery_py
from django.db import IntegrityError
seed()
for i in range(count):
asset = cls(ip='%s.%s.%s.%s' % (i, i, i, i),
hostname=forgery_py.internet.user_name(True),
admin_user=choice(AdminUser.objects.all()),
port=22,
created_by='Fake')
try:
asset.save()
asset.system_users = [choice(SystemUser.objects.all()) for i in range(3)]
logger.debug('Generate fake asset : %s' % asset.ip)
except IntegrityError:
print('Error continue')
continue