hoppscotch/helpers/codegen/generators/python-requests.js

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("")
  },
}