flutter/apidash
1
0
Fork 0
mirror of https://github.com/foss42/apidash.git synced 2025-08-06 05:32:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

python http.client update

Browse Source
This commit is contained in:
Ashita Prasad
2024-03-13 11:09:13 +05:30
parent 990b1f5e2a
commit 62551f05f4
2 changed files with 261 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
lib/codegen/python/http_client.dart
View File

@ -1,8 +1,7 @@
import 'dart:io';
import 'dart:convert';
import 'package:jinja/jinja.dart' as jj;
import 'package:apidash/utils/utils.dart'
show getValidRequestUri, padMultilineString;
import 'package:apidash/utils/utils.dart' show getValidRequestUri;
import 'package:apidash/models/models.dart' show RequestModel;
import 'package:apidash/consts.dart';
@ -20,7 +19,6 @@ queryParams = {{params}}
queryParamsStr = '?' + urlencode(queryParams)
""";
int kParamsPadding = 14;
String kTemplateBody = """
@ -36,8 +34,6 @@ headers = {{headers}}
String kTemplateFormHeaderContentType = '''
multipart/form-data; boundary={{boundary}}''';
int kHeadersPadding = 10;
String kTemplateConnection = """
conn = http.client.HTTP{{isHttps}}Connection("{{authority}}")""";
@ -116,43 +112,38 @@ body = b'\r\n'.join(dataList)
hasQuery = true;
var templateParams = jj.Template(kTemplateParams);
var paramsString = kEncoder.convert(params);
paramsString = padMultilineString(paramsString, kParamsPadding);
result += templateParams.render({"params": paramsString});
}
}
var method = requestModel.method;
var requestBody = requestModel.requestBody;
if (kMethodsWithBody.contains(method) &&
requestBody != null &&
!requestModel.hasFormData) {
var contentLength = utf8.encode(requestBody).length;
if (contentLength > 0) {
if (requestModel.hasData) {
hasBody = true;
if (requestModel.hasJsonData || requestModel.hasTextData) {
var templateBody = jj.Template(kTemplateBody);
result += templateBody.render({"body": requestBody});
result += templateBody.render({"body": requestModel.requestBody});
}
}
var headersList = requestModel.enabledRequestHeaders;
if (headersList != null || hasBody) {
var headers = requestModel.enabledHeadersMap;
if (requestModel.hasFormData) {
var formHeaderTemplate =
jj.Template(kTemplateFormHeaderContentType);
headers[HttpHeaders.contentTypeHeader] = formHeaderTemplate.render({
"boundary": boundary,
});
}
if (headers.isNotEmpty || hasBody) {
hasHeaders = true;
if (hasBody && !requestModel.hasContentTypeHeader) {
if (requestModel.hasJsonData || requestModel.hasTextData) {
headers[HttpHeaders.contentTypeHeader] =
requestModel.requestBodyContentType.header;
} else if (requestModel.hasFormData) {
var formHeaderTemplate =
jj.Template(kTemplateFormHeaderContentType);
headers[HttpHeaders.contentTypeHeader] =
formHeaderTemplate.render({
"boundary": boundary,
});
}
}
var headersString = kEncoder.convert(headers);
headersString = padMultilineString(headersString, kHeadersPadding);
var templateHeaders = jj.Template(kTemplateHeaders);
result += templateHeaders.render({"headers": headersString});
}
@ -174,7 +165,7 @@ body = b'\r\n'.join(dataList)
var templateRequest = jj.Template(kTemplateRequest);
result += templateRequest.render({
"method": method.name.toUpperCase(),
"method": requestModel.method.name.toUpperCase(),
"path": uri.path,
"queryParamsStr": hasQuery ? " + queryParamsStr" : "",
});

241
test/codegen/python_http_client_codegen_test.dart
View File

@ -414,17 +414,38 @@ print(data.decode("utf-8"))
test('POST 4', () {
const expectedCode = r"""import http.client
body = r'''{
"text": "I LOVE Flutter"
}'''
import mimetypes
from codecs import encode
headers = {
"content-type": "text/plain"
"content-type": "multipart/form-data; boundary=b9826c20-773c-1f0c-814d-a1b3d90cd6b3"
}
def build_data_list(fields):
dataList = []
for field in fields:
name = field.get('name', '')
value = field.get('value', '')
type_ = field.get('type', 'text')
dataList.append(encode('--b9826c20-773c-1f0c-814d-a1b3d90cd6b3'))
if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain'))
dataList.append(encode(''))
dataList.append(encode(value))
elif type_ == 'file':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"; filename="{value}"'))
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode(''))
dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--b9826c20-773c-1f0c-814d-a1b3d90cd6b3--'))
dataList.append(encode(''))
return dataList
dataList = build_data_list([{"name":"text","value":"API","type":"text"},{"name":"sep","value":"|","type":"text"},{"name":"times","value":"3","type":"text"}])
body = b'\r\n'.join(dataList)
conn = http.client.HTTPSConnection("api.apidash.dev")
conn.request("POST", "/case/lower",
conn.request("POST", "/io/form",
body= body,
headers= headers)
@ -435,23 +456,46 @@ print(data.decode("utf-8"))
""";
expect(
codeGen.getCode(
CodegenLanguage.pythonHttpClient, requestModelPost4, "https"),
CodegenLanguage.pythonHttpClient, requestModelPost4, "https",
boundary: "b9826c20-773c-1f0c-814d-a1b3d90cd6b3"),
expectedCode);
}, skip: true);
});
test('POST 5', () {
const expectedCode = r"""import http.client
body = r'''{
"text": "I LOVE Flutter"
}'''
import mimetypes
from codecs import encode
headers = {
"content-type": "text/plain"
"User-Agent": "Test Agent",
"content-type": "multipart/form-data; boundary=929dc910-7714-1f0c-814d-a1b3d90cd6b3"
}
def build_data_list(fields):
dataList = []
for field in fields:
name = field.get('name', '')
value = field.get('value', '')
type_ = field.get('type', 'text')
dataList.append(encode('--929dc910-7714-1f0c-814d-a1b3d90cd6b3'))
if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain'))
dataList.append(encode(''))
dataList.append(encode(value))
elif type_ == 'file':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"; filename="{value}"'))
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode(''))
dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--929dc910-7714-1f0c-814d-a1b3d90cd6b3--'))
dataList.append(encode(''))
return dataList
dataList = build_data_list([{"name":"text","value":"API","type":"text"},{"name":"sep","value":"|","type":"text"},{"name":"times","value":"3","type":"text"}])
body = b'\r\n'.join(dataList)
conn = http.client.HTTPSConnection("api.apidash.dev")
conn.request("POST", "/case/lower",
conn.request("POST", "/io/form",
body= body,
headers= headers)
@ -462,23 +506,45 @@ print(data.decode("utf-8"))
""";
expect(
codeGen.getCode(
CodegenLanguage.pythonHttpClient, requestModelPost5, "https"),
CodegenLanguage.pythonHttpClient, requestModelPost5, "https",
boundary: "929dc910-7714-1f0c-814d-a1b3d90cd6b3"),
expectedCode);
}, skip: true);
});
test('POST 6', () {
const expectedCode = r"""import http.client
body = r'''{
"text": "I LOVE Flutter"
}'''
import mimetypes
from codecs import encode
headers = {
"content-type": "text/plain"
"content-type": "multipart/form-data; boundary=9b1374c0-76e0-1f0c-814d-a1b3d90cd6b3"
}
def build_data_list(fields):
dataList = []
for field in fields:
name = field.get('name', '')
value = field.get('value', '')
type_ = field.get('type', 'text')
dataList.append(encode('--9b1374c0-76e0-1f0c-814d-a1b3d90cd6b3'))
if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain'))
dataList.append(encode(''))
dataList.append(encode(value))
elif type_ == 'file':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"; filename="{value}"'))
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode(''))
dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--9b1374c0-76e0-1f0c-814d-a1b3d90cd6b3--'))
dataList.append(encode(''))
return dataList
dataList = build_data_list([{"name":"token","value":"xyz","type":"text"},{"name":"imfile","value":"/Documents/up/1.png","type":"file"}])
body = b'\r\n'.join(dataList)
conn = http.client.HTTPSConnection("api.apidash.dev")
conn.request("POST", "/case/lower",
conn.request("POST", "/io/img",
body= body,
headers= headers)
@ -489,23 +555,45 @@ print(data.decode("utf-8"))
""";
expect(
codeGen.getCode(
CodegenLanguage.pythonHttpClient, requestModelPost6, "https"),
CodegenLanguage.pythonHttpClient, requestModelPost6, "https",
boundary: "9b1374c0-76e0-1f0c-814d-a1b3d90cd6b3"),
expectedCode);
}, skip: true);
});
test('POST 7', () {
const expectedCode = r"""import http.client
body = r'''{
"text": "I LOVE Flutter"
}'''
import mimetypes
from codecs import encode
headers = {
"content-type": "text/plain"
"content-type": "multipart/form-data; boundary=defdf240-76b4-1f0c-814d-a1b3d90cd6b3"
}
def build_data_list(fields):
dataList = []
for field in fields:
name = field.get('name', '')
value = field.get('value', '')
type_ = field.get('type', 'text')
dataList.append(encode('--defdf240-76b4-1f0c-814d-a1b3d90cd6b3'))
if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain'))
dataList.append(encode(''))
dataList.append(encode(value))
elif type_ == 'file':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"; filename="{value}"'))
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode(''))
dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--defdf240-76b4-1f0c-814d-a1b3d90cd6b3--'))
dataList.append(encode(''))
return dataList
dataList = build_data_list([{"name":"token","value":"xyz","type":"text"},{"name":"imfile","value":"/Documents/up/1.png","type":"file"}])
body = b'\r\n'.join(dataList)
conn = http.client.HTTPSConnection("api.apidash.dev")
conn.request("POST", "/case/lower",
conn.request("POST", "/io/img",
body= body,
headers= headers)
@ -516,23 +604,52 @@ print(data.decode("utf-8"))
""";
expect(
codeGen.getCode(
CodegenLanguage.pythonHttpClient, requestModelPost7, "https"),
CodegenLanguage.pythonHttpClient, requestModelPost7, "https",
boundary: "defdf240-76b4-1f0c-814d-a1b3d90cd6b3"),
expectedCode);
}, skip: true);
});
test('POST 8', () {
const expectedCode = r"""import http.client
import mimetypes
from codecs import encode
from urllib.parse import urlencode
body = r'''{
"text": "I LOVE Flutter"
}'''
queryParams = {
"size": "2",
"len": "3"
}
queryParamsStr = '?' + urlencode(queryParams)
headers = {
"content-type": "text/plain"
"content-type": "multipart/form-data; boundary=a990b150-7683-1f0c-814d-a1b3d90cd6b3"
}
def build_data_list(fields):
dataList = []
for field in fields:
name = field.get('name', '')
value = field.get('value', '')
type_ = field.get('type', 'text')
dataList.append(encode('--a990b150-7683-1f0c-814d-a1b3d90cd6b3'))
if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain'))
dataList.append(encode(''))
dataList.append(encode(value))
elif type_ == 'file':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"; filename="{value}"'))
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode(''))
dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--a990b150-7683-1f0c-814d-a1b3d90cd6b3--'))
dataList.append(encode(''))
return dataList
dataList = build_data_list([{"name":"text","value":"API","type":"text"},{"name":"sep","value":"|","type":"text"},{"name":"times","value":"3","type":"text"}])
body = b'\r\n'.join(dataList)
conn = http.client.HTTPSConnection("api.apidash.dev")
conn.request("POST", "/case/lower",
conn.request("POST", "/io/form" + queryParamsStr,
body= body,
headers= headers)
@ -543,23 +660,54 @@ print(data.decode("utf-8"))
""";
expect(
codeGen.getCode(
CodegenLanguage.pythonHttpClient, requestModelPost8, "https"),
CodegenLanguage.pythonHttpClient, requestModelPost8, "https",
boundary: "a990b150-7683-1f0c-814d-a1b3d90cd6b3"),
expectedCode);
}, skip: true);
});
test('POST 9', () {
const expectedCode = r"""import http.client
import mimetypes
from codecs import encode
from urllib.parse import urlencode
body = r'''{
"text": "I LOVE Flutter"
}'''
queryParams = {
"size": "2",
"len": "3"
}
queryParamsStr = '?' + urlencode(queryParams)
headers = {
"content-type": "text/plain"
"User-Agent": "Test Agent",
"Keep-Alive": "true",
"content-type": "multipart/form-data; boundary=79088e00-75ec-1f0c-814d-a1b3d90cd6b3"
}
def build_data_list(fields):
dataList = []
for field in fields:
name = field.get('name', '')
value = field.get('value', '')
type_ = field.get('type', 'text')
dataList.append(encode('--79088e00-75ec-1f0c-814d-a1b3d90cd6b3'))
if type_ == 'text':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"'))
dataList.append(encode('Content-Type: text/plain'))
dataList.append(encode(''))
dataList.append(encode(value))
elif type_ == 'file':
dataList.append(encode(f'Content-Disposition: form-data; name="{name}"; filename="{value}"'))
dataList.append(encode(f'Content-Type: {mimetypes.guess_type(value)[0] or "application/octet-stream"}'))
dataList.append(encode(''))
dataList.append(open(value, 'rb').read())
dataList.append(encode(f'--79088e00-75ec-1f0c-814d-a1b3d90cd6b3--'))
dataList.append(encode(''))
return dataList
dataList = build_data_list([{"name":"token","value":"xyz","type":"text"},{"name":"imfile","value":"/Documents/up/1.png","type":"file"}])
body = b'\r\n'.join(dataList)
conn = http.client.HTTPSConnection("api.apidash.dev")
conn.request("POST", "/case/lower",
conn.request("POST", "/io/img" + queryParamsStr,
body= body,
headers= headers)
@ -570,10 +718,11 @@ print(data.decode("utf-8"))
""";
expect(
codeGen.getCode(
CodegenLanguage.pythonHttpClient, requestModelPost9, "https"),
CodegenLanguage.pythonHttpClient, requestModelPost9, "https",
boundary: "79088e00-75ec-1f0c-814d-a1b3d90cd6b3"),
expectedCode);
});
}, skip: true);
});
group('PUT Request', () {
test('PUT 1', () {