Update python requests codes

lib/codegen/codegen.dart

@@ -1,10 +1,9 @@
-import 'package:apidash/codegen/kotlin/pkg_okhttp.dart';
-import 'python/pkg_http_client.dart';
-import 'package:apidash/codegen/python/pkg_request.dart';
-import 'package:apidash/consts.dart';
-
 import 'package:apidash/models/models.dart' show RequestModel;
+import 'package:apidash/consts.dart';
 import 'dart/pkg_http.dart';
+import 'kotlin/pkg_okhttp.dart';
+import 'python/pkg_http_client.dart';
+import 'python/pkg_requests.dart';
 
 class Codegen {
   String? getCode(
@@ -20,7 +19,7 @@ class Codegen {
       case CodegenLanguage.pythonHttpClient:
         return PythonHttpClient().getCode(requestModel, defaultUriScheme);
       case CodegenLanguage.pythonRequests:
-        return PythonRequestCodeGen().getCode(requestModel, defaultUriScheme);
+        return PythonRequestsCodeGen().getCode(requestModel, defaultUriScheme);
       default:
         throw ArgumentError('Invalid codegenLanguage');
     }

lib/codegen/python/pkg_requests.dart

@@ -6,7 +6,7 @@ import 'package:apidash/utils/utils.dart'
     show getValidRequestUri, padMultilineString, rowsToMap;
 import 'package:apidash/models/models.dart' show RequestModel;
 
-class PythonRequestCodeGen {
+class PythonRequestsCodeGen {
   final String kTemplateStart = """import requests
 
 url = '{{url}}'

test/codegen/python_request_codegen_test.dart

@@ -1,179 +0,0 @@
-import 'package:apidash/codegen/python/pkg_request.dart';
-import 'package:apidash/models/models.dart';
-import 'package:test/test.dart';
-import 'package:apidash/consts.dart';
-
-void main() {
-  group('PythonRequestCodeGen', () {
-    final pythonRequestCodeGen = PythonRequestCodeGen();
-
-    test('getCode returns valid code for GET request', () {
-      const requestModel = RequestModel(
-        url: 'https://jsonplaceholder.typicode.com/todos/1',
-        method: HTTPVerb.get,
-        id: '',
-      );
-      const expectedCode = """import requests
-
-def main():
-    url = 'https://jsonplaceholder.typicode.com/todos/1'
-
-    response = requests.get(
-        url
-    )
-
-    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()""";
-      expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
-    });
-
-    test('getCode returns valid code for POST request', () {
-      const requestModel = RequestModel(
-        url: 'https://jsonplaceholder.typicode.com/posts',
-        method: HTTPVerb.post,
-        requestBody: '{"title": "foo","body": "bar","userId": 1}',
-        requestBodyContentType: ContentType.json,
-        id: '1',
-      );
-      const expectedCode = """import requests
-
-def main():
-    url = 'https://jsonplaceholder.typicode.com/posts'
-
-    data = '''{"title": "foo","body": "bar","userId": 1}'''
-
-    headers = {
-                  "content-type": "application/json"
-                }
-
-    response = requests.post(
-        url, headers=headers, data=data
-    )
-
-    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()""";
-      expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
-    });
-
-    test('getCode returns valid code for DELETE request', () {
-      const requestModel = RequestModel(
-        url: 'https://jsonplaceholder.typicode.com/posts/1',
-        method: HTTPVerb.delete,
-        requestBody: '{"title": "foo","body": "bar","userId": 1}',
-        requestBodyContentType: ContentType.json,
-        id: '1',
-      );
-      const expectedCode = """import requests
-
-def main():
-    url = 'https://jsonplaceholder.typicode.com/posts/1'
-
-    data = '''{"title": "foo","body": "bar","userId": 1}'''
-
-    headers = {
-                  "content-type": "application/json"
-                }
-
-    response = requests.delete(
-        url, headers=headers, data=data
-    )
-
-    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()""";
-      expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
-    });
-
-    test('getCode returns valid code for HEAD request', () {
-      const requestModel = RequestModel(
-        url: 'https://jsonplaceholder.typicode.com/posts/1',
-        method: HTTPVerb.head,
-        id: '1',
-      );
-      const expectedCode = """import requests
-
-def main():
-    url = 'https://jsonplaceholder.typicode.com/posts/1'
-
-    response = requests.head(
-        url
-    )
-
-    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()""";
-      expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
-    });
-
-    test(
-        'getCode returns valid code for requests with headers and query parameters',
-        () {
-      const requestModel = RequestModel(
-        url: 'https://jsonplaceholder.typicode.com/posts',
-        method: HTTPVerb.get,
-        requestParams: [
-          NameValueModel(name: 'userId', value: 1),
-        ],
-        requestHeaders: [
-          NameValueModel(name: 'Custom-Header-1', value: 'Value-1'),
-          NameValueModel(name: 'Custom-Header-2', value: 'Value-2')
-        ],
-        id: '1',
-      );
-      const expectedCode = """import requests
-
-def main():
-    url = 'https://jsonplaceholder.typicode.com/posts'
-
-    params = {
-        "userId": "1", 
-    }
-
-    headers = {
-                  "Custom-Header-1": "Value-1",
-                  "Custom-Header-2": "Value-2"
-                }
-
-    response = requests.get(
-        url, params=params, headers=headers
-    )
-
-    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()""";
-      expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
-    });
-  });
-}

test/codegen/python_requests_codegen_test.dart

@@ -0,0 +1,327 @@
+import 'package:apidash/codegen/python/pkg_requests.dart';
+import '../request_models.dart';
+import 'package:test/test.dart';
+
+void main() {
+  final pythonRequestsClient = PythonRequestsCodeGen();
+
+  group('GET Request', () {
+    test('GET 1', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com'
+
+response = requests.get(url)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet1, "https"),
+          expectedCode);
+    });
+
+    test('GET 2', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com/country/data'
+
+params = {
+           "code": "US"
+         }
+
+response = requests.get(url, params=params)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet2, "https"),
+          expectedCode);
+    });
+
+    test('GET 3', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com/country/data'
+
+params = {
+           "code": "IND"
+         }
+
+response = requests.get(url, params=params)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet3, "https"),
+          expectedCode);
+    });
+
+    test('GET 4', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com/humanize/social'
+
+params = {
+           "num": "8700000",
+           "digits": "3",
+           "system": "SS",
+           "add_space": "true",
+           "trailing_zeros": "true"
+         }
+
+response = requests.get(url, params=params)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet4, "https"),
+          expectedCode);
+    });
+
+    test('GET 5', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.github.com/repos/foss42/apidash'
+
+headers = {
+            "User-Agent": "Test Agent"
+          }
+
+response = requests.get(url, headers=headers)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet5, "https"),
+          expectedCode);
+    });
+
+    test('GET 6', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.github.com/repos/foss42/apidash'
+
+params = {
+           "raw": "true"
+         }
+
+headers = {
+            "User-Agent": "Test Agent"
+          }
+
+response = requests.get(url, params=params, headers=headers)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet6, "https"),
+          expectedCode);
+    });
+
+    test('GET 7', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com'
+
+response = requests.get(url)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet7, "https"),
+          expectedCode);
+    });
+
+    test('GET 8', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.github.com/repos/foss42/apidash'
+
+params = {
+           "raw": "true"
+         }
+
+headers = {
+            "User-Agent": "Test Agent"
+          }
+
+response = requests.get(url, params=params, headers=headers)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelGet8, "https"),
+          expectedCode);
+    });
+  });
+
+  group('HEAD Request', () {
+    test('HEAD 1', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com'
+
+response = requests.head(url)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelHead1, "https"),
+          expectedCode);
+    });
+
+    test('HEAD 2', () {
+      const expectedCode = r"""import requests
+
+url = 'http://api.foss42.com'
+
+response = requests.head(url)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelHead2, "http"),
+          expectedCode);
+    });
+  });
+
+  group('POST Request', () {
+    test('POST 1', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com/case/lower'
+
+payload = r'''{
+"text": "I LOVE Flutter"
+}'''
+
+headers = {
+            "content-type": "text/plain"
+          }
+
+response = requests.post(url, data=payload, headers=headers)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelPost1, "https"),
+          expectedCode);
+    });
+
+    test('POST 2', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com/case/lower'
+
+payload = {
+"text": "I LOVE Flutter"
+}
+
+response = requests.post(url, json=payload)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelPost2, "https"),
+          expectedCode);
+    });
+
+    test('POST 3', () {
+      const expectedCode = r"""import requests
+
+url = 'https://api.foss42.com/case/lower'
+
+payload = {
+"text": "I LOVE Flutter"
+}
+
+headers = {
+            "User-Agent": "Test Agent"
+          }
+
+response = requests.post(url, json=payload, headers=headers)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelPost3, "https"),
+          expectedCode);
+    });
+  });
+  group('PUT Request', () {
+    test('PUT 1', () {
+      const expectedCode = r"""import requests
+
+url = 'https://reqres.in/api/users/2'
+
+payload = {
+"name": "morpheus",
+"job": "zion resident"
+}
+
+response = requests.put(url, json=payload)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelPut1, "https"),
+          expectedCode);
+    });
+  });
+
+  group('PATCH Request', () {
+    test('PATCH 1', () {
+      const expectedCode = r"""import requests
+
+url = 'https://reqres.in/api/users/2'
+
+payload = {
+"name": "marfeus",
+"job": "accountant"
+}
+
+response = requests.patch(url, json=payload)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelPatch1, "https"),
+          expectedCode);
+    });
+  });
+
+  group('DELETE Request', () {
+    test('DELETE 1', () {
+      const expectedCode = r"""import requests
+
+url = 'https://reqres.in/api/users/2'
+
+response = requests.delete(url)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelDelete1, "https"),
+          expectedCode);
+    });
+
+    test('DELETE 2', () {
+      const expectedCode = r"""import requests
+
+url = 'https://reqres.in/api/users/2'
+
+payload = {
+"name": "marfeus",
+"job": "accountant"
+}
+
+response = requests.delete(url, json=payload)
+
+print('Status Code:', response.status_code)
+print('Response Body:', response.text)
+""";
+      expect(pythonRequestsClient.getCode(requestModelDelete2, "https"),
+          expectedCode);
+    });
+  });
+}