Merge pull request #24 from montanaflynn/master

Add Python target

Author: ahmadnassri

README.md

@@ -34,6 +34,8 @@ currently the following output [targets](/src/targets) are supported:
   - [CoHTTP](https://github.com/mirage/ocaml-cohttp)
 - PHP
   - [ext-curl](http://php.net/manual/en/book.curl.php)
+- Python
+  - [Python 3](https://docs.python.org/3/library/http.client.html)
 - [Wget](https://www.gnu.org/software/wget/)
 
 ## Installation

src/targets/go/native.js

@@ -44,54 +44,57 @@ module.exports = function (source, options) {
 
   code.push(')\n')
 
-  code.push('func main() {')
+  code.push('func main() {\n')
 
   // Create client
   if (opts.timeout > 0) {
     code.push('\tclient := http.Client{')
     code.push(util.format('\t\tTimeout: time.Duration(%s * time.Second),', opts.timeout))
-    code.push('\t}')
+    code.push('\t}\n')
   } else {
-    code.push('\tclient := &http.Client{}')
+    code.push('\tclient := &http.Client{}\n')
   }
 
-  code.push(util.format('\turl := "%s"', source.fullUrl))
+  code.push(util.format('\turl := "%s"\n', source.fullUrl))
 
   // If we have body content or not create the var and reader or nil
   if (source.postData.text) {
-    code.push(util.format('\tpayload := strings.NewReader(%s)', JSON.stringify(source.postData.text)))
-    code.push(util.format('\treq, %s := http.NewRequest("%s", url, payload)', errorPlaceholder, source.method))
+    code.push(util.format('\tpayload := strings.NewReader(%s)\n', JSON.stringify(source.postData.text)))
+    code.push(util.format('\treq, %s := http.NewRequest("%s", url, payload)\n', errorPlaceholder, source.method))
   } else {
-    code.push(util.format('\treq, %s := http.NewRequest("%s", url, nil)', errorPlaceholder, source.method))
+    code.push(util.format('\treq, %s := http.NewRequest("%s", url, nil)\n', errorPlaceholder, source.method))
   }
 
   errorCheck()
 
   // Add headers
-  Object.keys(source.allHeaders).map(function (key) {
-    code.push(util.format('\treq.Header.Add("%s", "%s")', key, source.allHeaders[key]))
-  })
+  if (Object.keys(source.allHeaders).length) {
+    Object.keys(source.allHeaders).map(function (key) {
+      code.push(util.format('\treq.Header.Add("%s", "%s")', key, source.allHeaders[key]))
+    })
+    code.push(null)
+  }
 
   // Make request
   code.push(util.format('\tres, %s := client.Do(req)', errorPlaceholder))
   errorCheck()
 
   // Get Body
   if (opts.printBody) {
-    code.push('\tdefer res.Body.Close()')
+    code.push('\n\tdefer res.Body.Close()')
     code.push(util.format('\tbody, %s := ioutil.ReadAll(res.Body)', errorPlaceholder))
     errorCheck()
   }
 
   // Print it
-  code.push('\tfmt.Println(res)')
+  code.push('\n\tfmt.Println(res)')
 
   if (opts.printBody) {
     code.push('\tfmt.Println(string(body))')
   }
 
   // End main block
-  code.push('}')
+  code.push('\n}')
 
   return code.join('\n')
 }

src/targets/python/index.js

@@ -0,0 +1,3 @@
+'use strict'
+
+module.exports = require('require-directory')(module)

src/targets/python/info.js

@@ -0,0 +1,8 @@
+'use strict'
+
+module.exports = {
+  key: 'python',
+  title: 'Python',
+  extname: '.py',
+  default: 'python3'
+}

src/targets/python/python3.js

@@ -0,0 +1,75 @@
+'use strict'
+
+var util = require('util')
+
+module.exports = function (source, options) {
+  // Start Request
+  var code = []
+  code.push('import http.client\n')
+
+  // Check which protocol to be used for the client connection
+  var protocol = source.uriObj.protocol
+  if (protocol === 'https:') {
+    code.push(util.format('conn = http.client.HTTPSConnection("%s")\n', source.uriObj.host))
+  } else {
+    code.push(util.format('conn = http.client.HTTPConnection("%s")\n', source.uriObj.host))
+  }
+
+  // Create payload string if it exists
+  var payload = JSON.stringify(source.postData.text)
+  if (payload) {
+    code.push(util.format('payload = %s\n', payload))
+  }
+
+  // Create Headers
+  var header
+  var headers = source.allHeaders
+  var headerCount = Object.keys(headers).length
+  if (headerCount === 1) {
+    for (header in headers) {
+      code.push(util.format('headers = { \'%s\': "%s" }\n', header, headers[header]))
+    }
+  } else if (headerCount > 1) {
+    var headerLine
+    var count = 1
+    code.push('headers = {')
+    for (header in headers) {
+      if (count++ !== headerCount) {
+        headerLine = util.format('    \'%s\': "%s",', header, headers[header])
+      } else {
+        headerLine = util.format('    \'%s\': "%s"', header, headers[header])
+      }
+      code.push(headerLine)
+    }
+    code.push('    }\n')
+  }
+
+  // Make Request
+  var method = source.method
+  var path = source.uriObj.path
+  if (payload && headerCount) {
+    code.push(util.format('conn.request("%s", "%s", payload, headers)', method, path))
+  } else if (payload && !headerCount) {
+    code.push(util.format('conn.request("%s", "%s", payload)', method, path))
+  } else if (!payload && headerCount) {
+    code.push(util.format('conn.request("%s", "%s", headers = headers)', method, path))
+  } else {
+    code.push(util.format('conn.request("%s", "%s")', method, path))
+  }
+
+  // Get Response
+  code.push('\nres = conn.getresponse()')
+  code.push('data = res.read()')
+  code.push('\nprint(res.status)')
+  code.push('print(data)')
+
+  // console.log(code)
+  return code.join('\n')
+}
+
+module.exports.info = {
+  key: 'python3',
+  title: 'http.client',
+  link: 'https://docs.python.org/3/library/http.client.html',
+  description: 'Python3 HTTP Client'
+}

test/fixtures/available-targets.json

@@ -74,6 +74,20 @@
       }
     ]
   },
+  {
+    "key": "python",
+    "title": "Python",
+    "extname": ".py",
+    "default": "python3",
+    "clients": [
+      {
+        "key": "python3",
+        "title": "http.client",
+        "link": "https://docs.python.org/3/library/http.client.html",
+        "description": "Python3 HTTP Client"
+      }
+    ]
+  },
   {
     "key": "objc",
     "title": "Objective-C",

test/fixtures/output/go/native/application-form-encoded.go

@@ -8,14 +8,23 @@ import (
 )
 
 func main() {
+
 	client := &http.Client{}
+
 	url := "http://mockbin.com/har"
+
 	payload := strings.NewReader("foo=bar&hello=world")
+
 	req, _ := http.NewRequest("POST", url, payload)
+
 	req.Header.Add("content-type", "application/x-www-form-urlencoded")
+
 	res, _ := client.Do(req)
+
 	defer res.Body.Close()
 	body, _ := ioutil.ReadAll(res.Body)
+
 	fmt.Println(res)
 	fmt.Println(string(body))
+
 }

test/fixtures/output/go/native/application-json.go

@@ -8,14 +8,23 @@ import (
 )
 
 func main() {
+
 	client := &http.Client{}
+
 	url := "http://mockbin.com/har"
+
 	payload := strings.NewReader("{\"number\": 1, \"string\": \"f\\\"oo\", \"arr\": [1, 2, 3], \"nested\": {\"a\": \"b\"}, \"arr_mix\": [1, \"a\", {\"arr_mix_nested\": {}}] }")
+
 	req, _ := http.NewRequest("POST", url, payload)
+
 	req.Header.Add("content-type", "application/json")
+
 	res, _ := client.Do(req)
+
 	defer res.Body.Close()
 	body, _ := ioutil.ReadAll(res.Body)
+
 	fmt.Println(res)
 	fmt.Println(string(body))
+
 }

test/fixtures/output/go/native/cookies.go

@@ -7,13 +7,21 @@ import (
 )
 
 func main() {
+
 	client := &http.Client{}
+
 	url := "http://mockbin.com/har"
+
 	req, _ := http.NewRequest("POST", url, nil)
+
 	req.Header.Add("cookie", "foo=bar; bar=baz")
+
 	res, _ := client.Do(req)
+
 	defer res.Body.Close()
 	body, _ := ioutil.ReadAll(res.Body)
+
 	fmt.Println(res)
 	fmt.Println(string(body))
+
 }

test/fixtures/output/go/native/full.go

@@ -8,16 +8,25 @@ import (
 )
 
 func main() {
+
 	client := &http.Client{}
+
 	url := "http://mockbin.com/har?foo=bar&foo=baz&baz=abc&key=value"
+
 	payload := strings.NewReader("foo=bar")
+
 	req, _ := http.NewRequest("POST", url, payload)
+
 	req.Header.Add("cookie", "foo=bar; bar=baz")
 	req.Header.Add("accept", "application/json")
 	req.Header.Add("content-type", "application/x-www-form-urlencoded")
+
 	res, _ := client.Do(req)
+
 	defer res.Body.Close()
 	body, _ := ioutil.ReadAll(res.Body)
+
 	fmt.Println(res)
 	fmt.Println(string(body))
+
 }