mirror of https://github.com/jumpserver/jumpserver
26 lines
703 B
Python
26 lines
703 B
Python
from django.db.models import TextChoices
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
|
|
class Connectivity(TextChoices):
|
|
UNKNOWN = '-', _('Unknown')
|
|
OK = 'ok', _('Ok')
|
|
ERR = 'err', _('Error')
|
|
|
|
|
|
class AutomationTypes(TextChoices):
|
|
ping = 'ping', _('Ping')
|
|
ping_gateway = 'ping_gateway', _('Ping gateway')
|
|
gather_facts = 'gather_facts', _('Gather facts')
|
|
|
|
@classmethod
|
|
def get_type_model(cls, tp):
|
|
from assets.models import (
|
|
PingAutomation, GatherFactsAutomation,
|
|
)
|
|
type_model_dict = {
|
|
cls.ping: PingAutomation,
|
|
cls.gather_facts: GatherFactsAutomation,
|
|
}
|
|
return type_model_dict.get(tp)
|