mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
704 B
26 lines
704 B
from django.db.models import TextChoices
|
|
from django.utils.translation import ugettext_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)
|