介紹
本教程將會通過創建一個簡單的粘貼代碼項目,突出顯示WebAPI,過程中,會介紹組成REST框架的各種組件,讓你全面了解如何配合使用。
環境設置
建議使用virtualenv建立新的虛擬環境,確保本項目所依賴的任何包和其他環境保持隔離
virtualenv env
source env/bin/activate
現在在新的虛擬環境中安裝必須的包
pip install django
pip install djangorestframework
pip install pygments #代碼高亮使用
創建項目
我們將創建名為tutorial的項目,在項目中創建名為snippets的app
django-admin.py startproject tutorial
cd tutorial
python manage.py startapp snippets
最后我們需要在settings.py的INSTALLED_APPS配置里注冊我們的snippetsapp和rest_frameworkapp
INSTALLED_APPS = [
...
'rest_framework',
'snippets.apps.SnippetsConfig',
]
注意:如果使用的django版本低於1.9,直接使用snippets 來代替snippets.apps.SnippetsConfig即可
創建model
下面創建snippets所使用的數據庫表用來存儲代碼
from django.db import models
from pygments.lexers import get_all_lexers
from pygments.styles import get_all_styles
LEXERS = [item for item in get_all_lexers() if item[1]]
LANGUAGE_CHOICES = sorted([(item[1][0], item[0]) for item in LEXERS])
STYLE_CHOICES = sorted((item, item) for item in get_all_styles())
class Snippet(models.Model):
created = models.DateTimeField(auto_now_add=True)
title = models.CharField(max_length=100, blank=True, default='')
code = models.TextField()
linenos = models.BooleanField(default=False)
language = models.CharField(choices=LANGUAGE_CHOICES, default='python', max_length=100)
style = models.CharField(choices=STYLE_CHOICES, default='friendly', max_length=100)
class Meta:
ordering = ('created',)
然后我們需要在數據庫中創建數據庫表,生成表結構
python manage.py makemigrations snippets
python manage.py migrate
創建序列化的類
首先我們需要為api提供序列化和反序列化的方法,將snippet實例轉為json等方式呈現數據,我們可以使用Serializer達到這一目的, Serializer和django forms十分相似. 我們在snippetapp下新建serializers.py文件
from rest_framework import serializers
from snippets.models import Snippet, LANGUAGE_CHOICES, STYLE_CHOICES
class SnippetSerializer(serializers.Serializer):
id = serializers.IntegerField(read_only=True)
title = serializers.CharField(required=False, allow_blank=True, max_length=100)
code = serializers.CharField(style={'base_template': 'textarea.html'})
linenos = serializers.BooleanField(required=False)
language = serializers.ChoiceField(choices=LANGUAGE_CHOICES, default='python')
style = serializers.ChoiceField(choices=STYLE_CHOICES, default='friendly')
def create(self, validated_data):
"""
新建Snippet實例
"""
return Snippet.objects.create(**validated_data)
def update(self, instance, validated_data):
"""
更新Snippet示例
"""
instance.title = validated_data.get('title', instance.title)
instance.code = validated_data.get('code', instance.code)
instance.linenos = validated_data.get('linenos', instance.linenos)
instance.language = validated_data.get('language', instance.language)
instance.style = validated_data.get('style', instance.style)
instance.save()
return instance
serializer的首先定義了序列化和反序列化的字段,其次create()和update()方法定義了當調用serializer.save()時如何對實例進行創建或修改
serializer和django的form非常類似,而且包括一些類似的驗證字段的方法,如required,max_lengthanddefault.
使用Serializers
我們使用django的shell 環境來簡單的測試Serializers
創建兩個snippet實例
python3 manage.py shell
>>> from snippets.models import Snippet
>>> from snippets.serializers import SnippetSerializer
>>> from rest_framework.renderers import JSONRenderer
>>> from rest_framework.parsers import JSONParser
>>> snippet = Snippet(code='foo = "bar"\n')
>>> snippet.save()
>>> snippet2 = Snippet(code='print "hello, world"\n')
>>> snippet2.save()
序列化其中一個實例
>>> serializer = SnippetSerializer(snippet2)
>>> serializer.data
{'linenos': False, 'id': 2, 'code': 'print "hello, world"\n', 'language': 'python', 'style': 'friendly', 'title': ''}
我們已經將snippet2序列化為一個python的基本數據類型,現在我們生成json數據
>>> content = JSONRenderer().render(serializer.data)
>>> content
b'{"id":2,"title":"","code":"print \\"hello, world\\"\\n","linenos":false,"language":"python","style":"friendly"}'
反序列化更簡單,首先將stream轉為python基本類型
>>> from django.utils.six import BytesIO
>>> stream = BytesIO(content)
>>> data = JSONParser().parse(stream)
>>> data
{'linenos': False, 'id': 2, 'code': 'print "hello, world"\n', 'language': 'python', 'style': 'friendly', 'title': ''}
然后,我們將其轉換為snippet實例
>>> serializer = SnippetSerializer(data=data)
>>> serializer.is_valid()
True
>>> serializer.validated_data
OrderedDict([('title', ''), ('code', 'print "hello, world"'), ('linenos', False), ('language', 'python'), ('style', 'friendly')])
>>> serializer.save()
<Snippet: Snippet object>
我們也可以序列化querysets對象來代替model instances,因此在序列化的時候需要添加many=True參數
>>> serializer = SnippetSerializer(Snippet.objects.all(), many=True)
>>> serializer.data
[OrderedDict([('id', 1), ('title', ''), ('code', 'foo = "bar"\n'), ('linenos', False), ('language', 'python'), ('style', 'friendly')]), OrderedDict([('id', 2), ('title', ''), ('code', 'print "hello, world"\n'), ('linenos', False), ('language', 'python'), ('style', 'friendly')]), OrderedDict([('id', 3), ('title', ''), ('code', 'print "hello, world"'), ('linenos', False), ('language', 'python'), ('style', 'friendly')])]
使用ModelSerializers
我們的SnippetSerializer類很多信息和Snippet model中是重復的,為了保證我們代碼的簡潔,減少重復代碼,類似於django的提供的form和modelform,REST framework也提供了Serializerand ModelSerializer
下面使用ModelSerializer重構我們的serializer,重新打開serializers.py,使用下面代碼替換掉SnippetSerializer類
class SnippetSerializer(serializers.ModelSerializer):
class Meta:
model = Snippet
fields = ('id', 'title', 'code', 'linenos', 'language', 'style')
serializers有一個不錯的屬性是,可以通過print函數來打印它 表示檢查序列化實例中的所有字段
>>> from snippets.serializers import SnippetSerializer
>>> serializer = SnippetSerializer()
>>> print(repr(serializer))
重要的是要記住 ModelSerializer 不做任何格外的配置,它只是創建序列化類的快捷方式:
- 根據model里的字段自動定義字段集
- 簡單的實現
create()andupdate()方法
在django 視圖中使用我們的Serializer
讓我們看看我們如何使用我們新的Serializer類編寫一些API視圖。目前我們不使用任何REST框架的其他功能,我們只是將視圖寫為常規Django視圖。
我們將開始通過創建一個HttpResponse子類用來返回任何形式的數據,如json;
編輯views.py
from django.http import HttpResponse
from django.views.decorators.csrf import csrf_exempt
from rest_framework.renderers import JSONRenderer
from rest_framework.parsers import JSONParser
from snippets.models import Snippet
from snippets.serializers import SnippetSerializer
class JSONResponse(HttpResponse):
"""
An HttpResponse that renders its content into JSON.
"""
def __init__(self, data, **kwargs):
content = JSONRenderer().render(data)
kwargs['content_type'] = 'application/json'
super(JSONResponse, self).__init__(content, **kwargs)
@csrf_exempt
def snippet_list(request):
"""
列出所有的實例,或創建一個新的實例.
"""
if request.method == 'GET':
snippets = Snippet.objects.all()
serializer = SnippetSerializer(snippets, many=True)
return JSONResponse(serializer.data)
elif request.method == 'POST':
data = JSONParser().parse(request)
serializer = SnippetSerializer(data=data)
if serializer.is_valid():
serializer.save()
return JSONResponse(serializer.data, status=201)
return JSONResponse(serializer.errors, status=400)
注意我們在post測試的時候無法獲取到Django的csrf token,所以使用@csrf_exempt將此視圖排除,不在檢查csrf token
我們將還需要一個與單個代碼段對應的視圖,並且可以用來檢索、更新和刪除代碼段
@csrf_exempt
def snippet_detail(request, pk):
"""
Retrieve, update or delete a code snippet.
"""
try:
snippet = Snippet.objects.get(pk=pk)
except Snippet.DoesNotExist:
return HttpResponse(status=404)
if request.method == 'GET':
serializer = SnippetSerializer(snippet)
return JSONResponse(serializer.data)
elif request.method == 'PUT':
data = JSONParser().parse(request)
serializer = SnippetSerializer(snippet, data=data)
if serializer.is_valid():
serializer.save()
return JSONResponse(serializer.data)
return JSONResponse(serializer.errors, status=400)
elif request.method == 'DELETE':
snippet.delete()
return HttpResponse(status=204)
最后,我們需要配置url信息,在snippetapp下,新建urls.py
from django.conf.urls import url
from snippets import views
urlpatterns = [
url(r'^snippets/$', views.snippet_list),
url(r'^snippets/(?P<pk>[0-9]+)/$', views.snippet_detail),
]
我們也需要在tutorial項目的urls.py下將snippetapp的url配置加載進去
from django.conf.urls import url, include
urlpatterns = [
url(r'^', include('snippets.urls')),
]
嘗試測試我們的web api
啟動我們的django server
python manage.py runserver
Performing system checks...
System check identified no issues (0 silenced).
February 04, 2017 - 08:38:55
Django version 1.10, using settings 'tutorial.settings'
Starting development server at http://127.0.0.1:8000/
Quit the server with CONTROL-C.
最后使用curl 進行測試
curl -s http://127.0.0.1:8000/snippets/ | jq .
[
{
"id": 1,
"title": "",
"code": "foo = \"bar\"\n",
"linenos": false,
"language": "python",
"style": "friendly"
},
{
"id": 2,
"title": "",
"code": "print \"hello, world\"\n",
"linenos": false,
"language": "python",
"style": "friendly"
},
{
"id": 3,
"title": "",
"code": "print \"hello, world\"",
"linenos": false,
"language": "python",
"style": "friendly"
}
]
也可以指定id進行查詢
curl -s http://127.0.0.1:8000/snippets/1/ | jq .
{
"id": 1,
"title": "",
"code": "foo = \"bar\"\n",
"linenos": false,
"language": "python",
"style": "friendly"
}
