From 005272cdc08230c4cd5a95216752f780a76c4bbe Mon Sep 17 00:00:00 2001
From: Bai <bugatti_it@163.com>
Date: Tue, 7 Jan 2020 15:53:53 +0800
Subject: [PATCH] =?UTF-8?q?[Update]=20=E6=9B=B4=E6=96=B0=E7=B3=BB=E7=BB=9F?=
 =?UTF-8?q?=E7=94=A8=E6=88=B7=E8=AE=A4=E8=AF=81=E4=BF=A1=E6=81=AFprivate?=
 =?UTF-8?q?=5Fkey=E8=BD=AC=E6=8D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/assets/models/base.py             | 9 +++++++++
 apps/assets/serializers/system_user.py | 5 +++++
 2 files changed, 14 insertions(+)

diff --git a/apps/assets/models/base.py b/apps/assets/models/base.py
index 895e0134d..5d0f3cef1 100644
--- a/apps/assets/models/base.py
+++ b/apps/assets/models/base.py
@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 #
+import io
 import os
 import uuid
 from hashlib import md5
@@ -76,6 +77,14 @@ class AssetUser(OrgModelMixin):
         i = '-'.join(str(self.id).split('-')[:3])
         return i
 
+    def get_private_key(self):
+        if not self.private_key_obj:
+            return None
+        string_io = io.StringIO()
+        self.private_key_obj.write_private_key(string_io)
+        private_key = string_io.getvalue()
+        return private_key
+
     def get_related_assets(self):
         assets = self.assets.all()
         return assets
diff --git a/apps/assets/serializers/system_user.py b/apps/assets/serializers/system_user.py
index 5295d74b2..e16a7a271 100644
--- a/apps/assets/serializers/system_user.py
+++ b/apps/assets/serializers/system_user.py
@@ -123,6 +123,7 @@ class SystemUserAuthSerializer(AuthSerializer):
     """
     系统用户认证信息
     """
+    private_key = serializers.SerializerMethodField()
 
     class Meta:
         model = SystemUser
@@ -131,6 +132,10 @@ class SystemUserAuthSerializer(AuthSerializer):
             "login_mode", "password", "private_key",
         ]
 
+    @staticmethod
+    def get_private_key(obj):
+        return obj.get_private_key()
+
 
 class SystemUserSimpleSerializer(serializers.ModelSerializer):
     """