github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

[Update]优化资产设置节点的问题 (#2786) 

* [Update]优化资产设置节点的问题

* [Update]优化资产创建的节点设置
pull/2790/head
八千流 2019-06-12 17:49:30 +08:00 committed by BaiJiangJie
parent 3e86c07411
commit 69b91df96b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,10 +51,12 @@ class AssetViewSet(IDInCacheFilterMixin, LabelFilter, BulkModelViewSet):
def set_assets_node(self, assets): def set_assets_node(self, assets):
if not isinstance(assets, list): if not isinstance(assets, list):
assets = [assets] assets = [assets]
node = Node.root()
node_id = self.request.query_params.get('node_id') node_id = self.request.query_params.get('node_id')
if node_id: if not node_id:
node = get_object_or_none(Node, pk=node_id) return
node = get_object_or_none(Node, pk=node_id)
if not node:
return
node.assets.add(*assets) node.assets.add(*assets)
def perform_create(self, serializer): def perform_create(self, serializer):