github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-10-22 00:09:14 +00:00
Code Issues Projects Releases Wiki Activity

fix: 修复创建网关时, 没有保存到 Host 列表中的问题;

Browse Source
This commit is contained in:
Bai
2023-02-17 13:56:18 +08:00
committed by Jiangjie.Bai
parent 7040f46565
commit 1bfb434651
2 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
apps/assets/api/asset/asset.py
View File

@@ -16,7 +16,7 @@ from assets.tasks import (
)
from common.api import SuggestionMixin
from common.drf.filters import BaseFilterSet
from common.utils import get_logger
from common.utils import get_logger, is_uuid
from orgs.mixins import generics
from orgs.mixins.api import OrgBulkModelViewSet
from ..mixin import NodeFilterMixin
@@ -31,6 +31,7 @@ __all__ = [
class AssetFilterSet(BaseFilterSet):
labels = django_filters.CharFilter(method='filter_labels')
platform = django_filters.CharFilter(method='filter_platform')
domain = django_filters.CharFilter(method='filter_domain')
type = django_filters.CharFilter(field_name="platform__type", lookup_expr="exact")
category = django_filters.CharFilter(field_name="platform__category", lookup_expr="exact")
domain_enabled = django_filters.BooleanFilter(
@@ -59,7 +60,7 @@ class AssetFilterSet(BaseFilterSet):
model = Asset
fields = [
"id", "name", "address", "is_active", "labels",
"type", "category", "platform"
"type", "category", "platform",
]
@staticmethod
@@ -69,13 +70,21 @@ class AssetFilterSet(BaseFilterSet):
else:
return queryset.filter(platform__name=value)
@staticmethod
def filter_domain(queryset, name, value):
if is_uuid(value):
return queryset.filter(domain_id=value)
else:
return queryset.filter(domain__name__contains=value)
@staticmethod
def filter_labels(queryset, name, value):
if ':' in value:
n, v = value.split(':', 1)
queryset = queryset.filter(labels__name=n, labels__value=v)
else:
queryset = queryset.filter(Q(labels__name=value) | Q(labels__value=value))
q = Q(labels__name__contains=value) | Q(labels__value__contains=value)
queryset = queryset.filter(q)
return queryset