运维平台中RESTful的Token认证

我们来写一个新的view Token_api,内容如下：

from OpsManage.models import *

from django.contrib import auth

from django.http import HttpResponse, JsonResponsefrom django.views.decorators.csrf import csrf_exempt

from rest_framework.authtoken.models import Token

from django.utils.datastructures import SortedDict

@csrf_exempt

# @api_view(['POST'])

def gettoken(request):

if request.method == 'POST':

username = request.POST.get('username', '')

password = request.POST.get('password', '')

user = auth.authenticate(username=username, password=password)

if user is not None:

auth.login(request, user)

token = Token.objects.get(user=User.objects.get(username=username)).key

return JsonResponse({'token': token}, safe=True)

else:

return JsonResponse({'error': 'valid username or password'})

else:

return JsonResponse({'error': 'No GET Method!'})

然后写一个cmdb_api.py，内容如下，这个逻辑是根据输入的参数来得到相应的数据库信息列表，比如输入db_type=MySQL则输出MySQL信息，大概是这样的信息。

@api_view(['GET', 'POST' ])

@permission_classes((permissions.IsAuthenticated,))

def cmdb_server_list(request,format=None):

"""

List all order, or create a server assets order.

"""

if request.method == 'GET':

db_type = request.GET.get("db_type")

if db_type != None:

snippets = Cmdb_server.objects.filter(server_status=1).filter(server_db_type=db_type)

serializer = Cmdb_serverSerializer(snippets, many=True)

else:

snippets = Cmdb_server.objects.filter(server_status=1)[0:10]

serializer = Cmdb_serverSerializer(snippets, many=True)

return Response(serializer.data)

elif request.method == 'POST':

serializer = Cmdb_serverSerializer(data=request.data)

if serializer.is_valid():

serializer.save()

return Response(serializer.data, status=status.HTTP_201_CREATED)

return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

到了这里，其实cmdb_api和token_api还没有关联起来，调用的方式我们可以直接使用rest_framework来得到，当然这个是页面层面的信息，要得到更加正式的信息，我们需要使用requests，即独立客户端的调用方式,我们使用了requests的方式调用gettoken得到token_api.gettoken的结果，getresponse是一个相对通用的方法，可以传入url得到返回数据。

import requests

import json

def gettoken(username,password):

data = {'username': username, 'password': password}

print(data)

s = requests.post('http://xxxxx:8000/api/api-token-auth/', data, verify=False)

print(s.text)

token = eval(s.text).get('token')

return token

def getresponse(url, token):

headers = {"Authorization": "Token %s" % token}

ss = requests.Session()

r = ss.get(url, headers=headers, verify=False)

res = json.loads(r.text)

return res

token = gettoken('xxxx', 'xxxxx')

print(token)

rs = getresponse('http://xxxxx:8000/api/cmdb_list/',token)

print(rs)

我们模拟测试的话，可以使用python api.py的方式,得到的数据是这样的形式：

# python api.py

{"token": "2565018fb332f20f2885923a0e2313b3b769e7d9"}

2565018fb332f20f2885923a0e2313b3b769e7d9

[{u'update_date': None, u'create_date': u'2018-02-01T17:29:32', u'server_status': u'1', u'server_db_role': u'Master', u'server_db_type': u'MySQL', u'server_ip_addr': u'xxxxxxxxxxx', u'id': 168}, {u'update_date': None, u'create_date': u'2018-02-01T17:29:32', u'server_status': u'1', u'server_db_role': u'Master', u'server_db_type': u'MySQL', u'server_ip_addr': u'xxxxxxxxxxxxx', u'id': 169}...

到了这里，你就可以独立的调用RESTful接口，使用token认证得到数据。