使用 Django Rest Framework 创建 REST Api

Question

我正在尝试为以下方法创建一个 Restful Api 来运行 jenkins 作业以在 Saucelabs 上运行。 我想使用 restful API 对作业进行排队。 我正在使用 Django Restful 框架。

1. CreateMethod ：接受两个字段：ProjectName 和 URL，并返回一个令牌 ID。

2. VerifyStatus：接受 Token ID 并返回三个字段。 TokenID，运行：True/False 和 no_of_jobs：整数值（如果未指定，则为 0）

3. relseaseMethod：接受释放令牌调用，如果成功则返回真。

我是 Restful API 的新手，我正在尝试在酱实验室上运行 Jenkins 作业，并在 python Djano restframework 上使用 restful api 将它们排队。 引导我过去。

视图.py

class RegisterTestMethodView(APIView):
authentication_classes = [SessionAuthentication, TokenAuthentication, BasicAuthentication]
permission_classes = [IsAuthenticated]  #No access (not even read if not authorized)

def post(self, request, format=None):
    serializer = RegisterTestMethodSerializers(data=request.data)
    if serializer.is_valid():
        serializer.save()
        return Response({'tokenid':serializer.data['id']}, status=status.HTTP_201_CREATED)
    return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)


class CheckStatusView(APIView):


def get_object(self, pk):
    try:
        return Jobs.objects.get(pk=pk)
    except Jobs.DoesNotExist:
        raise Http404

def get(self, request, pk, format=None):
    snippet = self.get_object(pk)
    serializer = RegisterTestMethodSerializers(snippet)
    return Response({"tokenid":serializer.data["id"], "Runtestnow" : False, "VMcount" : 0 })


class ReleaseTokenView(APIView):

def get_object(self, pk):
    try:
        return Jobs.objects.get(pk=pk)
    except Jobs.DoesNotExist:
        raise Http404

def delete(self, request, pk, format=None):
    snippet = self.get_object(pk)
    snippet.delete()
    return Response(data={'Deleted':True}, status=status.HTTP_204_NO_CONTENT)

Serailizers.py

rom rest_framework import serializers
from .models import Jobs
from random import random



RegisterTestMethodSerializers(serializers.HyperlinkedModelSerializer):

    class Meta:
        model = Jobs
        fields = ('id','name','url')

模型.py

from django.db import models

# Create your models here.
class Jobs(models.Model):
    name = models.CharField(max_length=100)
    url = models.URLField()

    def __str__(self):
        return self.name

网址.py

from django.urls import path, include
from . import views
from .views import (RegisterTestMethodView,
                    RegisterTestMethodViewDetail,
                    CheckStatusView,
                    ReleaseTokenView
                    )
from rest_framework import routers
from rest_framework.authtoken.views import obtain_auth_token
from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView



urlpatterns = [
    path('', include(router.urls)),
    path('registertestmethod/',RegisterTestMethodView.as_view()),
    path('registertestmethod/<int:pk>/', 
    RegisterTestMethodViewDetail.as_view()),
    path('checkstatus/<int:pk>', CheckStatusView.as_view()),
    path('releasetoken/<int:pk>', ReleaseTokenView.as_view()),
]

我在这里添加了我的代码。 我的项目中还有其他类和函数。我试图删除所有这些。 因此，您可能会看到额外的导入。 我的代码做了以下事情。

POST --> http://localhost:8000/registertestmethod/

{

    "name": "Name",
    "url": "https://www.google.com"
}

返回

{
    "tokenid": 32 #unique token ID return 
}

这个tokenid应该很长，我现在正在使用 id。

GET --> http://localhost:8000/checkstatus/32

回来了

{
    "tokenid": 32, #unique tokenid refering to register info
    "Runtestnow": false, #if job is running
    "VMcount": 0 # number of VM used in sauce lab by the Jobs
}

DELETE --> http://localhost:8000/releasetoken/32 releasetoken DELETE --> http://localhost:8000/releasetoken/32 #完成后应该删除作业。 正在删除和返回

{
    "Deleted": true
}

我希望它是动态的并将信息存储在数据库中。 令牌应返回 checkstatus 上的所有内容。

Answer 1

您可以为此使用 ModelViewSet 方法，这是 API 端点的一个非常简单的示例。

视图.py

from rest_framework.viewsets import ModelViewSet
from api.serializers import SaucelabsSerializer
from rest_framework.response import Response


class SaucelabModelViewSet(ModelViewSet):
    serializer_class = SaucelabSerializer
    queryset = Sauselab.objects.all() 
    http_method_names = ['get', 'head', 'options', 'post']

    def create(self, request):
        pnam = request.data.get('project_name', None)
        url = request.data.get('url', None)
        if pnam and url:
            # do something here
            return Response({'success': 'Your success message'}, status=status.HTTP_200_OK)
        else:
            return Response({"error": "Your error message"}, status=status.HTTP_400_BAD_REQUEST)

序列化程序.py

from rest_framework.serializer import ModelSerializer
from appname.models import Saucelab

class SaucelabSerializer(ModelSerializer):
    class Meta:
        model = Saucelab
        fields = '__all__'

应用程序名称/模型.py

from django.db import models


class Saucelab(models.Model)
    project_name = models.CharField(max_length=255)
    url = models.URLField()

网址.py

from rest_framework.routers import DefaultRouter
from api import views


router = DefaultRouter()
router.register('your-endpoint-name', views.SaucelabModelViewSet, basename='your-url-name')

urlpatterns = []

urlpatterns += router.urls

这是一个非常基本的示例，您可以在其中创建一个名为saucelab的模型，其中包含您需要的两个字段，即。 project_name和url 。

我们创建了一个名为api的应用程序，其中包含两个不是自动生成的文件， serializers.py和urls.py 。 我们创建了最基本的模型序列化程序，并要求它序列化模型Saucelab所有字段。 然后我们创建一个简单的modelviewset ，它开箱即modelviewset为您提供CRUD功能。 如果您需要运行某些特定条件，您可以覆盖create方法，否则不要覆盖任何方法，只需使用适当的 HTTP 方法向端点发出请求。

这是您可以阅读的一些链接

https://www.django-rest-framework.org/api-guide/viewsets/#modelviewset https://www.django-rest-framework.org/api-guide/serializers/#modelserializer

生成随机令牌

from django.utils.crypto import get_random_string
print(get_random_string(length=25))

输出

u'rRXVe68NO7m3mHoBS488KdHaqQPD6Ofv'