jumpserver/apps/assets/models/utils.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#

from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_lazy as _

from common.utils import validate_ssh_private_key

__all__ = [
    'private_key_validator',
]


def private_key_validator(value):
    if not validate_ssh_private_key(value):
        raise ValidationError(
            _('%(value)s is not an even number'),
            params={'value': value},
        )


def update_internal_platforms(platform_model):
    platforms = [
        {'name': 'Linux', 'category': 'host', 'type': 'linux'},
        {'name': 'BSD', 'category': 'host', 'type': 'unix'},
        {'name': 'Unix', 'category': 'host', 'type': 'unix'},
        {'name': 'MacOS', 'category': 'host', 'type': 'unix'},
        {'name': 'Windows', 'category': 'host', 'type': 'unix'},
        {
            'name': 'AIX', 'category': 'host', 'type': 'unix',
            'change_secret_method': 'change_secret_aix',
        },
        {'name': 'Windows', 'category': 'host', 'type': 'windows'},
        {
            'name': 'Windows-TLS', 'category': 'host', 'type': 'windows',
            'protocols': [
                {'name': 'rdp', 'port': 3389, 'setting': {'security': 'tls'}},
                {'name': 'ssh', 'port': 22},
            ]
        },
        {
            'name': 'Windows-RDP', 'category': 'host', 'type': 'windows',
            'protocols': [
                {'name': 'rdp', 'port': 3389, 'setting': {'security': 'rdp'}},
                {'name': 'ssh', 'port': 22},
            ]
        },
        # 数据库
        {'name': 'MySQL', 'category': 'database', 'type': 'mysql'},
        {'name': 'PostgreSQL', 'category': 'database', 'type': 'postgresql'},
        {'name': 'Oracle', 'category': 'database', 'type': 'oracle'},
        {'name': 'SQLServer', 'category': 'database', 'type': 'sqlserver'},
        {'name': 'MongoDB', 'category': 'database', 'type': 'mongodb'},
        {'name': 'Redis', 'category': 'database', 'type': 'redis'},

        # 网络设备
        {'name': 'Generic', 'category': 'device', 'type': 'general'},
        {'name': 'Huawei', 'category': 'device', 'type': 'general'},
        {'name': 'Cisco', 'category': 'device', 'type': 'general'},
        {'name': 'H3C', 'category': 'device', 'type': 'general'},

        # Web
        {'name': 'Website', 'category': 'web', 'type': 'general'},

        # Cloud
        {'name': 'Kubernetes', 'category': 'cloud', 'type': 'k8s'},
        {'name': 'VMware vSphere', 'category': 'cloud', 'type': 'private'},
    ]

    platforms = platform_model.objects.all()

    updated = []
    for p in platforms:
        attrs = platform_ops_map.get((p.category, p.type), {})
        if not attrs:
            continue
        for k, v in attrs.items():
            setattr(p, k, v)
        updated.append(p)
    platform_model.objects.bulk_update(updated, list(default_ok.keys()))