Merge pull request kennknowles#12 from hakanw/jsonpath_script

kennknowles · kennknowles · commit b01e2e5ae7cb · 2014-02-07T17:43:09.000-05:00
Add jsonpath.py script that can be used when installing this package
diff --git a/jsonpath_rw/bin/jsonpath.py b/jsonpath_rw/bin/jsonpath.py
@@ -0,0 +1,51 @@
+#!/usr/bin/python
+# encoding: utf-8
+# Copyright © 2012 Felix Richter <wtfpl@syntax-fehler.de>
+# This work is free. You can redistribute it and/or modify it under the
+# terms of the Do What The Fuck You Want To Public License, Version 2,
+# as published by Sam Hocevar. See the COPYING file for more details.
+
+from jsonpath_rw import parse
+import json
+import sys
+import glob
+if len(sys.argv) < 2:
+    print("""usage: jsonpath.py expression [files]
+
+The expression is JSONPath and can be:
+
+    atomics:
+        $              - root object
+        `this`         - current object
+
+    operators:
+        path1.path2    - same as xpath /
+        path1|path2    - union
+        path1..path2   - somewhere in between
+
+    fiels:
+        fieldname       - field with name
+        *               - any field
+        [_start_?:_end_?] - array slice
+        [*]             - any array index
+""")
+    sys.exit(1)
+
+expr = parse(sys.argv[1])
+
+def find_matches_for_file(f):
+    return [unicode(match.value) for match in expr.find(json.load(f))]
+
+def print_matches(matches):
+    print(u"\n".join(matches).encode("utf-8"))
+
+if len(sys.argv) < 3:
+    # stdin mode
+    print_matches(find_matches_for_file(sys.stdin))
+else:
+    # file paths mode
+    for pattern in sys.argv[2:]:
+        for filename in glob.glob(pattern):
+            with open(filename) as f:
+                print_matches(find_matches_for_file(f))
+
diff --git a/setup.py b/setup.py
@@ -14,6 +14,7 @@
     license='Apache 2.0',
     long_description=io.open('README.rst', encoding='utf-8').read(),
     packages = ['jsonpath_rw'],
+    scripts = ['jsonpath_rw/bin/jsonpath.py'],
     test_suite = 'tests',
     install_requires = [ 'ply', 'decorator', 'six' ],
     classifiers = [
