Django 模拟登陆,验证接口可用行

import requests

LOGIN_URL = "http://mydjangosite.com/accounts/login/"
ENDPOINT_URL = 'http://mydjangosite.com/myendpoint/'

'''
Create a session.
A session will automatically store the cookies that Django
sends back to you, like the csrf token and a the session id. You
could do it without the session, but then you'd have to save off the
cookies manually and remember to pass them along with subsequent
requests.
'''
client = requests.session()
client.get(LOGIN_URL)

import requests

LOGIN_URL = "http://mydjangosite.com/accounts/login/"
ENDPOINT_URL = 'http://mydjangosite.com/myendpoint/'

'''
Create a session.
A session will automatically store the cookies that Django
sends back to you, like the csrf token and a the session id. You
could do it without the session, but then you'd have to save off the
cookies manually and remember to pass them along with subsequent
requests.
'''
client = requests.session()
client.get(LOGIN_URL)
# Django would like the csrf token passed with the data, so we do need to save it off seperately.
csrftoken = client.cookies['csrftoken']

'''
Log in.
'''
login_data = {login:"somepersonsname", password:"supergreatpassword", csrfmiddlewaretoken:csrftoken}
r1 = client.post(LOGIN_URL, data=login_data)
# For some reason, we are issued a new csrf token after logging in, so update your local copy.
csrftoken = client.cookies['csrftoken']

'''
Post some data to your login-only API endpoint.
'''
payload = {'somedata':'asdf', 'someotherdata':'1235', 'csrfmiddlewaretoken':csrftoken}
# We use client.post (not requests.post) so that we pass on the cookies that our session stored.
r2 = client.post(ENDPOINT_URL, data=payload)


# Django would like the csrf token passed with the data, so we do need to save it off seperately.
csrftoken = client.cookies['csrftoken']

'''
Log in.
'''
login_data = {login:"somepersonsname", password:"supergreatpassword", csrfmiddlewaretoken:csrftoken}
r1 = client.post(LOGIN_URL, data=login_data)
# For some reason, we are issued a new csrf token after logging in, so update your local copy.
csrftoken = client.cookies['csrftoken']

'''
Post some data to your login-only API endpoint.
'''
payload = {'somedata':'asdf', 'someotherdata':'1235', 'csrfmiddlewaretoken':csrftoken}
# We use client.post (not requests.post) so that we pass on the cookies that our session stored

headers = {"Content-Type": "application/json", "accept": "application/json",
"X-CSRFToken": csrftoken}

result = client.post(endpoint_url, json=payload, headers=headers)

 

 

posted @ 2020-05-19 14:27  进击的pythoner  阅读(328)  评论(0编辑  收藏  举报