mirror of https://github.com/jumpserver/jumpserver
86 lines
3.0 KiB
Python
86 lines
3.0 KiB
Python
#!/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):
|
|
from assets.const import AllTypes
|
|
|
|
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',
|
|
'create_account_method': 'create_account_aix',
|
|
'change_password_method': 'change_password_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()))
|