From 1c2963a1e0064a7d76e7f4d6c114cb45f95c1ba0 Mon Sep 17 00:00:00 2001
From: YE Qing <2chin.yip@gmail.com>
Date: Wed, 7 Oct 2020 01:26:10 +0800
Subject: [PATCH] add codegen for python-requests (#1244)

Co-authored-by: Qing Ye <ye.qing@go-jek.com>
Co-authored-by: Liyas Thomas <liyascthomas@gmail.com>
---
 helpers/codegen/codegen.js                    |  2 +
 helpers/codegen/generators/python-requests.js | 96 +++++++++++++++++++
 2 files changed, 98 insertions(+)
 create mode 100644 helpers/codegen/generators/python-requests.js

diff --git a/helpers/codegen/codegen.js b/helpers/codegen/codegen.js
index 9ca72464d..2ac7956b0 100644
--- a/helpers/codegen/codegen.js
+++ b/helpers/codegen/codegen.js
@@ -7,6 +7,7 @@ import { NodeJsRequestCodegen } from "./generators/nodejs-request"
 import { JSjQueryCodegen } from "./generators/js-jQuery"
 import { PowerShellRestMethod } from "./generators/powershell"
 import { PhpCurlCodegen } from "./generators/php-curl"
+import { PythonRequestsCodegen } from "./generators/python-requests"
 
 /* Register code generators here.
  * A code generator is defined as an object with the following structure.
@@ -26,6 +27,7 @@ export const codegens = [
   NodeJsRequestCodegen,
   PhpCurlCodegen,
   PowerShellRestMethod,
+  PythonRequestsCodegen,
 ]
 
 export function generateCodeWithGenerator(codegenID, context) {
diff --git a/helpers/codegen/generators/python-requests.js b/helpers/codegen/generators/python-requests.js
new file mode 100644
index 000000000..24cdc9657
--- /dev/null
+++ b/helpers/codegen/generators/python-requests.js
@@ -0,0 +1,96 @@
+import { isJSONContentType } from "~/helpers/utils/contenttypes"
+
+const printHeaders = (headers) => {
+  if (headers.length) {
+    return [`headers = {\n`, `  ${headers.join(",\n  ")}\n`, `}\n`]
+  }
+
+  return []
+}
+
+export const PythonRequestsCodegen = {
+  id: "python-requests",
+  name: "Python Requests",
+  generator: ({
+    url,
+    pathName,
+    queryString,
+    auth,
+    httpUser,
+    httpPassword,
+    bearerToken,
+    method,
+    rawInput,
+    rawParams,
+    rawRequestBody,
+    contentType,
+    headers,
+  }) => {
+    const requestString = []
+    let genHeaders = []
+
+    requestString.push(`import requests\n\n`)
+    requestString.push(`url = '${url}${pathName}${queryString}'\n`)
+
+    // auth headers
+    if (auth === "Basic Auth") {
+      const basic = `${httpUser}:${httpPassword}`
+      genHeaders.push(
+        `'Authorization': 'Basic ${window.btoa(unescape(encodeURIComponent(basic)))}'`
+      )
+    } else if (auth === "Bearer Token" || auth === "OAuth 2.0") {
+      genHeaders.push(`'Authorization': 'Bearer ${bearerToken}'`)
+    }
+
+    // custom headers
+    if (headers.length) {
+      headers.forEach(({ key, value }) => {
+        if (key) genHeaders.push(`'${key}': '${value}'`)
+      })
+    }
+
+    // initial request setup
+    let requestBody = rawInput ? rawParams : rawRequestBody
+    if (method == "GET") {
+      requestString.push(...printHeaders(genHeaders))
+      requestString.push(`response = requests.request(\n`)
+      requestString.push(`  '${method}',\n`)
+      requestString.push(`  '${url}${pathName}${queryString}',\n`)
+    }
+    if (["POST", "PUT", "PATCH", "DELETE"].includes(method)) {
+      genHeaders.push(`'Content-Type': '${contentType}'`)
+      requestString.push(...printHeaders(genHeaders))
+
+      if (isJSONContentType(contentType)) {
+        requestBody = JSON.stringify(requestBody)
+        requestString.push(`data = ${requestBody}\n`)
+      } else if (contentType.includes("x-www-form-urlencoded")) {
+        const formData = []
+        if (requestBody.indexOf("=") > -1) {
+          requestBody.split("&").forEach((rq) => {
+            const [key, val] = rq.split("=")
+            formData.push(`('${key}', '${val}')`)
+          })
+        }
+        if (formData.length) {
+          requestString.push(`data = [${formData.join(",\n      ")}]\n`)
+        }
+      } else {
+        requestString.push(`data = '''${requestBody}'''\n`)
+      }
+      requestString.push(`response = requests.request(\n`)
+      requestString.push(`  '${method}',\n`)
+      requestString.push(`  '${url}${pathName}${queryString}',\n`)
+      requestString.push(`  data=data,\n`)
+    }
+
+    if (genHeaders.length) {
+      requestString.push(`  headers=headers,\n`)
+    }
+
+    requestString.push(`)\n\n`)
+    requestString.push(`print(response)`)
+
+    return requestString.join("")
+  },
+}