mirror of https://github.com/jumpserver/jumpserver
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import random
|
|
|
|
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from common.utils import get_logger
|
|
from orgs.mixins.models import JMSOrgBaseModel
|
|
from .gateway import Gateway
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
__all__ = ['Domain']
|
|
|
|
|
|
class Domain(JMSOrgBaseModel):
|
|
name = models.CharField(max_length=128, verbose_name=_('Name'))
|
|
|
|
class Meta:
|
|
verbose_name = _("Domain")
|
|
unique_together = [('org_id', 'name')]
|
|
ordering = ('name',)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def select_gateway(self):
|
|
return self.random_gateway()
|
|
|
|
def random_gateway(self):
|
|
gateways = [gw for gw in self.active_gateways if gw.is_connective]
|
|
|
|
if not gateways:
|
|
gateways = self.active_gateways
|
|
if not gateways:
|
|
logger.warn(f'Not active gateway, domain={self}, pass')
|
|
return None
|
|
return random.choice(gateways)
|
|
|
|
@property
|
|
def active_gateways(self):
|
|
return self.gateways.filter(is_active=True)
|
|
|
|
@property
|
|
def gateways(self):
|
|
queryset = self.get_gateway_queryset().filter(domain=self)
|
|
return queryset
|
|
|
|
@classmethod
|
|
def get_gateway_queryset(cls):
|
|
return Gateway.objects.all()
|