Make into a package, split code and tests

Author: pradyunsg

json_minify.py

@@ -0,0 +1,62 @@
+"""A port of the `JSON-minify` utility to the Python language.
+
+Based on JSON.minify.js: https://github.com/getify/JSON.minify
+
+Contributers:
+  - Gerald Storer
+    - Contributed original version
+  - Felipe Machado
+    - Performance optimization
+  - Pradyun S. Gedam
+    - Conditions and variable names changed
+    - Reformatted tests and moved to separate file
+    - Made into a PyPI Package
+"""
+
+import re
+
+
+def json_minify(string, strip_space=True):
+    tokenizer = re.compile('"|(/\*)|(\*/)|(//)|\n|\r')
+    end_slashes_re = re.compile(r'(\\)*$')
+
+    in_string = False
+    in_multi = False
+    in_single = False
+
+    new_str = []
+    index = 0
+
+    for match in re.finditer(tokenizer, string):
+
+        if not (in_multi or in_single):
+            tmp = string[index:match.start()]
+            if not in_string and strip_space:
+                # replace white space as defined in standard
+                tmp = re.sub('[ \t\n\r]+', '', tmp)
+            new_str.append(tmp)
+
+        index = match.end()
+        val = match.group()
+
+        if val == '"' and not (in_multi or in_single):
+            escaped = end_slashes_re.search(string, 0, match.start())
+
+            # start of string or unescaped quote character to end string
+            if not in_string or (escaped is None or len(escaped.group()) % 2 == 0):  # noqa
+                in_string = not in_string
+            index -= 1  # include " character in next catch
+        elif not (in_string or in_multi or in_single):
+            if val == '/*':
+                in_multi = True
+            elif val == '//':
+                in_single = True
+        elif val == '*/' and in_multi and not (in_string or in_single):
+            in_multi = False
+        elif val in '\r\n' and not (in_multi or in_string) and in_single:
+            in_single = False
+        elif not ((in_multi or in_single) or (val in ' \r\n\t' and strip_space)):  # noqa
+            new_str.append(val)
+
+    new_str.append(string[index:])
+    return ''.join(new_str)

json_minify/__init__.py

@@ -0,0 +1,86 @@
+"""Tests for json_minify
+"""
+
+# Python 2.6+ needed to run tests
+import json
+import textwrap
+import unittest
+
+from json_minify import json_minify
+
+
+class JsonMinifyTestCase(unittest.TestCase):
+    """Tests for json_minify"""
+
+    tests = [
+        [
+            '''
+            // this is a JSON file with comments
+            {
+                "foo": "bar",    // this is cool
+                "bar": [
+                    "baz", "bum"
+                ],
+            /* the rest of this document is just fluff
+               in case you are interested. */
+                "something": 10,
+                "else": 20
+            }
+
+            /* NOTE: You can easily strip the whitespace and comments
+               from such a file with the JSON.minify() project hosted
+               here on github at http://github.com/getify/JSON.minify
+            */''',
+            '{"foo":"bar","bar":["baz","bum"],"something":10,"else":20}'
+        ],
+        [
+            '''
+            {"/*":"*/","//":"",/*"//"*/"/*/"://
+            "//"}''',
+            '{"/*":"*/","//":"","/*/":"//"}'
+        ],
+        [
+            r'''
+            /*
+            this is a
+            multi line comment */{
+
+            "foo"
+            :
+                "bar/*"// something
+                ,    "b\"az":/*
+            something else */"blah"
+
+            }
+            ''',
+            r'{"foo":"bar/*","b\"az":"blah"}'
+        ],
+        [
+            r'''
+            {"foo": "ba\"r//", "bar\\": "b\\\"a/*z",
+            "baz\\\\": /* yay */ "fo\\\\\"*/o"
+            }
+            ''',
+            r'{"foo":"ba\"r//","bar\\":"b\\\"a/*z","baz\\\\":"fo\\\\\"*/o"}'  # noqa
+        ]
+    ]
+
+    def template(self, in_string, expected):
+        in_dict = json.loads(json_minify(in_string))
+        expected_dict = json.loads(textwrap.dedent(expected))
+        self.assertEqual(in_dict, expected_dict)
+
+    def test_1(self):
+        self.template(*self.tests[0])
+
+    def test_2(self):
+        self.template(*self.tests[1])
+
+    def test_3(self):
+        self.template(*self.tests[2])
+
+    def test_4(self):
+        self.template(*self.tests[3])
+
+if __name__ == '__main__':
+    unittest.main()

json_minify/test_json_minify.py

@@ -20,7 +20,7 @@
     maintainer="Pradyun S. Gedam",
     maintainer_email="pradyunsg@gmail.com",
 
-    py_modules=["json_minify"],
+    packages=["json_minify"],
 
     license="MIT",
     classifiers=[