apidash/lib/codegen/python/pkg_request.dart

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';

class PythonRequestCodeGen {
  int kHeadersPadding = 16;
  String kPythonTemplate = '''
import requests

def main():
    url = '{{url}}'{{params}}{{body}}{{headers}}

    response = requests.{{method}}(
        url{{request_params}}{{request_headers}}{{request_body}}
    )

    status_code = response.status_code
    if 200 <= status_code < 300:
        print('Status Code:', status_code)
        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) {
    try {
      bool hasHeaders = false;
      bool hasBody = false;
      bool hasParams = false;

      String url = requestModel.url;
      if (!url.contains('://') && url.isNotEmpty) {
        url = '$defaultUriScheme://$url';
      }

      var paramsList = requestModel.requestParams;
      String params = '';
      if (paramsList != null) {
        for (var param in paramsList) {
          if (param.name.isNotEmpty) {
            hasParams = true;
            params += '\n        "${param.name}": "${param.value}",';
          }
        }
      }

      var method = requestModel.method.name.toLowerCase();

      var requestBody = requestModel.requestBody;
      String requestBodyString = '';
      if (requestBody != null && requestBody.isNotEmpty) {
        hasBody = true;
        var bodyType = requestModel.requestBodyContentType;
        if (bodyType == ContentType.json) {
          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;
      String headers = '';
      if (headersList != null || hasBody) {
        var head = rowsToMap(requestModel.requestHeaders) ?? {};
        if (head.isNotEmpty || hasBody) {
          if (hasBody) {
            head["content-type"] =
                kContentTypeMap[requestModel.requestBodyContentType] ?? "";
          }
          headers = kEncoder.convert(head);
          headers = padMultilineString(headers, kHeadersPadding);
        }
        hasHeaders = headers.isNotEmpty;
      }

      var template = jj.Template(kPythonTemplate);
      var pythonCode = template.render({
        'url': url,
        '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;
    } catch (e) {
      return null;
    }
  }
}