feat: python requests codegen

This commit is contained in:
Ashita Prasad
2023-09-30 22:02:50 +05:30
parent fd30817867
commit d3b6b701fa

View File

@ -1,102 +1,155 @@
import 'dart:io';
import 'dart:convert';
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/utils/utils.dart' show padMultilineString, rowsToMap;
import 'package:apidash/consts.dart'; import 'package:apidash/consts.dart';
import 'package:apidash/utils/utils.dart'
show getValidRequestUri, padMultilineString, rowsToMap;
import 'package:apidash/models/models.dart' show RequestModel;
class PythonRequestCodeGen { class PythonRequestCodeGen {
int kHeadersPadding = 16; final String kTemplateStart = """import requests
String kPythonTemplate = '''
import requests
def main(): url = '{{url}}'
url = '{{url}}'{{params}}{{body}}{{headers}}
response = requests.{{method}}( """;
url{{request_params}}{{request_headers}}{{request_body}}
)
status_code = response.status_code String kTemplateParams = """
if 200 <= status_code < 300:
print('Status Code:', status_code) params = {{params}}
""";
int kParamsPadding = 9;
String kTemplateBody = """
payload = r'''{{body}}'''
""";
String kTemplateJson = """
payload = {{body}}
""";
String kTemplateHeaders = """
headers = {{headers}}
""";
int kHeadersPadding = 10;
String kTemplateRequest = """
response = requests.{{method}}(url
""";
String kStringRequestParams = """, params=params""";
String kStringRequestBody = """, data=payload""";
String kStringRequestJson = """, json=payload""";
String kStringRequestHeaders = """, headers=headers""";
final String kStringRequestEnd = """)
print('Status Code:', response.status_code)
print('Response Body:', response.text) print('Response Body:', response.text)
else: """;
print('Error Status Code:', status_code)
print('Error Response Body:', response.reason)
main() String? getCode(
'''; RequestModel requestModel,
String defaultUriScheme,
String? getCode(RequestModel requestModel, String defaultUriScheme) { ) {
try { try {
String result = "";
bool hasParams = false;
bool hasHeaders = false; bool hasHeaders = false;
bool hasBody = false; bool hasBody = false;
bool hasParams = false; bool hasJsonBody = false;
String url = requestModel.url; String url = requestModel.url;
if (!url.contains('://') && url.isNotEmpty) { if (!url.contains("://") && url.isNotEmpty) {
url = '$defaultUriScheme://$url'; url = "$defaultUriScheme://$url";
} }
var paramsList = requestModel.requestParams; var rec = getValidRequestUri(url, requestModel.requestParams);
String params = ''; Uri? uri = rec.$1;
if (paramsList != null) { if (uri != null) {
for (var param in paramsList) { var templateStartUrl = jj.Template(kTemplateStart);
if (param.name.isNotEmpty) { result += templateStartUrl
.render({"url": "${uri.scheme}://${uri.authority}${uri.path}"});
if (uri.hasQuery) {
hasParams = true; hasParams = true;
params += '\n "${param.name}": "${param.value}",'; var params = uri.queryParameters;
} if (params.isNotEmpty) {
var templateParams = jj.Template(kTemplateParams);
var paramsString = kEncoder.convert(params);
paramsString = padMultilineString(paramsString, kParamsPadding);
result += templateParams.render({"params": paramsString});
} }
} }
var method = requestModel.method.name.toLowerCase(); var method = requestModel.method;
var requestBody = requestModel.requestBody; var requestBody = requestModel.requestBody;
String requestBodyString = ''; if (kMethodsWithBody.contains(method) && requestBody != null) {
if (requestBody != null && requestBody.isNotEmpty) { var contentLength = utf8.encode(requestBody).length;
if (contentLength > 0) {
if (requestModel.requestBodyContentType == ContentType.json) {
hasJsonBody = true;
var templateBody = jj.Template(kTemplateJson);
result += templateBody.render({"body": requestBody});
} else {
hasBody = true; hasBody = true;
var bodyType = requestModel.requestBodyContentType; var templateBody = jj.Template(kTemplateBody);
if (bodyType == ContentType.json) { result += templateBody.render({"body": requestBody});
int index = requestBody.lastIndexOf("}");
index--;
while (requestBody[index] == " " || requestBody[index] == "\n") {
index--;
}
if (requestBody[index] == ",") {
requestBody = requestBody.substring(0, index) +
requestBody.substring(index + 1);
} }
} }
requestBodyString = "data = '''$requestBody'''";
} }
var headersList = requestModel.requestHeaders; var headersList = requestModel.requestHeaders;
String headers = '';
if (headersList != null || hasBody) { if (headersList != null || hasBody) {
var head = rowsToMap(requestModel.requestHeaders) ?? {}; var headers = rowsToMap(requestModel.requestHeaders) ?? {};
if (head.isNotEmpty || hasBody) { if (headers.isNotEmpty || hasBody) {
hasHeaders = true;
if (hasBody) { if (hasBody) {
head["content-type"] = headers[HttpHeaders.contentTypeHeader] =
kContentTypeMap[requestModel.requestBodyContentType] ?? ""; kContentTypeMap[requestModel.requestBodyContentType] ?? "";
} }
headers = kEncoder.convert(head); var headersString = kEncoder.convert(headers);
headers = padMultilineString(headers, kHeadersPadding); headersString = padMultilineString(headersString, kHeadersPadding);
var templateHeaders = jj.Template(kTemplateHeaders);
result += templateHeaders.render({"headers": headersString});
} }
hasHeaders = headers.isNotEmpty;
} }
var template = jj.Template(kPythonTemplate); var templateRequest = jj.Template(kTemplateRequest);
var pythonCode = template.render({ result += templateRequest.render({
'url': url, "method": method.name.toLowerCase(),
'params': hasParams ? '\n\n params = {$params \n }' : '',
'body': hasBody ? '\n\n $requestBodyString' : '',
'headers': hasHeaders ? '\n\n headers = $headers' : '',
'method': method,
'request_params': hasParams ? ', params=params' : '',
'request_headers': hasHeaders ? ', headers=headers' : '',
'request_body': hasBody ? ', data=data' : '',
}); });
return pythonCode; if (hasParams) {
result += kStringRequestParams;
}
if (hasBody) {
result += kStringRequestBody;
}
if (hasJsonBody) {
result += kStringRequestJson;
}
if (hasHeaders) {
result += kStringRequestHeaders;
}
result += kStringRequestEnd;
}
return result;
} catch (e) { } catch (e) {
return null; return null;
} }