Merge branch 'main' of https://github.com/Mixel2004/api-dash into Mixel2004-main

2025-05-22 16:57:07 +08:00 · 2023-08-29 05:46:24 +05:30
parent 8992c486cb cb23555394
commit 7d8b493894
4 changed files with 288 additions and 1 deletions
--- a/lib/codegen/python/pkg_request.dart
+++ b/lib/codegen/python/pkg_request.dart
@ -0,0 +1,104 @@
+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.k.isNotEmpty) {
+            hasParams = true;
+            params += '\n        "${param.k}": "${param.v}",';
+          }
+        }
+      }
+
+      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;
+    }
+  }
+}