mirror of https://github.com/jumpserver/jumpserver
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
import uuid
|
|
import random
|
|
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orgs.mixins import OrgModelMixin
|
|
from .base import AssetUser
|
|
|
|
__all__ = ['Domain', 'Gateway']
|
|
|
|
|
|
class Domain(OrgModelMixin):
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
name = models.CharField(max_length=128, unique=True, verbose_name=_('Name'))
|
|
comment = models.TextField(blank=True, verbose_name=_('Comment'))
|
|
date_created = models.DateTimeField(auto_now_add=True, null=True,
|
|
verbose_name=_('Date created'))
|
|
|
|
class Meta:
|
|
verbose_name = _("Domain")
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def has_gateway(self):
|
|
return self.gateway_set.filter(is_active=True).exists()
|
|
|
|
@property
|
|
def gateways(self):
|
|
return self.gateway_set.filter(is_active=True)
|
|
|
|
def random_gateway(self):
|
|
return random.choice(self.gateways)
|
|
|
|
|
|
class Gateway(AssetUser):
|
|
SSH_PROTOCOL = 'ssh'
|
|
RDP_PROTOCOL = 'rdp'
|
|
PROTOCOL_CHOICES = (
|
|
(SSH_PROTOCOL, 'ssh'),
|
|
(RDP_PROTOCOL, 'rdp'),
|
|
)
|
|
ip = models.GenericIPAddressField(max_length=32, verbose_name=_('IP'), db_index=True)
|
|
port = models.IntegerField(default=22, verbose_name=_('Port'))
|
|
protocol = models.CharField(choices=PROTOCOL_CHOICES, max_length=16, default=SSH_PROTOCOL, verbose_name=_("Protocol"))
|
|
domain = models.ForeignKey(Domain, on_delete=models.CASCADE, verbose_name=_("Domain"))
|
|
comment = models.CharField(max_length=128, blank=True, null=True, verbose_name=_("Comment"))
|
|
is_active = models.BooleanField(default=True, verbose_name=_("Is active"))
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Meta:
|
|
unique_together = [('name', 'org_id')]
|
|
verbose_name = _("Gateway")
|