From 90abf36c61dbb53379ee0188bda9d46a9ba8bd22 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 7 May 2021 16:12:17 +0800
Subject: [PATCH] =?UTF-8?q?nginx=20=E6=96=87=E4=BB=B6=E9=89=B4=E6=9D=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client/src/views/qualityinspect/mytask.vue |  2 +-
 server/apps/system/views.py                | 17 ++++++++++++++++-
 server/server/urls.py                      |  3 ++-
 3 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/client/src/views/qualityinspect/mytask.vue b/client/src/views/qualityinspect/mytask.vue
index ba02b51..f5b7d71 100644
--- a/client/src/views/qualityinspect/mytask.vue
+++ b/client/src/views/qualityinspect/mytask.vue
@@ -1,7 +1,7 @@
 <template>
   <div class="app-container">
     
-    <el-card style="margin-top: 10px">
+    <el-card>
       <el-table
         v-loading="listLoading"
         :data="subinspecttasklist"
diff --git a/server/apps/system/views.py b/server/apps/system/views.py
index 85f2e5c..5be7896 100644
--- a/server/apps/system/views.py
+++ b/server/apps/system/views.py
@@ -9,7 +9,7 @@ from django.http import request
 from django.http.response import JsonResponse
 from django_filters.rest_framework import DjangoFilterBackend
 from rest_framework import status
-from rest_framework.decorators import (action, authentication_classes,
+from rest_framework.decorators import (action, api_view, authentication_classes,
                                        permission_classes)
 from rest_framework.filters import OrderingFilter, SearchFilter
 from rest_framework.mixins import (CreateModelMixin, DestroyModelMixin,
@@ -408,6 +408,21 @@ class FileViewSet(ModelViewSet):
         instance = serializer.save(create_by = self.request.user, name=name, size=size, type=type, mime=mime)
         instance.path = settings.MEDIA_URL + instance.file.name
         instance.save()
+    
+
+from rest_framework_simplejwt.state import token_backend
+from django.http import HttpResponseForbidden, HttpResponse
+def mediaauth(request):
+    token = request.COOKIES.get('token', None)
+    if token:
+        try:
+            payload = token_backend.decode(token)
+            if payload.get('user_id', None):
+                return HttpResponse()
+        except:
+            return HttpResponseForbidden()
+    return HttpResponseForbidden()
+
 
 
 
diff --git a/server/server/urls.py b/server/server/urls.py
index 8b37a0d..4d6369a 100644
--- a/server/server/urls.py
+++ b/server/server/urls.py
@@ -30,13 +30,14 @@ from django.conf.urls import url
 
 from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
 from rest_framework_simplejwt.views import TokenViewBase
-from apps.system.views import WXMPlogin
+from apps.system.views import WXMPlogin,mediaauth
 
 
 
 urlpatterns = [
     path('', TemplateView.as_view(template_name="index.html")),
     path('api/admin/', admin.site.urls),
+    path('api/mediaauth/',mediaauth),
     path('api/wxmplogin/',WXMPlogin.as_view()),
     path('api/token/', TokenObtainPairView.as_view(), name='token_obtain_pair'),
     path('api/token2/', Login2View.as_view(), name='token_obtain_2'),