瀏覽代碼

EmployeeViewSet

master
kj1352 3 年之前
父節點
當前提交
92a2affe9f
共有 3 個檔案被更改,包括 12 行新增11 行删除
  1. +2
    -2
      client/models.py
  2. +6
    -2
      employee/urls.py
  3. +4
    -7
      employee/views.py

+ 2
- 2
client/models.py 查看文件

@@ -37,8 +37,8 @@ class Address(models.Model):
address_line1 = models.CharField(max_length=50)
address_line2 = models.CharField(max_length=50)
postal_code = models.CharField(max_length=15)
tax_type = models.ForeignKey(TaxType, on_delete=models.CASCADE)
tax_id = models.CharField(max_length=30)
tax_type = models.ForeignKey(TaxType, on_delete=models.CASCADE, blank=True, null=True)
tax_id = models.CharField(max_length=30, blank=True, null=True)

def __str__(self):
return self.address_line1


+ 6
- 2
employee/urls.py 查看文件

@@ -1,6 +1,10 @@
from django.urls import path
from django.urls import path, include
from . import views
from rest_framework import routers

router = routers.DefaultRouter()
router.register(r'employees', views.EmployeeViewSet)

urlpatterns = [
path('employees/', views.EmployeeList.as_view()),
path('', include(router.urls)),
]

+ 4
- 7
employee/views.py 查看文件

@@ -1,16 +1,13 @@
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework import viewsets
from rest_framework.authentication import TokenAuthentication, SessionAuthentication
from rest_framework.permissions import IsAuthenticated
from .models import Employee
from .serializers import EmployeeSerializer


class EmployeeList(APIView):
class EmployeeViewSet(viewsets.ReadOnlyModelViewSet):
queryset = Employee.objects.all()
serializer_class = EmployeeSerializer
authentication_classes = [TokenAuthentication, SessionAuthentication]
permission_classes = [IsAuthenticated]

def get(self, request):
employees = Employee.objects.all()
serializer = EmployeeSerializer(employees, many=True)
return Response(serializer.data)