1

scripts/postman/mocking.py

@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 
+import os
+import sys
 import json
+from urllib import parse as urlparse
 
 class Mocking(object):
 
@@ -9,3 +12,95 @@ class Mocking(object):
 
     def convert(self):
         self._env = json.loads(open(self._workDir + 'env.json', 'r').read())
+        self._projObj = {
+            'info': {
+                '_postman_id': '',
+                'name': self._env['name'],
+                'schema': 'https://schema.getpostman.com/json/collection/v2.1.0/collection.json'
+            },
+            'variable': self._env['variables'],
+            'item': []
+        }
+        sys.path.append(self._workDir + 'collections')
+        for file in os.listdir(self._workDir + 'collections'):
+            if file[0:1] == '_':
+                continue
+            m = __import__(file.replace('.py', ''))
+            for name in dir(m):
+                if name[0:1] != '_':
+                    c = getattr(m, name)()
+                    self._convertCollection(c)
+        self._out()
+
+    def _out(self):
+        wf = open(self._workDir + self._env['name'] + '.postman.json', 'w')
+        wf.write(json.dumps(self._projObj, indent = 4))
+        wf.close()
+
+    def _convertCollection(self, c):
+        for item in c.items:
+            itemObj = {
+                'name': item.get('name', ''),
+                'event': [],
+                'request': {
+                    'method': item.get('method', 'GET'),
+                    'header': [],
+                    'url': {
+                        'raw': self._getRaw(item),
+                        'protocol': self._getProtocol(item['url']),
+                        'host': self._getHost(item['url']),
+                        'path': self._getPath(item['url']),
+                        'query': []
+                    }
+                },
+                'response': []
+            }
+            if 'prerequest' in item:
+                itemObj['event'].append({
+                    'listen': 'prerequest',
+                    'script': {
+                        'exec': [
+                            item['prerequest'].strip()
+                        ],
+                        'type': 'text/javascript'
+                    }
+                })
+            if 'test' in item:
+                itemObj['event'].append({
+                    'listen': 'test',
+                    'script': {
+                        'exec': [
+                            item['test'].strip()
+                        ],
+                        'type': 'text/javascript'
+                    }
+                })
+            for param in item['params']:
+                itemObj['request']['url']['query'].append({
+                    'key': param[0],
+                    'value': param[1],
+                })
+            self._projObj['item'].append(itemObj)
+
+    def _getRaw(self, item):
+        raw = item['url']
+        for param in item['params']:
+            if raw.index('?') == -1:
+                raw += '?'
+            if raw[-1] in ('&', '?'):
+                raw += param[0] + '=' + param[1]
+            else:
+                raw += '&' + param[0] + '=' + param[1]
+        return raw
+
+    def _getProtocol(self, url):
+        parsed_tuple = urlparse.urlparse(url)
+        return parsed_tuple.scheme
+
+    def _getHost(self, url):
+        parsed_tuple = urlparse.urlparse(url)
+        return [parsed_tuple.netloc]
+
+    def _getPath(self, url):
+        parsed_tuple = urlparse.urlparse(url)
+        return [path for path in parsed_tuple.path.split('/') if path != '']