jumpserver/apps/assets/const/device.py

60 lines
1.5 KiB
Python

from django.utils.translation import gettext_lazy as _
from .base import BaseType
class DeviceTypes(BaseType):
GENERAL = 'general', _("General")
SWITCH = 'switch', _("Switch")
ROUTER = 'router', _("Router")
FIREWALL = 'firewall', _("Firewall")
@classmethod
def _get_base_constrains(cls) -> dict:
return {
'*': {
'charset_enabled': False,
'domain_enabled': True,
'su_enabled': False,
}
}
@classmethod
def _get_protocol_constrains(cls) -> dict:
return {
'*': {
'choices': ['ssh', 'telnet']
}
}
@classmethod
def _get_automation_constrains(cls) -> dict:
return {
'*': {
'ansible_enabled': True,
'ansible_config': {
'ansible_connection': 'local',
},
'ping_enabled': True,
'gather_facts_enabled': False,
'gather_accounts_enabled': False,
'verify_account_enabled': False,
'change_secret_enabled': False,
}
}
@classmethod
def internal_platforms(cls):
return {
cls.GENERAL: [{'name': 'General'}, {'name': 'Cisco'}, {'name': 'Huawei'}, {'name': 'H3C'}],
cls.SWITCH: [],
cls.ROUTER: [],
cls.FIREWALL: []
}
@classmethod
def get_community_types(cls):
return [
cls.GENERAL, cls.SWITCH, cls.ROUTER, cls.FIREWALL
]