forked from postmanlabs/httpsnippet-fsless
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequests.js
More file actions
104 lines (80 loc) · 2.16 KB
/
requests.js
File metadata and controls
104 lines (80 loc) · 2.16 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/**
* @description
* HTTP code snippet generator for Python using Requests
*
* @author
* @montanaflynn
*
* for any questions or issues regarding the generated code snippet, please open an issue mentioning the author.
*/
'use strict'
var util = require('util')
var CodeBuilder = require('../../helpers/code-builder')
module.exports = function (source, options) {
// Start snippet
var code = new CodeBuilder(' ')
// Import requests
code.push('import requests')
.blank()
// Set URL
code.push('url = "%s"', source.url)
.blank()
// Construct query string
if (source.queryString.length) {
var qs = 'querystring = ' + JSON.stringify(source.queryObj)
code.push(qs)
.blank()
}
// Construct payload
var payload = JSON.stringify(source.postData.text)
if (payload) {
code.push('payload = %s', payload)
}
// Construct headers
var header
var headers = source.allHeaders
var headerCount = Object.keys(headers).length
if (headerCount === 1) {
for (header in headers) {
code.push('headers = {\'%s\': \'%s\'}', header, headers[header])
.blank()
}
} else if (headerCount > 1) {
var count = 1
code.push('headers = {')
for (header in headers) {
if (count++ !== headerCount) {
code.push(1, '\'%s\': "%s",', header, headers[header])
} else {
code.push(1, '\'%s\': "%s"', header, headers[header])
}
}
code.push(1, '}')
.blank()
}
// Construct request
var method = source.method
var request = util.format('response = requests.request("%s", url', method)
if (payload) {
request += ', data=payload'
}
if (headerCount > 0) {
request += ', headers=headers'
}
if (qs) {
request += ', params=querystring'
}
request += ')'
code.push(request)
.blank()
// Print response
.push('print(response.text)')
return code.join()
}
module.exports.info = {
key: 'requests',
title: 'Requests',
link: 'http://docs.python-requests.org/en/latest/api/#requests.request',
description: 'Requests HTTP library'
}
// response = requests.request("POST", url, data=payload, headers=headers, params=querystring)