mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-05-06 15:16:32 +00:00
3159a4e794
* perf: change domain to zone * perf: change domain to zone * perf: change some word * perf: update gateway enabled i18n * perf: change migrations --------- Co-authored-by: ibuler <ibuler@qq.com>
58 lines
1.4 KiB
Python
58 lines
1.4 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, lazyproperty
|
|
from labels.mixins import LabeledMixin
|
|
from orgs.mixins.models import JMSOrgBaseModel
|
|
from .gateway import Gateway
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
__all__ = ['Zone']
|
|
|
|
|
|
class Zone(LabeledMixin, JMSOrgBaseModel):
|
|
name = models.CharField(max_length=128, verbose_name=_('Name'))
|
|
|
|
class Meta:
|
|
verbose_name = _("Zone")
|
|
unique_together = [('org_id', 'name')]
|
|
ordering = ('name',)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def select_gateway(self):
|
|
return self.random_gateway()
|
|
|
|
@lazyproperty
|
|
def assets_amount(self):
|
|
return self.assets.gateways(0).count()
|
|
|
|
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(zone=self)
|
|
return queryset
|
|
|
|
@classmethod
|
|
def get_gateway_queryset(cls):
|
|
return Gateway.objects.all()
|