From 87e0e1f2c4ddb8ba052e3dc32e3c6096d9949002 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Wed, 1 Aug 2018 13:06:50 +0800
Subject: [PATCH] =?UTF-8?q?[Update]=20Asset=E8=BF=94=E5=9B=9Eorg=20name?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/assets/models/asset.py      | 6 ++++++
 apps/assets/serializers/asset.py | 8 ++++----
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/apps/assets/models/asset.py b/apps/assets/models/asset.py
index 3219ca75b..97bcd4b47 100644
--- a/apps/assets/models/asset.py
+++ b/apps/assets/models/asset.py
@@ -161,6 +161,12 @@ class Asset(OrgModelMixin):
             nodes = list(reduce(lambda x, y: set(x) | set(y), nodes))
         return nodes
 
+    @property
+    def org_name(self):
+        from orgs.models import Organization
+        org = Organization.get_instance(self.org_id)
+        return org.name
+
     @property
     def hardware_info(self):
         if self.cpu_count:
diff --git a/apps/assets/serializers/asset.py b/apps/assets/serializers/asset.py
index 4c648c0ab..36639a17e 100644
--- a/apps/assets/serializers/asset.py
+++ b/apps/assets/serializers/asset.py
@@ -25,7 +25,7 @@ class AssetSerializer(BulkSerializerMixin, serializers.ModelSerializer):
     def get_field_names(self, declared_fields, info):
         fields = super().get_field_names(declared_fields, info)
         fields.extend([
-            'hardware_info', 'is_connective',
+            'hardware_info', 'is_connective', 'org_name'
         ])
         return fields
 
@@ -43,7 +43,7 @@ class AssetGrantedSerializer(serializers.ModelSerializer):
         fields = (
             "id", "hostname", "ip", "port", "system_users_granted",
             "is_active", "system_users_join", "os", 'domain',
-            "platform", "comment", "protocol", "org_id",
+            "platform", "comment", "protocol", "org_id", "org_name",
         )
 
     @staticmethod
@@ -61,6 +61,6 @@ class MyAssetGrantedSerializer(AssetGrantedSerializer):
         model = Asset
         fields = (
             "id", "hostname", "system_users_granted",
-            "is_active", "system_users_join",
-            "os", "platform", "comment",
+            "is_active", "system_users_join", "org_name",
+            "os", "platform", "comment", "org_id", "protocol"
         )