-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathpython3.js
More file actions
121 lines (102 loc) · 3.13 KB
/
python3.js
File metadata and controls
121 lines (102 loc) · 3.13 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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/**
* @description
* HTTP code snippet generator for native Python3.
*
* @author
* @montanaflynn
*
* for any questions or issues regarding the generated code snippet, please open an issue mentioning the author.
*/
'use strict'
const CodeBuilder = require('../../helpers/code-builder')
const helpers = require('../../helpers/headers')
module.exports = function (source, options) {
const code = new CodeBuilder()
// Start Request
code.push('import http.client')
if (options.insecureSkipVerify) {
code.push('import ssl')
}
const mayBeGzipped = helpers.hasHeader(source.allHeaders, 'accept-encoding') &&
helpers.getHeader(source.allHeaders, 'accept-encoding').includes('gzip')
if (mayBeGzipped) {
code.push('import gzip')
}
code.blank()
// Check which protocol to be used for the client connection
const protocol = source.uriObj.protocol
if (protocol === 'https:') {
if (options.insecureSkipVerify) {
code.push(
'conn = http.client.HTTPSConnection("%s", context = ssl._create_unverified_context())',
source.uriObj.host
).blank()
} else {
code.push('conn = http.client.HTTPSConnection("%s")', source.uriObj.host)
.blank()
}
} else {
code.push('conn = http.client.HTTPConnection("%s")', source.uriObj.host)
.blank()
}
// Create payload string if it exists
const payload = JSON.stringify(source.postData.text)
if (payload) {
code.push('payload = %s', payload)
.blank()
}
// Create Headers
const headers = source.allHeaders
const headerCount = Object.keys(headers).length
if (headerCount === 1) {
for (const header in headers) {
code.push('headers = { "%s": "%qd" }', header, headers[header])
.blank()
}
} else if (headerCount > 1) {
let count = 1
code.push('headers = {')
for (const header in headers) {
if (count++ !== headerCount) {
code.push(' "%s": "%qd",', header, headers[header])
} else {
code.push(' "%s": "%qd"', header, headers[header])
}
}
code.push('}')
.blank()
}
// Make Request
const method = source.method
const path = source.uriObj.path
if (payload && headerCount) {
code.push('conn.request("%s", "%s", payload, headers)', method, path)
} else if (payload && !headerCount) {
code.push('conn.request("%s", "%s", payload)', method, path)
} else if (!payload && headerCount) {
code.push('conn.request("%s", "%s", headers=headers)', method, path)
} else {
code.push('conn.request("%s", "%s")', method, path)
}
// Get Response
code.blank()
.push('res = conn.getresponse()')
.push('data = res.read()')
.blank()
// Decode response
if (mayBeGzipped) {
code.push("if res.headers['content-encoding'] == 'gzip':")
code.push(' print(gzip.decompress(data).decode("utf-8"))')
code.push('else:')
code.push(' print(data.decode("utf-8"))')
} else {
code.push('print(data.decode("utf-8"))')
}
return code.join()
}
module.exports.info = {
key: 'python3',
title: 'http.client',
link: 'https://docs.python.org/3/library/http.client.html',
description: 'Python3 HTTP Client'
}