56完善用户信息查看

backend/api/api_graphicFile.py

@@ -0,0 +1,62 @@
+from django.http import HttpResponse
+
+from rest_framework.views import APIView
+
+from api.models import Graph , User
+from api.utils import *
+
+class GraphicSelectAPI(APIView):
+    def get(self , request):
+        user = request.user
+        print(user)
+        graphics = []
+        if(user.identity == 'admin'):
+            graphiclist = Graph.objects.all()
+            print(graphiclist)
+            for graph in graphiclist:
+                graphics.append({
+                    "id": graph.id,
+                    "nodes": graph.nodes,
+                    "type": graph.type,
+                    "edges": graph.edges,
+                    "createTime": graph.create_time,
+                    "updateTime": graph.update_time,
+                    "resultId": graph.result_id,
+                    "userId": graph.user_id 
+                })
+        else:
+            graphiclist = Graph.objects.filter(user_id = user.id)
+            for graph in graphiclist:
+                graphics.append({
+                    "id": graph.id,
+                    "nodes": graph.nodes,
+                    "type": graph.type,
+                    "edges": graph.edges,
+                    "createTime": graph.create_time,
+                    "updateTime": graph.update_time,
+                    "resultId": graph.result_id,
+                    "userId": graph.user_id 
+                })
+        return success(message = "图形数据查询成功" , 
+                        data = graphics , code = 200)
+    
+
+class GraphicDownloadAPI(APIView):
+    def post(self, request):
+        try:
+            print(request.data)
+            graph = Graph.objects.get(id = request.data.get("id"))
+            print(graph)
+        except Graph.DoesNotExist:
+            return failed(message = "文件不存在", status=404)
+
+        # 创建CSV响应
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = f'attachment; filename="{graph.name}.csv"'
+        
+        # 写入CSV内容，假设file.content是CSV格式的字符串
+        response.write(graph.nodes)
+        
+        return success(message = "图形数据下载成功" , 
+                        data = response , code = 200)
+

backend/api/api_taskFile.py

@@ -0,0 +1,40 @@
+from rest_framework.views import APIView
+from api.models import Mission , User
+from api.utils import *
+
+class TaskSelectAPI(APIView):
+        authentication_classes = []
+        permission_classes = []
+
+        def post(self , request):
+            print(request.data.get('username'))
+            user = User.objects.get(username = request.data.get('username'))
+            tasks = []
+            if(user.identity == 'admin'):
+                tasklist = Mission.objects.all()
+                for file in tasklist:
+                    tasks.append({
+                        "id": file.id,
+                        "name": file.name,
+                        "nodeFileId": file.nodeFile_id,
+                        "createTime": file.create_time,
+                        "updateTime": file.update_time,
+                        "state": file.state,
+                        "userId": file.user_id,
+                        "edgeFileId": file.edgeFile_id 
+                    })
+            else:
+                tasklist = Mission.objects.filter(user_id = user.id) 
+                for file in tasklist:
+                    tasks.append({
+                        "id": file.id,
+                        "name": file.name,
+                        "nodeFileId": file.nodeFile_id,
+                        "createTime": file.create_time,
+                        "updateTime": file.update_time,
+                        "userId": file.user_id,
+                        "edgeFileId": file.edgeFile_id 
+                    })
+            return success(message = "任务数据查询成功" , 
+                           data = tasks , code = 200)
+    

backend/api/api_user.py

@@ -6,6 +6,8 @@ from .serializers import UserRegisterSerializer
 
 from django.middleware.csrf import get_token
 from django.contrib.auth import login
+from api.models import User
+from django.contrib.auth.hashers import make_password , check_password
 
 from api.utils import *
 
@@ -81,4 +83,58 @@ class getDashboard(APIView):
         return Response({
             'data': 'yes'
         })
-        
+
+
+#查询用户信息
+class UserSelectAPI(APIView):
+    def get(self, request):
+        user = request.user
+        return success(message="查询用户信息成功", data = {
+                'username': user.username,
+                'displayname': user.displayname,
+                'createtime': user.create_time,
+                'identity': user.identity,
+                'lastlogin': user.last_login
+            }, code=200)
+    
+#修改用户信息
+class UserUpdateAPI(APIView):
+    def post(self, request):
+        try:
+            user = request.user
+            user.username = request.data.get('username')
+            user.displayname = request.data.get('displayname')
+            user.identity = request.data.get('identity')
+            user.save()
+            return success(message="修改用户信息成功", data = {
+                'username': user.username,
+                'displayname': user.displayname,
+                'createtime': user.create_time,
+                'identity': user.identity,
+                'lastlogin': user.last_login
+            }, code=200)
+        except Exception as e:
+            return failed(message="修改用户信息失败", code=400)
+    
+#修改用户密码
+class PassUpdateAPI(APIView):
+    def post(self, request):
+        try:
+            user = request.user
+            password = make_password(
+                request.data.get("oldPassword"),
+                salt='vrviewer',
+                hasher='pbkdf2_sha256'
+            )
+            if(check_password(request.data.get("oldPassword") , user.password)):
+                user.password = make_password(
+                request.data.get("newPassword"),
+                salt='vrviewer',
+                hasher='pbkdf2_sha256'
+                )
+                user.save()
+                return success(message="用户密码修改成功", code=200)
+            else:
+                return failed(message="用户密码修改失败,原密码错误", code=400)
+        except Exception as e:
+            return failed(message="用户密码修改失败", code=400)

backend/api/models/file.py

@@ -4,6 +4,7 @@ import csv
 from api.utils import *
 import json
 from random import randint
+import logging
 
 types = [
     ('csv', 'csv'),
@@ -21,8 +22,7 @@ contents = [
     ('edge', 'edge'),
 ]
 
-
-
+logger = logging.getLogger("file-model")
 
 class FileManager(models.Manager):
     def getHistory(self, user):
@@ -36,7 +36,7 @@ class FileManager(models.Manager):
             try:
                 size = os.path.getsize(path)
             except FileNotFoundError:
-                print("未找到对应文件，现将记录删除1", fileId, file.name)
+                print("未找到对应文件，现将记录删除", fileId, file.name)
                 self.get(id=fileId).delete()
                 continue
             except Exception as error:
@@ -49,12 +49,30 @@ class FileManager(models.Manager):
             else:
                 size = size / 1024
                 size = f"{size:.2f} KB"
+            if file.content == 'node':
+                missions = file.own_missions_node.all()
+                fileInfo = {
+                    '节点总数': file.own_file_info.nodes,
+                    'S节点数': file.own_file_info.sNodes,
+                    'D节点数': file.own_file_info.dNodes,
+                    'I节点数': file.own_file_info.iNodes,
+                }
+            elif file.content == 'edge':
+                missions = file.own_missions_edge.all()
+                fileInfo = {
+                    '边总数': file.own_file_info.edges,
+                }
+            else:
+                logger.error(f"获取历史文件出错，文件格式错误 content: {file.content}")
+                return FAILED
             history.append({
                 'id': file.id,
                 'name': file.name,
                 'uploadTime': file.update_time,
                 'size': size,
                 'content': file.content,
+                'missions': [{'id': mission.id, 'name': mission.name} for mission in missions],
+                'fileInfo': fileInfo,
             })
         return history
         # except Exception as error:

backend/api/urls.py

@@ -1,5 +1,5 @@
 from django.urls import path
-from .api_user import UserRegisterAPI, UserLoginAPI
+from .api_user import UserRegisterAPI, UserLoginAPI, UserUpdateAPI , UserSelectAPI , PassUpdateAPI
 from .api_user import getDashboard
 from .api_prepare import UploadFileAPI, PlanAPI, InputFileAPI, MissionsAPI
 from .api_calculate import CalculateAPI
@@ -9,6 +9,9 @@ from .api_graph import GenerateGraph, ViewGraphByToken
 from .api_alert import AlertAPI, AlertCheck
 from .api_system import SystemPerformanceAPI
 
+from .api_graphicFile import GraphicSelectAPI , GraphicDownloadAPI
+from .api_taskFile import TaskSelectAPI
+
 urlpatterns = [
     path('register/', UserRegisterAPI.as_view(), name='user_register_api'),
     path('login/', UserLoginAPI.as_view(), name='user_login_api'),
@@ -25,4 +28,11 @@ urlpatterns = [
     path('systemPerformance/', SystemPerformanceAPI.as_view(), name="system_performance_api"),
     path('inputFile/', InputFileAPI.as_view(), name='input_file_api'),
     path('missions/', MissionsAPI.as_view(), name="missions_api"),
+
+    path('selectUser/', UserSelectAPI.as_view(), name='user_select_api'),
+    path('updateUser/', UserUpdateAPI.as_view(), name='user_update_api'),
+    path('updatePass/', PassUpdateAPI.as_view(), name='pass_update_api'),
+    path('selectGraphic/' , GraphicSelectAPI.as_view() , name = "graphic_select_api"),
+    path('selectTask/' , TaskSelectAPI.as_view() , name = "task_select_api"),
+    path('downloadGraphic/' , GraphicDownloadAPI.as_view() , name = "download_graphic_api")
 ]