mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-17 07:58:52 +00:00
8291a81efd
* perf: 支持全局的 labels * perf: stash * stash * stash * stash * stash * perf: 优化 labels * stash * perf: add debug sql * perf: 修改 labels * perf: 优化提交 * perf: 优化提交 labels * perf: 基本完成 * perf: 完成 labels 搜索 * perf: 优化 labels * perf: 去掉不用 debug --------- Co-authored-by: ibuler <ibuler@qq.com>
54 lines
1.3 KiB
Python
54 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 labels.mixins import LabeledMixin
|
|
from orgs.mixins.models import JMSOrgBaseModel
|
|
from .gateway import Gateway
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
__all__ = ['Domain']
|
|
|
|
|
|
class Domain(LabeledMixin, 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()
|