mirror of
https://github.com/foss42/apidash.git
synced 2025-06-28 03:19:40 +08:00
Fixed the bug
This commit is contained in:
@ -1,20 +1,26 @@
|
|||||||
import 'package:jinja/jinja.dart' as jj;
|
import 'package:jinja/jinja.dart' as jj;
|
||||||
import 'package:apidash/models/models.dart' show RequestModel;
|
import 'package:apidash/models/models.dart' show RequestModel;
|
||||||
|
import 'package:apidash/utils/utils.dart' show padMultilineString, rowsToMap;
|
||||||
|
import 'package:apidash/consts.dart';
|
||||||
|
|
||||||
class PythonRequestCodeGen {
|
class PythonRequestCodeGen {
|
||||||
|
int kHeadersPadding = 16;
|
||||||
String kPythonTemplate = '''
|
String kPythonTemplate = '''
|
||||||
import requests
|
import requests
|
||||||
|
import json
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
url = '{{url}}'
|
url = '{{url}}'
|
||||||
|
|
||||||
{{params}}
|
params = {{params}}
|
||||||
|
|
||||||
{{body}}
|
{{body}}
|
||||||
|
|
||||||
headers = {{headers}}
|
headers = {{headers}}
|
||||||
|
|
||||||
response = requests.{{method}}(url{{request_headers}}{{request_body}})
|
response = requests.{{method}}(
|
||||||
|
url{{request_params}}{{request_headers}}{{request_body}}
|
||||||
|
)
|
||||||
|
|
||||||
status_code = response.status_code
|
status_code = response.status_code
|
||||||
if 200 <= status_code < 300:
|
if 200 <= status_code < 300:
|
||||||
@ -31,6 +37,7 @@ main()
|
|||||||
try {
|
try {
|
||||||
bool hasHeaders = false;
|
bool hasHeaders = false;
|
||||||
bool hasBody = false;
|
bool hasBody = false;
|
||||||
|
bool hasParams = false;
|
||||||
|
|
||||||
String url = requestModel.url;
|
String url = requestModel.url;
|
||||||
if (!url.contains('://') && url.isNotEmpty) {
|
if (!url.contains('://') && url.isNotEmpty) {
|
||||||
@ -40,9 +47,9 @@ main()
|
|||||||
var paramsList = requestModel.requestParams;
|
var paramsList = requestModel.requestParams;
|
||||||
String params = '';
|
String params = '';
|
||||||
if (paramsList != null) {
|
if (paramsList != null) {
|
||||||
params = '';
|
hasParams = true;
|
||||||
for (var param in paramsList) {
|
for (var param in paramsList) {
|
||||||
params += '${param.k} = ${param.v}\n ';
|
params += '\n "${param.k}": "${param.v}",';
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -52,19 +59,31 @@ main()
|
|||||||
String requestBodyString = '';
|
String requestBodyString = '';
|
||||||
if (requestBody != null) {
|
if (requestBody != null) {
|
||||||
hasBody = true;
|
hasBody = true;
|
||||||
|
var bodyType = requestModel.requestBodyContentType;
|
||||||
|
if (bodyType == ContentType.text) {
|
||||||
requestBodyString = "data = '''$requestBody'''\n";
|
requestBodyString = "data = '''$requestBody'''\n";
|
||||||
|
} else if (bodyType == ContentType.json) {
|
||||||
|
int index = requestBody.lastIndexOf("}");
|
||||||
|
if (requestBody[index - 1] == ",") {
|
||||||
|
requestBody = requestBody.substring(0, index - 1) +
|
||||||
|
requestBody.substring(index);
|
||||||
|
}
|
||||||
|
requestBodyString =
|
||||||
|
"data = '''$requestBody''' \ndata = json.loads(data)\n";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var headersList = requestModel.requestHeaders;
|
var headersList = requestModel.requestHeaders;
|
||||||
String headers = '';
|
String headers = '';
|
||||||
if (headersList != null || hasBody) {
|
if (headersList != null || hasBody) {
|
||||||
headers = '';
|
var head = rowsToMap(requestModel.requestHeaders) ?? {};
|
||||||
for (var header in headersList ?? []) {
|
if (head.isNotEmpty || hasBody) {
|
||||||
headers += "'${header.k}': '${header.v}',\n ";
|
|
||||||
}
|
|
||||||
if (hasBody) {
|
if (hasBody) {
|
||||||
headers +=
|
head["content-type"] =
|
||||||
"'Content-Type': '${requestModel.requestBodyContentType}',\n ";
|
kContentTypeMap[requestModel.requestBodyContentType] ?? "";
|
||||||
|
}
|
||||||
|
headers = kEncoder.convert(head);
|
||||||
|
headers = padMultilineString(headers, kHeadersPadding);
|
||||||
}
|
}
|
||||||
hasHeaders = headers.isNotEmpty;
|
hasHeaders = headers.isNotEmpty;
|
||||||
}
|
}
|
||||||
@ -72,12 +91,13 @@ main()
|
|||||||
var template = jj.Template(kPythonTemplate);
|
var template = jj.Template(kPythonTemplate);
|
||||||
var pythonCode = template.render({
|
var pythonCode = template.render({
|
||||||
'url': url,
|
'url': url,
|
||||||
'params': params,
|
'params': '{$params \n }',
|
||||||
'body': requestBodyString,
|
'body': requestBodyString,
|
||||||
'headers': '{\n $headers}',
|
'headers': headers,
|
||||||
'method': method,
|
'method': method,
|
||||||
'request_headers': hasHeaders ? 'headers=headers,\n ' : '',
|
'request_params': hasParams ? ', params=params' : '',
|
||||||
'request_body': hasBody ? 'data=data,\n ' : '',
|
'request_headers': hasHeaders ? ', headers=headers' : '',
|
||||||
|
'request_body': hasBody ? ', data=data' : '',
|
||||||
});
|
});
|
||||||
|
|
||||||
return pythonCode;
|
return pythonCode;
|
||||||
|
Reference in New Issue
Block a user